使用 Redis 的 SETNX 命令实现分布式锁

使用 Redis 实现分布式锁，具体实现思想是使用 Redis 的 SETNX 命令尝试获取锁，如果 SETNX 返回 1，则表示成功获取锁；如果 SETNX 返回 0，则表示锁已经存在，需要判断锁是否已经过期，如果锁已经过期，则使用 GETSET 命令尝试获取锁。如果 GETSET 返回的值等于之前获取的值，则表示成功获取锁；如果 GETSET 返回的值不等于之前获取的值，则表示锁已经被其他进程获取。释放锁时，使用 DEL 命令删除锁的键。

import redis.clients.jedis.Jedis;

public class DistributedLock {

	// 锁的键名
    private static final String LOCK_KEY = "mylock";
    // 锁的过期时间，单位毫秒
    private static final int LOCK_EXPIRE = 30000;
	// Redis 客户端
    private Jedis jedis;

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

    /**
     * 获取分布式锁
     *
     * @return 是否成功获取锁
     */
    public boolean acquire() {
        long now = System.currentTimeMillis();
        // 锁的过期时间
        long expireTime = now + LOCK_EXPIRE + 1;
        String expireStr = String.valueOf(expireTime);
        if (jedis.setnx(LOCK_KEY, expireStr) == 1) {
        	// 成功获取锁
            return true;
        }
        String currentValueStr = jedis.get(LOCK_KEY);
        	// 锁已经过期
        if (currentValueStr != null && Long.parseLong(currentValueStr) < now) {
            String oldValueStr = jedis.getSet(LOCK_KEY, expireStr);
            // 判断是否有其他进程已经获取了锁
            if (oldValueStr != null && oldValueStr.equals(currentValueStr)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 释放分布式锁
     */
    public void release() {
        String currentValueStr = jedis.get(LOCK_KEY);
        if (currentValueStr != null) {
            jedis.del(LOCK_KEY);
        }
    }

    // 使用示例
    public static void main(String[] args) {
    	// 连接本地 Redis 服务
        Jedis jedis = new Jedis("localhost");
        DistributedLock lock = new DistributedLock(jedis);
        // 获取锁成功
        if (lock.acquire()) {
            try {
                // TODO: 执行需要加锁的业务逻辑
            } finally {
            	// 释放锁
                lock.release();
            }
        } else { // 获取锁失败
            System.out.println("获取锁失败");
        }
    }

}