秒懂Redis实现分布式锁的两种方案，小白必备指南

2023-06-24 06:23:48

分布式锁：协调共享资源的访问

在分布式系统中，多个进程或线程可能同时尝试访问共享资源。为了避免并发访问导致的数据不一致问题，分布式锁至关重要，它允许一次只有一个进程或线程访问共享资源。

Redis实现分布式锁

Redis提供了两种实现分布式锁的方案：

基于setnx命令

setnx命令仅当键不存在时才设置其值。利用这一特性，我们可以通过setnx命令实现分布式锁：

// key：锁定的资源名称
// value：客户端ID
// expireTime：锁的过期时间
String result = jedis.set(key, value, NX, EX, expireTime);

如果result为"OK"，则表示客户端成功获取锁。

基于Redisson框架

Redisson是一个Java分布式锁框架，它提供了一种更高级的分布式锁实现：

RLock lock = redissonClient.getLock(key);

// 尝试获取锁，等待waitTime，最多持有leaseTime
boolean acquired = lock.tryLock(waitTime, leaseTime, TimeUnit.MILLISECONDS);

如果acquired为true，则表示客户端获取了锁。

Java代码示例

基于setnx命令

import redis.clients.jedis.Jedis;

public class RedisDistributedLock {

    private Jedis jedis;

    public RedisDistributedLock(Jedis jedis) {
        this.jedis = jedis;
    }

    public boolean tryLock(String key, String value, long expireTime) {
        String result = jedis.set(key, value, NX, EX, expireTime);
        return "OK".equals(result);
    }

    public void unlock(String key, String value) {
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        jedis.eval(script, 1, key, value);
    }
}

基于Redisson框架

import org.redisson.api.RedissonClient;
import org.redisson.api.RLock;

public class RedissonDistributedLock {

    private RedissonClient redissonClient;

    public RedissonDistributedLock(RedissonClient redissonClient) {
        this.redissonClient = redissonClient;
    }

    public RLock lock(String key) {
        return redissonClient.getLock(key);
    }

    public boolean tryLock(String key, long waitTime, long leaseTime) {
        RLock lock = redissonClient.getLock(key);
        try {
            return lock.tryLock(waitTime, leaseTime, TimeUnit.MILLISECONDS);
        } catch (InterruptedException e) {
            return false;
        }
    }

    public void unlock(String key) {
        RLock lock = redissonClient.getLock(key);
        lock.unlock();
    }
}