Redis实现分布式锁

redis实现分布式锁

---

1.典型错误的做法

下面做法正常情况可以实现分布式锁,但是极端情况下,如果执行了setnx但是系统挂掉了,没有执行expire. 那么就会导致锁没法过期,如果这时候使用的锁的一方刚好有问题,那就成死锁了,锁永远无法释放.

    Long ok = jedis.setnx(lockKey, "OK");
    if(ok==1){
        jedis.expire(lockKey,fc.initNum());
    }

2.正确做法一:jedis

在Redis2.6之后,Redis就已经提供了原子方法去实现nx + expire的方法.NX表示不存在才会去设置,EX表示过期时间单位是秒

    String res = jedis.set(lockKey, "OK", "NX", "EX", currentNum);

源码:

     /**
       * Set the string value as value of the key. The string can't be longer than 1073741824 bytes (1
       * GB).
       * @param key
       * @param value
       * @param nxxx NX|XX, NX -- Only set the key if it does not already exist. XX -- Only set the key
       *          if it already exist.
       * @param expx EX|PX, expire time units: EX = seconds; PX = milliseconds
       * @param time expire time in the units of <code>expx</code>
       * @return Status code reply
       */
      public String set(final String key, final String value, final String nxxx, final String expx,
          final long time) {
        checkIsInMultiOrPipeline();
        client.set(key, value, nxxx, expx, time);
        return client.getStatusCodeReply();
      }

3.正确做法二:Lua脚本

当使用RedisTemplate的时候,由于没有jedis那种直接提供的方法,使用lua脚本进行原子化操作.

    if redis.call('setNx',KEYS[1],ARGV[1])==1 then
        if redis.call('get',KEYS[1])==ARGV[1] then
            return redis.call('expire',KEYS[1],ARGV[2])
        else
            return 0
        end
    else
        return 0
    end

java代码

     String script = "if redis.call('setNx',KEYS[1],ARGV[1])==1 then\n" +
                    "    if redis.call('get',KEYS[1])==ARGV[1] then\n" +
                    "        return redis.call('expire',KEYS[1],ARGV[2])\n" +
                    "    else\n" +
                    "        return 0\n" +
                    "    end\n" +
                    "else\n" +
                    "    return 0\n" +
                    "end";
            RedisScript<Long> redisScript = new DefaultRedisScript<>(script, Long.class);
            Long result = template.execute(redisScript, Collections.singletonList(lockKey),"2019",String.valueOf(fc.initNum()));
            logger.info("lua lock result="+result);
            return result==1;