基于Zookeeper 简单实现分布式任务协调组件

import org.apache.zookeeper.*;
 
public class DistributedTaskCoordinator {
 
    private ZooKeeper zooKeeper;
    private String taskPath;
 
    public DistributedTaskCoordinator(String host, int sessionTimeout, String taskPath) throws Exception {
        this.taskPath = taskPath;
        zooKeeper = new ZooKeeper(host, sessionTimeout, new Watcher() {
            @Override
            public void process(WatchedEvent event) {
                // 事件处理逻辑
            }
        });
        // 确保父目录存在
        if (zooKeeper.exists(taskPath.substring(0, taskPath.lastIndexOf('/')), false) == null) {
            zooKeeper.create(taskPath.substring(0, taskPath.lastIndexOf('/')), new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
        }
    }
 
    public void startTask() throws KeeperException, InterruptedException {
        // 创建临时节点表示开始任务
        zooKeeper.create(taskPath, "started".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
    }
 
    public void waitForTasks() throws KeeperException, InterruptedException {
        // 等待其他任务节点
        while (zooKeeper.exists(taskPath, event -> {}) == null) {
            // 处理其他任务节点的到来
        }
    }
 
    public void close() throws InterruptedException {
        zooKeeper.close();
    }
 
    public static void main(String[] args) {
        try {
            DistributedTaskCoordinator coordinator = new DistributedTaskCoordinator("localhost:2181", 30000, "/tasks/task-1");
            coordinator.startTask();
            coordinator.waitForTasks();
            // 执行任务逻辑
            coordinator.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

这个简易的示例展示了如何使用Zookeeper来协调分布式任务。它首先创建一个与Zookeeper的连接，并在指定的路径下创建一个临时节点来表示任务的开始。然后，它进入一个循环等待其他任务节点的出现，在这个过程中，它定义了一个事件处理器来响应Zookeeper的watch事件。当其他任务节点出现时，它们之间的协调完成，可以执行相关的任务逻辑。最后，任务完成后，它关闭与Zookeeper的连接。