|
|
|
|
@ -0,0 +1,162 @@
|
|
|
|
|
package com.wayn.data.redis.manager;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.util.IdUtil;
|
|
|
|
|
import com.wayn.data.redis.util.ThreadUtil;
|
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
|
import org.springframework.data.redis.core.RedisTemplate;
|
|
|
|
|
import org.springframework.data.redis.core.script.DefaultRedisScript;
|
|
|
|
|
import org.springframework.data.redis.core.script.RedisScript;
|
|
|
|
|
import org.springframework.stereotype.Component;
|
|
|
|
|
|
|
|
|
|
import java.util.Collections;
|
|
|
|
|
import java.util.concurrent.ExecutorService;
|
|
|
|
|
import java.util.concurrent.Executors;
|
|
|
|
|
import java.util.concurrent.ScheduledExecutorService;
|
|
|
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* redis锁
|
|
|
|
|
*/
|
|
|
|
|
@Slf4j
|
|
|
|
|
@Component
|
|
|
|
|
public class RedisLock {
|
|
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
|
public RedisTemplate redisTemplate;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 默认锁过期时间30秒
|
|
|
|
|
*/
|
|
|
|
|
public static final Integer DEFAULT_TIME_OUT = 30;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 保存线程id-ThreadLocal
|
|
|
|
|
*/
|
|
|
|
|
private ThreadLocal<String> stringThreadLocal = new ThreadLocal<>();
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 保存定时任务（watch-dog）-ThreadLocal
|
|
|
|
|
*/
|
|
|
|
|
private ThreadLocal<ExecutorService> executorServiceThreadLocal = new ThreadLocal<>();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 加锁，不指定过期时间
|
|
|
|
|
*
|
|
|
|
|
* @param key key名称
|
|
|
|
|
* @return boolean
|
|
|
|
|
*/
|
|
|
|
|
public boolean lock(String key) {
|
|
|
|
|
return lock(key, null);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 加锁
|
|
|
|
|
*
|
|
|
|
|
* @param key key名称
|
|
|
|
|
* @param timeout 过期时间/分钟
|
|
|
|
|
* @return boolean
|
|
|
|
|
*/
|
|
|
|
|
public boolean lock(String key, Integer timeout) {
|
|
|
|
|
Integer timeoutTmp;
|
|
|
|
|
if (timeout == null) {
|
|
|
|
|
timeoutTmp = DEFAULT_TIME_OUT;
|
|
|
|
|
} else {
|
|
|
|
|
timeoutTmp = timeout;
|
|
|
|
|
}
|
|
|
|
|
String nanoId;
|
|
|
|
|
if (stringThreadLocal.get() != null) {
|
|
|
|
|
nanoId = stringThreadLocal.get();
|
|
|
|
|
} else {
|
|
|
|
|
nanoId = IdUtil.nanoId();
|
|
|
|
|
stringThreadLocal.set(nanoId);
|
|
|
|
|
}
|
|
|
|
|
RedisScript<Long> redisScript = new DefaultRedisScript<>(buildLuaLockScript(), Long.class);
|
|
|
|
|
Long execute = (Long) redisTemplate.execute(redisScript, Collections.singletonList(key), nanoId, timeoutTmp);
|
|
|
|
|
boolean flag = execute != null && execute == 1;
|
|
|
|
|
if (flag) {
|
|
|
|
|
ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
|
|
|
|
|
executorServiceThreadLocal.set(scheduledExecutorService);
|
|
|
|
|
scheduledExecutorService.scheduleWithFixedDelay(() -> {
|
|
|
|
|
RedisScript<Long> renewRedisScript = new DefaultRedisScript<>(buildLuaRenewScript(), Long.class);
|
|
|
|
|
Long result = (Long) redisTemplate.execute(renewRedisScript, Collections.singletonList(key), nanoId, timeoutTmp);
|
|
|
|
|
if (result != null && result == 2) {
|
|
|
|
|
ThreadUtil.shutdownAndAwaitTermination(scheduledExecutorService);
|
|
|
|
|
}
|
|
|
|
|
}, 0, timeoutTmp / 3, TimeUnit.SECONDS);
|
|
|
|
|
}
|
|
|
|
|
return flag;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 释放锁
|
|
|
|
|
*
|
|
|
|
|
* @param key key名称
|
|
|
|
|
* @return boolean
|
|
|
|
|
*/
|
|
|
|
|
public boolean unLock(final String key) {
|
|
|
|
|
String nanoId = stringThreadLocal.get();
|
|
|
|
|
RedisScript<Long> redisScript = new DefaultRedisScript<>(buildLuaUnLockScript(), Long.class);
|
|
|
|
|
Long execute = (Long) redisTemplate.execute(redisScript, Collections.singletonList(key), nanoId);
|
|
|
|
|
boolean flag = execute != null && execute == 1;
|
|
|
|
|
if (flag) {
|
|
|
|
|
if (executorServiceThreadLocal.get() != null) {
|
|
|
|
|
ThreadUtil.shutdownAndAwaitTermination(executorServiceThreadLocal.get());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return flag;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private String buildLuaLockScript() {
|
|
|
|
|
return """
|
|
|
|
|
local key = KEYS[1]
|
|
|
|
|
local value = ARGV[1]
|
|
|
|
|
local time_out = ARGV[2]
|
|
|
|
|
local result = redis.call('get', key)
|
|
|
|
|
if result == value then
|
|
|
|
|
return 1;
|
|
|
|
|
end
|
|
|
|
|
local lock_result = redis.call('setnx', key, value)
|
|
|
|
|
if tonumber(lock_result) == 1 then
|
|
|
|
|
redis.call('expire', key, time_out)
|
|
|
|
|
return 1;
|
|
|
|
|
else
|
|
|
|
|
return 0;
|
|
|
|
|
end
|
|
|
|
|
""";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private String buildLuaUnLockScript() {
|
|
|
|
|
return """
|
|
|
|
|
local key = KEYS[1]
|
|
|
|
|
local value = ARGV[1]
|
|
|
|
|
local result = redis.call('get', key)
|
|
|
|
|
if result ~= value then
|
|
|
|
|
return 0;
|
|
|
|
|
else
|
|
|
|
|
redis.call('del', key)
|
|
|
|
|
end
|
|
|
|
|
return 1;
|
|
|
|
|
""";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private String buildLuaRenewScript() {
|
|
|
|
|
return """
|
|
|
|
|
local key = KEYS[1]
|
|
|
|
|
local value = ARGV[1]
|
|
|
|
|
local timeout = ARGV[2]
|
|
|
|
|
local result = redis.call('get', key)
|
|
|
|
|
if result ~= value then
|
|
|
|
|
return 2;
|
|
|
|
|
end
|
|
|
|
|
local ttl = redis.call('ttl', key)
|
|
|
|
|
if tonumber(ttl) < tonumber(timeout) / 2 then
|
|
|
|
|
redis.call('expire', key, timeout)
|
|
|
|
|
return 1;
|
|
|
|
|
else
|
|
|
|
|
return 0;
|
|
|
|
|
end
|
|
|
|
|
""";
|
|
|
|
|
}
|
|
|
|
|
}
|
wayn
1 year ago