如何在Java上使用zookeeper客户端库递归创建节点?

时间:2013-04-17 20:10:58

标签: java recursion hadoop apache-zookeeper

我知道这个问题已经被一个使用python的动物园管理员问过并回答了。答案很好,但是,我想要一些与代码更相关的东西。我已经实现了一个创建节点的方法,但我想以递归方式进行。我的节点的结构将是这样的:

ZOOKEEPER

  • WEB服务器
    • SERVER1
    • SERVER2
  • MODULE CONNECTED
    • 数据库模块
      • 电脑1
      • 电脑2
    • 服务模块
      • COMPUTER3
    • 搜索模块
      • COMPUTER4

我有类似的东西:

Zookeeper zk = new Zookeeper(...);

public void createNodeRecursively(String type) {
    final String node = "/" + type + "/" + info.getIP() + ":" + info.getPort(); // Correct line
    if (zk.exists("/" + type, null) == null) {
        Object ctx = new Object();
        StringCallback cb = new StringCallback() {
            public void processResult(int rc, String path,
                                Object ctx, String name) {
                if (name.equals("/" + type))// just in case
                    try {
                        zk.create(node, info.getBytes(),
                            Ids.OPEN_ACL_UNSAFE,CreateMode.EPHEMERAL);
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
            }
        };
        zk.create("/" + type, info.getBytes(), Ids.OPEN_ACL_UNSAFE,
                            CreateMode.PERSISTENT, cb, ctx);

    } else
        zk.create(node, info.getBytes(), Ids.OPEN_ACL_UNSAFE,
                            CreateMode.EPHEMERAL);
    } 
}

正如你所看到我多次使用zk.create,所以我想让方法递归以获得性能并拥有更好的代码,但我不知道如何开始,我会非常如果有人可以帮助我,我感激不尽。非常感谢你提前。

2 个答案:

答案 0 :(得分:0)

Zookeeper具有有用的属性:

  1. (写)请求的总顺序
  2. 其异步性。
  3. 你可以继续使用它。 简单地将整个树作为一组异步请求以正确的顺序发出,然后等待,直到所有这些请求成功执行。当然,您可以忽略“NodeExists”的异常(但由于这样的错误将写入日志这一事实并不好)。

答案 1 :(得分:0)

我设法取得了更好的表现:

public void createNode(NodePath nodePath, NodeData nodeData, NodeRights nodeRights, NodeCreationHandler nodeCreationHandler) throws KeeperException, InterruptedException, ZookeeperCreationException {

        if (zk == null) {
            throw new ZookeeperCreationException("The zookeeper client has not been instanced.");
        }       
        String targetPath = nodePath.getFullNodePath();
        targetPath = targetPath.substring(1, targetPath.length());
        byte[] serializedData = nodeData.serialize(new Object());
        String[] array = targetPath.split(ICoordinationConstants.BASE_ROOT_SPTR);
        String acum="";
        for (int i = 0; i < array.length-1; i++) {
            acum+=(ICoordinationConstants.BASE_ROOT_SPTR+array[i]);
            if (zk.exists(acum, null) == null) {
                zk.create(acum, serializedData, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
            }
        }
        zk.create(acum+ICoordinationConstants.BASE_ROOT_SPTR+array[array.length-1], serializedData, Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);             
    }