Redis学习（12）｜使用Redis在Spring Boot中实现分布式锁

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;
 
import java.util.Collections;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
 
@Component
public class RedisLock {
 
    @Autowired
    private StringRedisTemplate redisTemplate;
 
    @Autowired
    private DefaultRedisScript<Long> lockScript;
 
    @Autowired
    private DefaultRedisScript<String> unlockScript;
 
    public boolean lock(String key, long expireTime) {
        String value = UUID.randomUUID().toString();
        Long result = redisTemplate.execute(lockScript, Collections.singletonList(key), value, String.valueOf(expireTime));
        return result != null && result.equals(1L);
    }
 
    public boolean unlock(String key, String value) {
        String result = redisTemplate.execute(unlockScript, Collections.singletonList(key), value);
        return "OK".equals(result);
    }
}
 
// Lua 脚本用于加锁
String lockScript = "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 elseif redis.call('hexists', KEYS[1], ARGV[1]) == 1 then return 0 else return -1 end";
 
// Lua 脚本用于解锁
String unlockScript = "if redis.call('hexists', KEYS[1], ARGV[1]) == 1 then redis.call('hdel', KEYS[1], ARGV[1]) return 'OK' else return 'INVALID_REQUEST' end";

在这个代码实例中，我们使用了Lua脚本来实现Redis分布式锁。这样做的好处是可以保证加锁和解锁的操作是原子的，避免了在解锁之前有其他线程获得锁的情况发生。同时，我们使用了UUID来生成唯一的客户端标识，避免了因为客户端重复请求导致的问题。这个实现是简化版的，但是它展示了如何在Spring Boot应用程序中使用Redis实现分布式锁的基本概念。