java常用工具

token 机制实现幂等性
幂等性工具

import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.serializer.StringRedisSerializer;
import org.springframework.stereotype.Component;

import java.util.Collections;
import java.util.List;

@Component
public class IdempotentTokenUtil {

    private final RedisTemplate<String, String> redisTemplate;
    // 定义 Lua 脚本
    private final String LUA_SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

    public IdempotentTokenUtil(RedisTemplate<String, String> redisTemplate) {
        this.redisTemplate = redisTemplate;
        // 设置 Redis 序列化方式（避免键值乱码）
        this.redisTemplate.setKeySerializer(new StringRedisSerializer());
        this.redisTemplate.setValueSerializer(new StringRedisSerializer());
    }

    /**
     * 校验并删除幂等性 Token
     * @param tokenKey Redis 键名
     * @param userId   当前请求的用户 ID
     * @return true=Token 有效，false=Token 无效
     */
    public boolean validateAndDeleteToken(String tokenKey, String userId) {
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptText(LUA_SCRIPT);
        redisScript.setResultType(Long.class); // 脚本返回值类型为 Long

        // 执行脚本：KEYS 传 tokenKey，ARGV 传 userId
        List<String> keys = Collections.singletonList(tokenKey);
        Long result = redisTemplate.execute(redisScript, keys, userId);

        // 返回 1 表示校验成功，否则失败
        return result != null && result == 1;
    }
}