一. 基于Redis的分布式锁

1.名词解释

Redis 是一种内存数据库,其特点之一就是支持原子性操作,这使得它成为实现分布式锁的理想选择。基于 Redis 实现的分布式锁主要依赖于 Redis 的 SETNX(SET if Not eXists)命令,该命令用于设置键的值,但仅在键不存在时设置成功。锁的释放则可以通过 DEL 命令或设置过期时间来实现。

实现步骤:
  1. 客户端请求获取锁时,调用 SETNX 命令尝试将某个唯一标识符作为键名存入 Redis 中,如果 SETNX 执行成功,则获取到锁。
  2. 如果 SETNX 执行失败,则表示锁已被其他客户端占用,客户端可以选择重试或放弃。
  3. 获取到锁的客户端在完成操作后,通过 DEL 命令删除键来释放锁,或者可以为键设置过期时间,确保即使客户端崩溃或网络中断,锁也会自动释放。
优点:
  • 实现简单,易于理解和部署。
  • Redis 的高性能和可靠性保证了分布式锁的性能和稳定性。
缺点:
  • 锁的获取和释放不是原子操作,可能会导致死锁或锁过期等问题。
  • 锁的粒度较粗,无法支持细粒度的锁。

2.代码实现 

下面是一个基于 Redis 的分布式锁的 Java 代码示例:

import redis.clients.jedis.Jedis;

public class RedisDistributedLock {
    private final Jedis jedis;
    private final String lockKey;
    private final int expireTimeMs;
    private boolean locked = false;

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

    public boolean acquireLock() {
        // 使用 SETNX 命令尝试获取锁
        String result = jedis.set(lockKey, "1", "NX", "PX", expireTimeMs);
        if ("OK".equals(result)) {
            locked = true;
            return true;
        }
        return false;
    }

    public void releaseLock() {
        // 释放锁
        if (locked) {
            jedis.del(lockKey);
            locked = false;
        }
    }

    // 示例用法
    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost");
        RedisDistributedLock lock = new RedisDistributedLock(jedis, "mylock", 10000); // 锁10秒钟

        try {
            if (lock.acquireLock()) {
                System.out.println("获取锁成功,执行业务逻辑...");
                // 执行业务逻辑
            } else {
                System.out.println("获取锁失败,其他线程正在操作...");
            }
        } finally {
            lock.releaseLock(); // 释放锁
        }

        jedis.close();
    }
}

这个示例代码创建了一个 `RedisDistributedLock` 类,它封装了 Redis 的连接和操作方法,并提供了获取锁和释放锁的功能。在 `main` 方法中,首先创建了一个 `RedisDistributedLock` 实例,并尝试获取锁。如果获取锁成功,则执行业务逻辑;否则输出获取锁失败的消息。最后,无论是否获取到锁,都会释放锁。

需要注意的是,该示例中使用的是 Redis 的单机模式,如果要在生产环境中使用 Redis 集群,需要使用相应的连接池和集群模式进行连接。

二. 基于ZooKeeper的分布式锁

1.名称解释

ZooKeeper 是一个分布式协调服务,提供了一套完善的分布式原语,可以用来实现分布式锁。基于 ZooKeeper 实现的分布式锁主要依赖于其顺序节点和监听机制。

实现步骤:
  1. 客户端请求获取锁时,创建一个顺序临时节点。
  2. 客户端获取父节点下所有的子节点,并监视比自己小的那个节点。
  3. 如果发现自己是最小的节点,则表示获取到了锁,可以执行业务逻辑。
  4. 如果发现自己不是最小的节点,则等待监听的节点被删除,然后重复步骤 2。
优点:
  • 锁的获取和释放是原子操作,能够确保强一致性。
  • 支持细粒度的锁。
缺点:
  • 实现相对复杂,对 ZooKeeper 的依赖性较强。
  • 性能可能不如基于 Redis 的实现方式。

2.代码实现  

下面是一个基于 ZooKeeper 的分布式锁的 Java 代码示例:

import org.apache.zookeeper.*;

import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.CountDownLatch;

public class ZooKeeperDistributedLock implements Watcher {
    private ZooKeeper zooKeeper;
    private String rootNode = "/locks";
    private String lockName;
    private String lockPath;
    private CountDownLatch countDownLatch;

    public ZooKeeperDistributedLock(String host, int sessionTimeout, String lockName) throws IOException {
        this.lockName = lockName;
        countDownLatch = new CountDownLatch(1);
        zooKeeper = new ZooKeeper(host, sessionTimeout, this);
        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void lock() {
        try {
            String lockNode = zooKeeper.create(rootNode + "/" + lockName, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
            List<String> nodes = zooKeeper.getChildren(rootNode, false);
            Collections.sort(nodes);
            if (lockNode.equals(rootNode + "/" + nodes.get(0))) {
                // 如果创建的节点是最小的节点,则表示获取到了锁
                lockPath = lockNode;
                return;
            }
            String currentNode = lockNode.substring(lockNode.lastIndexOf("/") + 1);
            // 获取比当前节点小的最大节点,并设置监听
            String preNode = nodes.get(Collections.binarySearch(nodes, currentNode) - 1);
            zooKeeper.exists(rootNode + "/" + preNode, new LockWatcher());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void unlock() {
        try {
            zooKeeper.delete(lockPath, -1);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void process(WatchedEvent event) {
        if (event.getState() == Event.KeeperState.SyncConnected) {
            countDownLatch.countDown();
        }
    }

    private class LockWatcher implements Watcher {
        @Override
        public void process(WatchedEvent event) {
            if (event.getType() == Event.EventType.NodeDeleted) {
                lock();
            }
        }
    }

    // 示例用法
    public static void main(String[] args) {
        String host = "localhost:2181";
        int sessionTimeout = 5000;
        String lockName = "mylock";

        try {
            ZooKeeperDistributedLock lock = new ZooKeeperDistributedLock(host, sessionTimeout, lockName);

            lock.lock();
            System.out.println("获取锁成功,执行业务逻辑...");
            // 执行业务逻辑

            lock.unlock();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

 这个示例代码创建了一个 ZooKeeperDistributedLock 类,它封装了 ZooKeeper 的连接和操作方法,并提供了获取锁和释放锁的功能。在 lock 方法中,创建了一个临时顺序节点,并获取了父节点下的所有子节点,并对这些子节点进行排序。如果当前创建的节点是最小的节点,则表示获取到了锁;否则,设置对比当前节点小的最大节点的监听器,当该节点被删除时重新尝试获取锁。

需要注意的是,该示例中创建的节点是临时顺序节点,当客户端与 ZooKeeper 断开连接时,该节点会自动删除,从而释放锁。

Logo

更多推荐