package com.ruoyi.web.tool;
|
|
import org.springframework.data.redis.core.RedisTemplate;
|
|
import java.util.UUID;
|
import java.util.concurrent.TimeUnit;
|
|
/**
|
* @author zhibing.pu
|
* @Date 2024/9/26 18:43
|
*/
|
public class RedisLock {
|
private RedisTemplate redisTemplate;
|
private String lockKey;
|
private int expireTime; // 锁的超时时间
|
private int timeout; // 获取锁的超时时间
|
|
|
public RedisLock(RedisTemplate redisTemplate, String lockKey, int expireTime, int timeout) {
|
this.redisTemplate = redisTemplate;
|
this.lockKey = lockKey;
|
this.expireTime = expireTime;
|
this.timeout = timeout;
|
}
|
|
public boolean lock() {
|
String identifier = UUID.randomUUID().toString();
|
long end = System.currentTimeMillis() + timeout;
|
while (System.currentTimeMillis() < end) {
|
if (redisTemplate.opsForValue().setIfAbsent(lockKey, identifier, expireTime, TimeUnit.SECONDS)) {
|
return true;
|
}
|
// 可以使用延时来减少CPU占用
|
try {
|
Thread.sleep(100);
|
} catch (InterruptedException e) {
|
Thread.currentThread().interrupt();
|
}
|
}
|
return false;
|
}
|
|
public void unlock() {
|
redisTemplate.delete(lockKey);
|
}
|
}
|
