ZooKeeper 分布式锁如何实现？

Zookeeper 通过临时有序节点和事件监听机制来实现分布式锁。具体实现步骤如下： #### 1. 创建临时有序节点 - **节点创建**： - 每个客户端在尝试获取锁时，会在锁的目录下创建一个临时有序节点。 - 例如，客户端1创建的节点名为`/lock/lock-000000001`，客户端2创建的节点名为`/lock/lock-000000002`，依此类推。 #### 2. 排队机制 - **节点排序**： - 所有客户端创建的节点按照节点名进行排序，节点名越小，表示其顺序越靠前。 - **判断是否获取锁**： - 每个客户端在创建完节点后，获取锁目录下的所有子节点，并判断自己创建的节点是否是最小的那个。 - 如果是最小的节点，说明该客户端获取到了锁。 - 如果不是最小的节点，说明锁被其他客户端持有。 #### 3. 设置监听器 - **监听前一个节点**： - 如果客户端未能获取锁，它需要对比自己前一个节点（比自己节点名小的最大节点）设置一个监听器。 - 监听器用于监听前一个节点的删除事件。 #### 4. 处理节点删除事件 - **释放锁**： - 持有锁的客户端在完成操作后，会删除自己创建的临时有序节点。 - **触发监听**： - 前一个节点被删除时，其监听器会触发通知下一个节点（紧随其后的节点）。 - 被通知的节点再次检查自己是否是最小的节点，如果是，则获取到锁。 #### 5. 自动释放锁 - **宕机处理**： - 如果某个客户端在持有锁期间宕机，由于节点是临时的，Zookeeper 会自动删除该客户端的临时节点，自动释放锁。 #### 关键点 1. **死锁问题**： - 使用 Zookeeper 的临时节点来避免死锁。客户端失效时，临时节点会自动删除，锁自动释放。 2. **锁等待问题**： - 使用有序节点来实现锁的排队。节点按顺序排列，保证锁的公平性。 3. **锁管理问题**： - 使用监听机制来管理锁的竞争。锁释放时，通过监听器通知下一个排队的节点。 4. **监听的羊群效应**： - 每个竞争者只监听自己前面的那个节点，避免过多的监听通知。 - 例如，如果2号节点释放锁，只有3号节点会收到通知，其他节点不会受到影响。 ### 分布式锁实现的示例代码 以下是一个使用 ZooKeeper 实现分布式锁的简单示例代码： ```java import org.apache.zookeeper.*; import org.apache.zookeeper.data.Stat; import java.util.Collections; import java.util.List; public class DistributedLock { private static final String LOCK_ROOT = "/locks"; private static final String LOCK_NODE = LOCK_ROOT + "/lock_"; private ZooKeeper zk; private String currentNode; public DistributedLock(ZooKeeper zk) { this.zk = zk; } public void acquireLock() throws Exception { // 创建临时有序节点 currentNode = zk.create(LOCK_NODE, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL); while (true) { // 获取锁节点下的所有子节点 List<String> nodes = zk.getChildren(LOCK_ROOT, false); // 排序 Collections.sort(nodes); // 获取最小的节点 String smallestNode = nodes.get(0); // 判断当前节点是否是最小节点 if (currentNode.endsWith(smallestNode)) { System.out.println("Acquired lock: " + currentNode); return; } // 获取前一个节点 String previousNode = null; for (String node : nodes) { if (currentNode.endsWith(node)) { break; } previousNode = node; } // 监听前一个节点的删除事件 final String watchNode = previousNode; final Object lock = new Object(); Stat stat = zk.exists(LOCK_ROOT + "/" + watchNode, event -> { if (event.getType() == Watcher.Event.EventType.NodeDeleted) { synchronized (lock) { lock.notify(); } } }); // 如果前一个节点不存在，重新获取锁 if (stat == null) { continue; } // 等待通知 synchronized (lock) { lock.wait(); } } } public void releaseLock() throws Exception { if (currentNode != null) { zk.delete(currentNode, -1); currentNode = null; System.out.println("Released lock: " + currentNode); } } } ``` ### 说明 - **创建临时有序节点**：客户端在`/locks`目录下创建一个临时有序节点。 - **排序和判断锁**：客户端获取所有子节点并排序，判断自己是否为最小节点，如果是，则获取锁；否则，监听前一个节点的删除事件。 - **释放锁**：客户端在完成操作后，删除自己的节点，触发下一个节点的监听器。 通过上述步骤和示例代码，Zookeeper 可以实现一个高效的分布式锁机制，确保多个客户端在分布式环境下安全、有序地访问共享资源。