【Redis】Redis实现分布式锁合理的控制锁的有效时长的方法

灰_灰丶灰

于 2024-06-09 10:00:00 发布

阅读量1.4k

点赞数 14

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

本文链接：https://blog.csdn.net/hui_zai_/article/details/139512694

版权

在分布式系统中，合理地控制 Redis 分布式锁的有效时长（即过期时间）非常重要，以确保锁既能防止死锁又能提供高可用性。设置合理的过期时间可以防止客户端在持有锁期间崩溃而导致其他客户端无法获取锁的情况，同时也能确保锁在操作未完成时不会过早失效。

以下是实现合理控制 Redis 分布式锁有效时长的方法和注意事项：

方法一：估算操作时间并加上缓冲时间

最简单的方法是根据操作的预期时间设置锁的有效时长。通常会估算操作需要的时间，并加上一些缓冲时间，以应对网络延迟和其他不确定因素。

示例代码：

public class RedisDistributedLock {
    private Jedis jedis;
    private String lockKey;
    private int expireTime; // 锁的过期时间（秒）

    public RedisDistributedLock(Jedis jedis, String lockKey, int expireTime) {
        this.jedis = jedis;
        this.lockKey = lockKey;
        this.expireTime = expireTime;
    }

    public boolean acquireLock(String requestId) {
        String result = jedis.set(lockKey, requestId, "NX", "EX", expireTime);
        return "OK".equals(result);
    }

    public boolean releaseLock(String requestId) {
        String script =
                "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                "return redis.call('del', KEYS[1]) " +
                "else " +
                "return 0 " +
                "end";
        Object result = jedis.eval(script, 1, lockKey, requestId);
        return "1".equals(result.toString());
    }
}

在上述代码中，expireTime 应根据操作的预期时间加上缓冲时间来设置。

方法二：锁续命（Lock Renewal）

对于一些可能超过预期时间的长时间操作，可以使用锁续命机制。在锁快要过期时，客户端可以通过续命机制延长锁的有效时间。

实现思路：

在锁快要过期时，客户端定期检查锁是否仍然持有，并尝试续命。
续命可以通过重新设置过期时间实现。

示例代码：

public class RedisDistributedLock {
    private Jedis jedis;
    private String lockKey;
    private int expireTime; // 锁的过期时间（秒）
    private volatile boolean keepExtending = true;

    public RedisDistributedLock(Jedis jedis, String lockKey, int expireTime) {
        this.jedis = jedis;
        this.lockKey = lockKey;
        this.expireTime = expireTime;
    }

    public boolean acquireLock(String requestId) {
        String result = jedis.set(lockKey, requestId, "NX", "EX", expireTime);
        if ("OK".equals(result)) {
            startRenewalTask(requestId);
            return true;
        }
        return false;
    }

    public boolean releaseLock(String requestId) {
        keepExtending = false;
        String script =
                "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                "return redis.call('del', KEYS[1]) " +
                "else " +
                "return 0 " +
                "end";
        Object result = jedis.eval(script, 1, lockKey, requestId);
        return "1".equals(result.toString());
    }

    private void startRenewalTask(String requestId) {
        Thread renewalThread = new Thread(() -> {
            while (keepExtending) {
                try {
                    Thread.sleep(expireTime * 1000 / 2);
                    String script =
                            "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                            "return redis.call('expire', KEYS[1], ARGV[2]) " +
                            "else " +
                            "return 0 " +
                            "end";
                    jedis.eval(script, 1, lockKey, requestId, String.valueOf(expireTime));
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }
        });
        renewalThread.setDaemon(true);
        renewalThread.start();
    }
}