redis实现分布式可重入锁
/*** 保存每个线程独有的token
*/
private static ThreadLocal<String> tokenMap = new ThreadLocal<>();
/**
* redis实现分布式可重入锁,并不保证在过期时间内完成锁定内的任务，需根据业务逻辑合理分配seconds
*
* @param lock
* 锁的名称
* @param seconds
* 锁定时间，单位 秒
*token
* 对于同一个lock,相同的token可以再次获取该锁，不相同的token线程需等待到unlock之后才能获取
* @return
*/
public boolean lock(final String lock,finalint seconds) {
// @param token 对于同一个lock,相同的token可以再次获取该锁，不相同的token线程需等待到unlock之后才能获取
String token = tokenMap.get();
if (StringUtil.isBlank(token)) {
token = UUIDGenerator.getUUID();
tokenMap.set(token);
}
boolean flag = false;
Jedis client = null;
try {
client = jedisPool.getResource();
String ret = client.set(lock, token, "NX", "EX", seconds);
if (ret == null) {// 该lock的锁已经存在
String origToken = client.get(lock);// 即使lock已经过期也可以
if (token.equals(origToken)||origToken==null) {// token相同默认为同一线程，所以token应该尽量长且随机，保证不同线程的该值不相同
ret = client.set(lock, token, "NX", "EX", seconds);//
if ("OK".equalsIgnoreCase(ret))
flag = true;
}
} else if ("OK".equalsIgnoreCase(ret))
flag = true;
} catch (Exception e) {
logger.error(" lock{}失败");
throw new RedisException(e);
} finally {
if (client != null)
client.close();
}
return flag;
}
/**
* redis可以保证lua中的键的原子操作 unlock:lock调用完之后需unlock,否则需等待lock自动过期
*
* @param lock
*token
* 只有线程已经获取了该锁才能释放它（token相同表示已获取）
*/
public void unlock(final String lock) {
Jedis client = null;
final String token = tokenMap.get();
if (StringUtil.isBlank(token))
return;
try {
client = jedisPool.getResource();
final String script = "if redis.call(\"get\",\"" + lock + "\") == \"" + token + "\"thenreturn redis.call(\"del\",\"" + lock + "\") else return 0 end ";
client.eval(script);
} catch (Exception e) {
logger.error(" unlock{}失败");
throw new RedisException(e);
} finally {
if (client != null)
client.close();
}
}
public static void main(String[] args) {
>
final RedisUtil redis = ctx.getBean(RedisUtil.class);
for (int i = 0; i < 100; i++) {
new Thread(new Runnable() {
String key = "cheng";
@Override
public void run() {
boolean lock = redis.lock(key, 30);
System.out.print(lock + "-");
}
}).start();
;
}
// redis.unlock(key);
// ctx.close();
}
页:
[1]