网上到处都是分布式锁的代码,基本都是通过setNX 和 expire

这两个不是原子操作,肯定会有问题,不乏好多人通过用setNX的value当做过期时间来弥补等等。但是好像都不太好,或者多少有点问题。

从一个大神那里得来的代码

 package com.abc.def.util;

 import redis.clients.jedis.Jedis;

 import java.util.Collections;

 public class RedisDistributedLock {

     private static final String LOCK_SUCCESS = "OK";
private static final String SET_IF_NOT_EXIST = "NX";
private static final String SET_WITH_EXPIRE_TIME = "PX";
private static final Long RELEASE_SUCCESS = 1L; /**
* 尝试获取分布式锁
* @param jedis Redis客户端
* @param lockKey 锁
* @param requestId 请求标识
* @param expireTime 超期时间
* @return 是否获取成功
*/
public static boolean tryGetDistributedLock(Jedis jedis, String lockKey, String requestId, int expireTime) { String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime); if (LOCK_SUCCESS.equals(result)) {
return true;
}
return false; } /**
* 释放分布式锁
* @param jedis Redis客户端
* @param lockKey 锁
* @param requestId 请求标识
* @return 是否释放成功
*/
public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) { String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId)); if (RELEASE_SUCCESS.equals(result)) {
return true;
}
return false; } }

获取锁,通过一个条指令来获取并且同时设置超时。

另外,解锁是通过获取锁的时候设置的key  这个key应该是一个随机值 推荐使用 UUID 来生成。 这样只能解锁自己的锁

另外解锁操作用的lua脚本来执行,把三条语句集合成一个原子操作。

使用示例代码:

 while(true){
  String uuid = UUID.randomUUID().toString();
boolean ret = lock.tryGetDistributedLock(redis,"NX_TEST", uuid, 5000);
if(ret != false){
//TODO 实现业务逻辑
lock.releaseDistributedLock(redis,"NX_TEST", uuid);
break;
}else{
Thread.sleep(500);
}
12 }

这里,设置的5000(5秒)超时时间,这个时间一定要大于业务逻辑的执行时间,否则就没办法锁住了。。

参考博客 https://wudashan.cn/2017/10/23/Redis-Distributed-Lock-Implement/

PS: 哪位大神知道怎么从RedisTemplate 中获取 jedis ,RedisTemplate 并没有封装set 的相关函数。

05-11 22:51