Redis setNx实现分布式锁的坑

因为redis高效、原子等特性,redis是分布式锁实现中的一种方式,最常见的锁核心的代码如下


@Override
    public Boolean setNx(final IKeyBuilder key, final Serializable value, final int expire) throws Exception {
        return this.redisTemplate.execute(new RedisCallback<Boolean>() {
            @Override
            public Boolean doInRedis(RedisConnection conn)
                    throws DataAccessException {
                byte[] keyBytes = serializeKey(key.getKey());
                Boolean result = conn.setNX(keyBytes, serialize(value));
                if (result){
                    conn.expire(keyBytes, expire);
                }
                return result;
            }
        });
    }

不仔细观察这段代码或者没碰到是不会留意setNx居然还有坑,因为redis给我们的印象就是,高效性、原子性,根本就不会留意到setNx还有坑?对,setNx本身是一个原子性操作,但是通常会给key设置一个expire时间,这两步并不是一个原子操作
因此上面的代码块,可能会出现死锁,原因是在setNx操作成功之后,此时如果发生异常,导致expire设置失败,这个时候setNx设置的key的TTL值为-1,这个值将永远不会失效,并且redis不会主动删除这个设置,导致其他资源无法获取锁。
为了解决上述问题,Redis2.6.12以上版本,可以用set获取锁。set可以实现setnx和expire,这个是原子操作。

SET key value [EX seconds] [PX millisecounds] [NX|XX]
EX seconds:设置键的过期时间为second秒
PX millisecounds:设置键的过期时间为millisecounds 毫秒
NX:只在键不存在的时候,才对键进行设置操作
XX:只在键已经存在的时候,才对键进行设置操作
SET操作成功后,返回的是OK,失败返回NIL

对应的JAVA代码如下:

@Override
    public Boolean setNx(final IKeyBuilder key, final Serializable value, final int expire) throws Exception {
        return this.redisTemplate.execute(new RedisCallback<Boolean>() {
            @Override
            public Boolean doInRedis(RedisConnection conn)
                    throws DataAccessException {
                byte[] keyBytes = serializeKey(key.getKey());
                return conn.set(keyBytes, serialize(value),Expiration.seconds(expire), RedisStringCommands.SetOption.ifAbsent());
            }
        });
    }

set方法中的参数及含义

@Nullable
    Boolean set(byte[] var1, byte[] var2, Expiration var3, RedisStringCommands.SetOption var4);

var1:是要设置的key
var2:设置key的value值
var3:Expiration是一个对象,有两个字段分别是过期时间值和过期时间单位,默认秒
var4:RedisStringCommands.SetOption 是一个枚举变量,具体内容如下:

public static enum SetOption {
        
       **
         * Do not set any additional command argument.
         * 
         * @return
         */
        UPSERT,

        /**
         * {@code NX}
         * 只在键不存在的时候,才对键进行设置操作
         * @return
         */
        SET_IF_ABSENT,

        /**
         * {@code XX}
         * 只在键已经存在的时候,才对键进行设置操作
         * @return
         */
        SET_IF_PRESENT;

        private SetOption() {
        }

        public static RedisStringCommands.SetOption upsert() {
            return UPSERT;
        }

        public static RedisStringCommands.SetOption ifPresent() {
            return SET_IF_PRESENT;
        }

        public static RedisStringCommands.SetOption ifAbsent() {
            return SET_IF_ABSENT;
        }
    }

因此用redis做分布式锁时,请使用set命令。

可关注公众号查看更多文章。

原文链接:

https://mp.weixin.qq.com/s/rpYEOFPz5UlxvHHZsl0TsA

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容