redis分布式锁setnx

package com.realize;

import redis.clients.jedis.Jedis;
import java.util.UUID;
import redis.clients.jedis.params.SetParams;

public class DistributedLock {

    private Jedis jedis;
    private String lockKey;
    private String lockValue;
    private int expireTime;

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

    // 获取锁
    public boolean acquireLock() {
        this.lockValue = UUID.randomUUID().toString();
        //String result = jedis.set(lockKey, lockValue, "NX", "EX", expireTime);
        String result = jedis.set(lockKey, lockValue,SetParams.setParams().nx().ex(10L));
        return "OK".equals(result);
    }

    // 释放锁
    public boolean releaseLock() {
        String luaScript =
                "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                        "return redis.call('del', KEYS[1]) " +
                        "else " +
                        "return 0 " +
                        "end";
        Object result = jedis.eval(luaScript, 1, lockKey, lockValue);
        return result.equals(1L);
    }

    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost", 6379);
        DistributedLock lock = new DistributedLock(jedis, "resource_lock", 10);

        if (lock.acquireLock()) {
            try {
                System.out.println("Lock acquired, doing some work...");
                // 执行业务逻辑
            } finally {
                lock.releaseLock();
                System.out.println("Lock released");
            }
        } else {
            System.out.println("Failed to acquire lock");
        }

        jedis.close();
    }
}

  • 注意事项
  • 唯一标识符:使用唯一标识符来标识每个持有锁的客户端,防止锁被错误释放。
  • 过期时间:设置合理的过期时间,防止死锁。如果客户端在持有锁期间崩溃,过期时间可以自动释放锁。
  • 重试机制:获取锁时可以设置重试机制,防止短暂的争用导致获取锁失败。
  • 使用场景

    SET NX 命令通常用于实现分布式锁,以确保在分布式环境中,同一时间只有一个客户端可以持有锁,从而避免并发访问导致的数据不一致问题。

  • 获取锁:

    lockValue:生成一个唯一标识符,作为锁的值。
    jedis.set(lockKey, lockValue, "NX", "EX", expireTime):尝试使用 SET NX 命令设置锁。只有在 lockKey 不存在时,才能成功设置该键,并指定过期时间 expireTime。

你可能感兴趣的:(redis,分布式,数据库)