当前位置：首页 > article >正文

如何向zookeeper中注册内容

article 2025/2/25 18:05:52

我来为你展示如何在Java项目中使用Apache ZooKeeper注册内容。这里提供一个简单但完整的示例，包含依赖配置和代码实现。

首先需要在pom.xml中添加ZooKeeper依赖（假设使用Maven）：

<dependency>
    <groupId>org.apache.zookeeper</groupId>
    <artifactId>zookeeper</artifactId>
    <version>3.8.4</version>
</dependency>

下面是一个完整的Java示例代码，展示如何连接ZooKeeper并注册内容：

import org.apache.zookeeper.*;
import java.io.IOException;
import java.util.concurrent.CountDownLatch;

public class ZookeeperRegisterExample {
    private static final String ZK_CONNECT_STRING = "localhost:2181"; // ZooKeeper地址
    private static final int SESSION_TIMEOUT = 5000; // 会话超时时间(ms)
    private static ZooKeeper zooKeeper;
    private static final CountDownLatch connectedSignal = new CountDownLatch(1);

    public static void main(String[] args) {
        try {
            // 建立连接
            connectToZookeeper();
            
            // 注册内容
            registerContent("/my-service", "service-data".getBytes());
            
            // 保持程序运行以观察结果
            Thread.sleep(60000);
            
            // 关闭连接
            zooKeeper.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    // 连接ZooKeeper
    private static void connectToZookeeper() throws IOException {
        zooKeeper = new ZooKeeper(ZK_CONNECT_STRING, SESSION_TIMEOUT, new Watcher() {
            @Override
            public void process(WatchedEvent event) {
                if (event.getState() == Event.KeeperState.SyncConnected) {
                    connectedSignal.countDown();
                    System.out.println("成功连接到ZooKeeper");
                }
            }
        });
        
        try {
            connectedSignal.await(); // 等待连接建立
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    // 注册内容到指定路径
    private static void registerContent(String path, byte[] data) throws KeeperException, InterruptedException {
        // 检查节点是否存在
        if (zooKeeper.exists(path, false) == null) {
            // 创建持久节点
            zooKeeper.create(path, 
                           data, 
                           ZooDefs.Ids.OPEN_ACL_UNSAFE, 
                           CreateMode.PERSISTENT);
            System.out.println("成功创建节点: " + path);
        } else {
            // 如果节点已存在，更新数据
            zooKeeper.setData(path, data, -1);
            System.out.println("成功更新节点: " + path);
        }
        
        // 验证注册结果
        byte[] registeredData = zooKeeper.getData(path, false, null);
        System.out.println("节点数据: " + new String(registeredData));
    }
}

使用说明：

前提条件：
- 需要先安装并运行ZooKeeper服务（默认端口2181）
- 修改ZK_CONNECT_STRING为你的ZooKeeper实际地址
代码功能说明：
- 连接到ZooKeeper服务
- 在"/my-service"路径下注册内容"service-data"
- 如果节点已存在则更新数据，如果不存在则创建新节点
- 使用持久节点（PERSISTENT），即使客户端断开连接，节点依然保留
常用节点类型：
- CreateMode.PERSISTENT: 持久节点
- CreateMode.EPHEMERAL: 临时节点（客户端断开后自动删除）
- CreateMode.PERSISTENT_SEQUENTIAL: 持久顺序节点
- CreateMode.EPHEMERAL_SEQUENTIAL: 临时顺序节点
异常处理建议：

// 添加更多错误处理
try {
    registerContent("/my-service", "service-data".getBytes());
} catch (KeeperException.ConnectionLossException e) {
    System.out.println("连接丢失，正在重试...");
    // 实现重试逻辑
} catch (KeeperException.NodeExistsException e) {
    System.out.println("节点已存在");
}

实际应用场景示例：

// 服务注册示例
String servicePath = "/services/service1";
String serviceData = "{\"host\":\"192.168.1.100\",\"port\":8080}";
registerContent(servicePath, serviceData.getBytes());