redis分布式锁的代码实现(加入看门狗)

最新推荐文章于 2024-06-19 18:10:55 发布

pad长世界第一

最新推荐文章于 2024-06-19 18:10:55 发布

阅读量215

点赞数 1

文章标签： redis 分布式数据库

本文链接：https://blog.csdn.net/weixin_44045520/article/details/136989114

版权

本文介绍了一个使用Jedis库在Java中实现的分布式锁类RedisLock，包括acquireLock获取锁、releaseLock释放锁以及watchdog机制确保锁的持续有效。实例展示了如何在主函数中使用这个锁进行资源保护。

摘要由CSDN通过智能技术生成

import redis.clients.jedis.Jedis;
import redis.clients.jedis.Transaction;

import java.util.UUID;

public class RedisLock {
private Jedis jedis;
private String lockKey;
private String lockValue;
private int expireTime;
private ThreadLocal lockThreadLocal;
private Thread watchdogThread;

public RedisLock(Jedis jedis, String lockKey, int expireTime) {
    this.jedis = jedis;
    this.lockKey = lockKey;
    this.expireTime = expireTime;
    this.lockThreadLocal = new ThreadLocal<>();
}

public boolean acquireLock() {
    String value = UUID.randomUUID().toString();
    String result = jedis.set(lockKey, value, "NX", "EX", expireTime);
    if ("OK".equals(result)) {
        lockValue = value;
        lockThreadLocal.set(value);
        startWatchdog();
        return true;
    }
    return false;
}

public boolean releaseLock() {
    String value = lockThreadLocal.get();
    if (value != null) {
        Transaction transaction = jedis.multi();
        transaction.watch(lockKey);
        if (value.equals(jedis.get(lockKey))) {
            transaction.del(lockKey);
            transaction.exec();
            lockThreadLocal.remove();
            stopWatchdog();
            return true;
        } else {
            transaction.unwatch();
        }
    }
    return false;
}

private void renewLock() {
    while (lockThreadLocal.get() != null) {
        // 续约锁的有效期
        jedis.expire(lockKey, expireTime);
        try {
            Thread.sleep(expireTime * 1000 / 2); // 续约时间为锁有效期的一半
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

private void startWatchdog() {
    watchdogThread = new Thread(this::renewLock);
    watchdogThread.setDaemon(true);
    watchdogThread.start();
}

private void stopWatchdog() {
    if (watchdogThread != null) {
        watchdogThread.interrupt();
    }
}

}
使用示例：

java
import redis.clients.jedis.Jedis;

public class Main {
public static void main(String[] args) {
Jedis jedis = new Jedis(“localhost”, 6379);
RedisLock lock = new RedisLock(jedis, “my_resource”, 10);
if (lock.acquireLock()) {
try {
// 执行需要加锁的代码
} finally {
lock.releaseLock();
}
}
}
}