package org.springblade.common.utils;
|
|
import org.springblade.core.redis.cache.BladeRedis;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.locks.LockSupport;
|
|
@Component
|
public class SpinLockUtil {
|
@Autowired
|
private BladeRedis bladeRedis;
|
|
private static final String LOCK_PREFIX = "thread-safe-lock:";
|
private static final Long DEFAULT_LOCK_TIMEOUT = 6L; // 默认锁超时时间，单位：秒
|
|
|
/**
|
* 尝试获取锁，如果获取失败则自旋等待。
|
*
|
* @param lockKey 锁的键
|
* @return 是否成功获取锁
|
*/
|
public boolean tryLock(String lockKey) {
|
String threadId = Thread.currentThread().getName() + Thread.currentThread().getId(); // 使用线程名和ID作为标识
|
long startTime = System.currentTimeMillis();
|
while (System.currentTimeMillis() - startTime < DEFAULT_LOCK_TIMEOUT * 1000) {
|
if (bladeRedis.getRedisTemplate().opsForValue().setIfAbsent(LOCK_PREFIX + lockKey, threadId,
|
DEFAULT_LOCK_TIMEOUT, TimeUnit.SECONDS)) {
|
return true;
|
}
|
// 短暂休眠，避免高并发下的CPU过度消耗
|
LockSupport.parkNanos(100_000_000L); // 使用LockSupport进行更细粒度的暂停，单位为纳秒
|
}
|
return false;
|
}
|
|
/**
|
* 释放锁，只有加锁的线程才能解锁
|
*
|
* @param lockKey 锁的键
|
* @return 是否成功释放锁
|
*/
|
public boolean unlock(String lockKey) {
|
String threadId = Thread.currentThread().getName() + Thread.currentThread().getId();
|
String currentValue = (String) bladeRedis.getRedisTemplate().opsForValue().get(LOCK_PREFIX + lockKey);
|
if (threadId.equals(currentValue)) {
|
bladeRedis.getRedisTemplate().delete(LOCK_PREFIX + lockKey);
|
return true;
|
}
|
return false; // 如果不是当前线程持有的锁，则返回false，不进行解锁操作
|
}
|
|
}
|
