redis实现一个简单的分布式锁

1.准备工作：SETNX 是 Redis SET 命令的一种选项，用于在键不存在时才进行设置。
可以利用redis这个命令的特效来实现分布式锁的效果,如果键存在则返回0，不存在则返回1

2.大致思路：

上锁：
1.Thread.currentThread().getId()先通过这个获取当前线程的id
2.然后通过set nx来保存这个线程id值,并设置缓存的时间

这样做的目的是，只有第一个尝试设置键的线程会成功，后续的线程将无法设置该键。通过缓存当前线程的ID，并在释放锁时进行验证，确保只有当前线程才能释放锁

释放锁
1.先查询刚刚创建的缓存id值
2.判断这个id是否为当前线程id
3.如果为当前线程id则通过删除这个缓存来释放锁

释放锁的操作用lua脚本来写：这样做的目的是保证redis操作的原子性，防止多线程并发锁误删的问题。

3.编写获取锁的代码

这里利用RedisTemplate的一个方法：setIfAbsent 这个方法等同于setnx操作

缓存当前线程的ID，并在释放锁时进行验证，确保只有当前线程才能释放锁

   public Boolean tryLock(String key){
        long id = Thread.currentThread().getId();

        System.out.println(id);

        return stringRedisTemplate.opsForValue().setIfAbsent(key,String.valueOf(id),60, TimeUnit.SECONDS);
    }

4.编写释放锁的代码

SETNX如果键存在则返回0，不存在则返回1

可以利用这个redis命令的特性来编写释放锁的lua脚本：

--key
local klock = KEYS[1]
--value：线程id
local vid = ARGV[1]

--判断当前缓存id是否为当前线程id
if (redis.call('get',klock) == vid)  then

--如果是则执行del删除这个缓存以达到释放锁的效果
return redis.call('del',klock)

end

return 0

5.然后通过execute方法来执行lua脚本：

execute方法的三个参数 ：

public <T> T execute(RedisScript<T> script, List<K> keys, Object... args) {
        return this.scriptExecutor.execute(script, keys, args);
    }

1.可以看到第一个参数是RedisScript：

这里需要new一个RedisScript对象，我们点进这个类找到他的实现类DefaultRedisScript

 可以看到这个类有这两个属性很重要

//lua文件的路径
private ScriptSource scriptSource;

//返回值类型
private Class<T> resultType;

这里调用setLocation方法和setResultType方法给这两个属性传值

private static final DefaultRedisScript<Long> defaultRedisScript;

static{
	//new出对象
        defaultRedisScript = new DefaultRedisScript();
	//调用setLocation方法来指定lua文件的路径
        defaultRedisScript.setLocation(new ClassPathResource("unlock.lua"));
	//设置lua  return的返回值类型
        defaultRedisScript.setResultType(long.class);
 }

2.第二个参数是List<K> keys

需要一个list，可以调用这个方法 List<String> list = Collections.singletonList(key);把字符key转成字符集合key

3.第三个参数是 Object... args 

这个直接传一个value值进去就可，这里直接传线程的id，然后转换成字符串string.valueof(id)

最后释放锁的代码就长这样了：

 public void unLock(String key){

        //获取当前线程id
        long id = Thread.currentThread().getId();

        //通过execute方法执行lua脚本，传入参数key，value-id
        stringRedisTemplate.execute(defaultRedisScript, Collections.singletonList(key), String.valueOf(id));
    }

总结：

public class lockTest {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    public Boolean tryLock(String key){

        long id = Thread.currentThread().getId();

        System.out.println(id);

        return stringRedisTemplate.opsForValue().setIfAbsent(key,String.valueOf(id),60, TimeUnit.SECONDS);
    }

    private static final DefaultRedisScript<Long> defaultRedisScript;

    static {
        defaultRedisScript = new DefaultRedisScript<>();
        defaultRedisScript.setLocation(new ClassPathResource("unlock.lua"));
        defaultRedisScript.setResultType(long.class);
    }
    /**
     * 释放锁
     * @param key
     * @return
     */
    public void unLock(String key){

        //获取当前线程id
        long id = Thread.currentThread().getId();

        //通过execute方法执行lua脚本，传入参数key，value-id
        stringRedisTemplate.execute(defaultRedisScript, Collections.singletonList(key), String.valueOf(id));
    }