Redis分布式锁的try-with-resources实现

public class RedisDistributedLock implements AutoCloseable {
    private final Jedis jedis;
    private final String lockKey;
    private final String lockValue;
    private final long expireTime;

    public RedisDistributedLock(Jedis jedis, String lockKey, long expireTime) {
        this.jedis = jedis;
        this.lockKey = lockKey;
        this.lockValue = UUID.randomUUID().toString();
        this.expireTime = expireTime;
    }

    public boolean tryLock() {
        String result = jedis.set(lockKey, lockValue, "NX", "PX", expireTime);
        return "OK".equals(result);
    }

    public void unlock() {
        if (lockValue.equals(jedis.get(lockKey))) {
            jedis.del(lockKey);
        }
    }

    @Override
    public void close() {
        unlock();
    }
}

try (RedisDistributedLock lock = new RedisDistributedLock(jedis, "myLock", 10000)) {
    if (lock.tryLock()) {
        // 执行业务逻辑
    } else {
        // 获取锁失败
    }
} // 锁会自动释放

public class RedisLockExample {
    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost", 6379);
        String lockKey = "myLock";
        long expireTime = 10000; // 10秒

        try (RedisDistributedLock lock = new RedisDistributedLock(jedis, lockKey, expireTime)) {
            if (lock.tryLock()) {
                System.out.println("Lock acquired, executing business logic...");
                Thread.sleep(3000); // 模拟业务逻辑执行
            } else {
                System.out.println("Failed to acquire lock");
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            jedis.close();
        }
    }
}