Class CoolLock

src/main/java/com/cool/core/lock/CoolLock.java:18–108

import org.springframework.stereotype.Component;

@Component
@RequiredArgsConstructor
public class CoolLock {
    // 缓存类型
    @Value("${spring.cache.type}")
    private String type;

    @Value("${cool.cacheName}")
    private String cacheName;

    private final CacheManager cacheManager;
    private RedisCacheWriter redisCache ;

    // 非redis方式时使用
    private static final Map<String, Lock> lockMap = new ConcurrentHashMap<>();

    private static final String LOCK_PREFIX = "lock:";

    @PostConstruct
    private void init() {
        this.type = type.toLowerCase();
        if (type.equalsIgnoreCase(CacheType.REDIS.name())) {
            redisCache = (RedisCacheWriter) Objects.requireNonNull(cacheManager.getCache(cacheName))
                .getNativeCache();
        }
    }
    /**
     * 尝试获取锁
     *
     * @param key 锁的 key
     * @param expireTime 锁的过期时间
     * @return 如果成功获取锁则返回 true，否则返回 false
     */
    public boolean tryLock(String key, Duration expireTime) {
        String lockKey = getLockKey(key);
        if (type.equalsIgnoreCase(CacheType.CAFFEINE.name())) {
            Lock lock = lockMap.computeIfAbsent(lockKey, k -> new ReentrantLock());
            return lock.tryLock();
        }
        byte[] lockKeyBytes = lockKey.getBytes();
        // 使用 putIfAbsent 来尝试设置锁，如果成功返回 true，否则返回 false
        return redisCache.putIfAbsent(cacheName, lockKeyBytes, new byte[0], expireTime) == null;
    }

    /**
     * 释放锁
     */
    public void unlock(String key) {
        String lockKey = getLockKey(key);
        if (type.equalsIgnoreCase(CacheType.CAFFEINE.name())) {
            Lock lock = lockMap.get(lockKey);
            if (lock != null && lock.tryLock()) {
                lock.unlock();
                lockMap.remove(lockKey);
            }
            return;
        }
        redisCache.remove(cacheName, lockKey.getBytes());