怎样写一个分布式锁，用Java

缺乏、安全感 2024-03-25 15:56 40阅读 0赞

可以使用 Java 的 `java.util.concurrent.locks.Lock` 接口来实现分布式锁。

你可以使用任意一种实现了 `Lock` 接口的锁，比如 `ReentrantLock` 或者 `ReentrantReadWriteLock`。这些锁都可以用来保证多线程之间的同步访问。

如果你想要实现一个分布式锁，你需要使用一个分布式存储来保存锁的状态，比如 Redis 或者 Zookeeper。

下面是一个简单的分布式锁的例子，使用了 Redis 来保存锁的状态：

import redis.clients.jedis.Jedis;
    
    public class DistributedLock {
      private static final String LOCK_SUCCESS = "OK";
      private static final String SET_IF_NOT_EXIST = "NX";
      private static final String SET_WITH_EXPIRE_TIME = "PX";
      private static final Long RELEASE_SUCCESS = 1L;
    
      /**
       * 尝试获取分布式锁
       * @param jedis Redis客户端
       * @param lockKey 锁
       * @param requestId 请求标识
       * @param expireTime 超期时间
       * @return 是否获取成功
       */
      public static boolean tryGetDistributedLock(Jedis jedis, String lockKey, String requestId, int expireTime) {
    
        String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
    
        if (LOCK_SUCCESS.equals(result)) {
          return true;
        }
        return false;
      }
    
      /**
       * 释放分布式锁
       * @param jedis Redis客户端
       * @param lockKey 锁
       * @param requestId 请求标识
       * @return 是否释放成功
       */
      public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) {
    
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";