因为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命令。
可关注公众号查看更多文章。