如何用Redis实现一个分布式锁?
1 什么是分布式锁
mp.weixin.qq.com/s/8fdBKAyHZ…
2 Redis实现分布式锁
2.1 加锁
最开始的分布式锁是使用setnx+expire命令来实现的。setnx设置成功返回1,表示获取到锁,返回0,表示没有获取到锁,同时为了避免显示释放锁失败,导致资源永远也不释放,获取到锁后还会用expire命令设置锁超时的时间。
但有个问题就是setnx+expire不是原子性的,有可能获取到锁后,还没执行expire命令,也没执行释放锁的操作,服务就挂了,这样这个资源就永远也不会访问到了。
为了解决这个问题,Redis 2.6.12版本以后,为set命令增加了一系列的参数,我们此时用NX和PX参数就可以解决这个问题。
所以现在Redis分布式锁的加锁命令如下:
SET resource_key random_value NX PX 30000
NX只会在key不存在的时候给key赋值,PX通知Redis保存这个key 30000ms,当资源被锁定超过这个时间时,锁将自动释放 random_value最好是全局唯一的值,保证释放锁的安全性
设置成功返回OK:
127.0.0.1:6379> SET lock1 100 NX PX 30000
OK
127.0.0.1:6379> SET lock1 100 NX PX 30000
(nil)
当某个key不存在时才能设置成功。这就可以让多个并发线程同时去设置同一个key,只有一个能设置成功。而其他线程设置失败,也就是获得锁失败
2.2 释放锁
释放锁不能简单的使用如下命令:
del resource_key
因为有可能节点A加锁后执行超时,锁被释放了。节点B又重新加锁,A正常执行到del命令的话就把节点B的锁给释放了。所以在解锁之前先判断一下是不是自己加的锁,是自己加的锁再释放,不是就不释放。所以伪代码如下:
if (random_value .equals(redisClient.get(resource_key))) {
del(key)
}
因为判断和解锁是2个独立的操作,不具有原子性,还是有可能会出问题。所以解锁的过程要执行如下的Lua脚本 ,通过Lua脚本来保证判断和解锁具有原子性。
if redis.call("get", KEYS[1]) == ARGV[1] then
return redis.call("del", KEYS[1])
else
return 0
end
如果key对应的value一致,则删除这个key,通过这个方式释放锁是为了避免Client释放了其他Client申请的锁. 到此已经彻底说明了该如何实现一个分布式锁了,以及为什么要这样做的原因。 2.3 分布式锁工具类
public class LockUtil {
private static final String OK = "OK";
private static final Long LONG_ONE = 1L;
private static final String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
public static boolean tryLock(String key, String value, long expire) {
Jedis jedis = RedisPool.getJedis();
SetParams setParams = new SetParams();
setParams.nx().px(expire);
return OK.equals(jedis.set(key, value, setParams));
}
public static boolean releaseLock(String key, String value) {
Jedis jedis = RedisPool.getJedis();
return LONG_ONE.equals(jedis.eval(script, 1, key, value));
}
}
3 LUA脚本
从Redis2.6开始,内嵌Lua环境,通过EVAL和EVALSHA命令可以执行脚本
EVAL script numkeys key [key...] arg [arg...]
Redis执行Lua脚本的命令
EVAL命令:
EVAL "return {KEYS[1], KEYS[2], ARGV[1], ARGV[2]}" 2 key1 key2 first second
输出为:
1) "key1"
2) "key2"
3) "first"
4) "second"
EVALSHA命令:
EVAL命令是直接执行给定的脚本 EVALSHA命令可以根据给定的sha1校验码,执行缓存在服务器中的脚本 首先要将Lua脚本加载到Redis服务端,得到该脚本的SHA1校验和,EVALSHA命令使用SHA1作为参数可以直接执行对应Lua脚本,避免每次发送Lua脚本的开销。而脚本也会常驻在服务端,脚本功能得到了复用。
通过 SCRIPT LOAD 命令可以将脚本缓存到服务器,这个命令会返回脚本的sha1值
SCRIPT LOAD script
EVALSHA sha1 numkeys key [key ...] arg [arg ...]
sha1为脚本sha1值
在Lua脚本中调用Redis方法:
有2种方式redis.call()和redis.pcall()
redis.call()与redis.pcall()非常类似,唯一的区别是,如果Redis命令调用发生了错误,redis.call() 将抛出一个Lua类型的错误,再强制EVAL命令把错误返回给命令的调用者,而redis.pcall()将捕获错误并返回表示错误的Lua表类型
127.0.0.1:6379> set testKey testValue
OK
127.0.0.1:6379> get testKey
"testValue"
用脚本实现上述功能:
127.0.0.1:6379> eval "return redis.call('set', KEYS[1], ARGV[1])" 1 scriptKey scriptValue
OK
127.0.0.1:6379> eval "return redis.call('get', KEYS[1])" 1 scriptKey
"scriptValue"
用SCRIPT LOAD命令缓存脚本:
127.0.0.1:6379> SCRIPT LOAD "return redis.call('set', KEYS[1], ARGV[1])"
"55b22c0d0cedf3866879ce7c854970626dcef0c3"
127.0.0.1:6379> evalsha 55b22c0d0cedf3866879ce7c854970626dcef0c3 1 sha1Key sha1Value
OK
127.0.0.1:6379> get sha1Key
"sha1Value"
总结:
当有一些操作Redis的功能想和其他人共享,或者实现命令的原子性时,就可以考虑用lua脚本来实现