import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;
import java.util.Collections;
@Service
public class RedisService {
private final StringRedisTemplate stringRedisTemplate;
private final DefaultRedisScript<Long> lockScript;
private final DefaultRedisScript<Long> unlockScript;
public RedisService(StringRedisTemplate stringRedisTemplate) {
this.stringRedisTemplate = stringRedisTemplate;
// 定义Lua脚本用于加锁
String lockScriptSource = "if redis.call('exists', KEYS[1]) == 0 then redis.call('hset', KEYS[1], ARGV[1], 1); redis.call('pexpire', KEYS[1], ARGV[2]); return 1; else return 0; end";
lockScript = new DefaultRedisScript<>();
lockScript.setScriptText(lockScriptSource);
lockScript.setResultType(Long.class);
// 定义Lua脚本用于解锁
String unlockScriptSource = "if redis.call('hexists', KEYS[1], ARGV[1]) == 1 then redis.call('hdel', KEYS[1], ARGV[1]); return 1; else return 0; end";
unlockScript = new DefaultRedisScript<>();
unlockScript.setScriptText(unlockScriptSource);
unlockScript.setResultType(Long.class);
}
public boolean lock(String lockKey, String requestId, int expireTime) {
Long result = stringRedisTemplate.execute(lockScript, Collections.singletonList(lockKey),
Collections.singletonList(requestId), Collections.singletonList(String.valueOf(expireTime)));
return result != null && result.intValue() == 1;
}
public boolean unlock(String lockKey, String requestId) {
Long result = stringRedisTemplate.execute(unlockScript, Collections.singletonList(lockKey),
Collections.singletonList(requestId));
return result != null && result.intValue() == 1;
}
}这段代码示例展示了如何使用Spring Data Redis的StringRedisTemplate和DefaultRedisScript来实现分布式锁。通过Lua脚本来保证在解锁时的原子性，避免了由于网络分区或其他问题导致的解锁失败问题。这是一个简化的示例，但在实际应用中，你可能需要添加更多的错误处理和重试逻辑。