packagecom.utils;importjava.util.Collections;importjava.util.UUID;importredis.clients.jedis.Jedis;public classRedisTool {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 String SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";private static final Long RELEASE_SUCCESS = 1L;//test
private int count = 0;/*** 获取请求Id
*@return返回UUID*/
public staticString getRequestId(){returnUUID.randomUUID().toString();
}/*** 尝试获取分布式锁
*@paramjedis Redis客户端
*@paramlockKey 锁
*@paramrequestId 请求标识
*@paramexpireTime 超期时间
*@return是否获取成功*/
public static boolean tryGetDistributedLock(Jedis jedis, String lockKey, String requestId, intexpireTime) {
String result=jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);if(LOCK_SUCCESS.equals(result)) {return true;
}return false;
}/*** 释放分布式锁
*@paramjedis Redis客户端
*@paramlockKey 锁
*@paramrequestId 请求标识
*@return是否释放成功*/
public static booleanreleaseDistributedLock(Jedis jedis, String lockKey, String requestId) {
Object result=jedis.eval(SCRIPT, Collections.singletonList(lockKey), Collections.singletonList(requestId));if(RELEASE_SUCCESS.equals(result)) {return true;
}return false;
}public static voidmain(String[] args) {
RedisTool redisTool= newRedisTool();
redisTool.testDistributedLock();
}public voidtestDistributedLock() {for (int i = 0; i < 10; i++) {new Thread(newRunnable() {
@Overridepublic voidrun() {
Jedis jedis= new Jedis("localhost");
String lockKey= "lock";
String requestId=getRequestId();//加锁
if (tryGetDistributedLock(jedis, lockKey, requestId, 10)) {
count++;
System.out.println(requestId+ ":" +count);//解锁
releaseDistributedLock(jedis, lockKey, requestId);
}else{
System.out.println(requestId+ ": tryGetDistributedLock fail");
}
}
});
}
}
}