ipu-zk测试代码

.gitignore

@@ -12,3 +12,5 @@
# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
hs_err_pid*
.idea/
.project

ipu-zk-example/ipu-zk-example.iml

@@ -0,0 +1,55 @@
<?xml version="1.0" encoding="UTF-8"?>
<module org.jetbrains.idea.maven.project.MavenProjectsManager.isMavenModule="true" type="JAVA_MODULE" version="4">
  <component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_5">
    <output url="file://$MODULE_DIR$/target/classes" />
    <output-test url="file://$MODULE_DIR$/target/test-classes" />
    <content url="file://$MODULE_DIR$">
      <sourceFolder url="file://$MODULE_DIR$/src/main/java" isTestSource="false" />
      <sourceFolder url="file://$MODULE_DIR$/src/main/resources" type="java-resource" />
      <sourceFolder url="file://$MODULE_DIR$/src/test/java" isTestSource="true" />
      <excludeFolder url="file://$MODULE_DIR$/target" />
    </content>
    <orderEntry type="inheritedJdk" />
    <orderEntry type="sourceFolder" forTests="false" />
    <orderEntry type="library" name="Maven: junit:junit:4.12" level="project" />
    <orderEntry type="library" name="Maven: org.hamcrest:hamcrest-core:1.3" level="project" />
    <orderEntry type="library" name="Maven: com.ai.ipu:ipu-zk:3.1-SNAPSHOT" level="project" />
    <orderEntry type="library" name="Maven: org.apache.zookeeper:zookeeper:3.4.9" level="project" />
    <orderEntry type="library" name="Maven: org.slf4j:slf4j-api:1.6.1" level="project" />
    <orderEntry type="library" name="Maven: org.slf4j:slf4j-log4j12:1.6.1" level="project" />
    <orderEntry type="library" name="Maven: log4j:log4j:1.2.16" level="project" />
    <orderEntry type="library" name="Maven: jline:jline:0.9.94" level="project" />
    <orderEntry type="library" name="Maven: io.netty:netty:3.10.5.Final" level="project" />
    <orderEntry type="library" name="Maven: org.apache.curator:curator-framework:2.8.0" level="project" />
    <orderEntry type="library" name="Maven: org.apache.curator:curator-client:2.8.0" level="project" />
    <orderEntry type="library" name="Maven: org.apache.curator:curator-recipes:2.8.0" level="project" />
    <orderEntry type="library" name="Maven: com.101tec:zkclient:0.10" level="project" />
    <orderEntry type="library" name="Maven: com.google.guava:guava:16.0.1" level="project" />
    <orderEntry type="library" name="Maven: com.ai.ipu:ipu-basic:3.1-SNAPSHOT" level="project" />
    <orderEntry type="library" name="Maven: com.ai.ipu:ipu-common:3.1-SNAPSHOT" level="project" />
    <orderEntry type="library" name="Maven: com.mashape.unirest:unirest-java:1.3.8" level="project" />
    <orderEntry type="library" name="Maven: org.apache.httpcomponents:httpclient:4.3.2" level="project" />
    <orderEntry type="library" name="Maven: org.apache.httpcomponents:httpcore:4.3.1" level="project" />
    <orderEntry type="library" name="Maven: commons-logging:commons-logging:1.1.3" level="project" />
    <orderEntry type="library" name="Maven: org.apache.httpcomponents:httpasyncclient:4.0.1" level="project" />
    <orderEntry type="library" name="Maven: org.apache.httpcomponents:httpcore-nio:4.3.2" level="project" />
    <orderEntry type="library" name="Maven: org.apache.httpcomponents:httpmime:4.3.2" level="project" />
    <orderEntry type="library" name="Maven: org.json:json:20131018" level="project" />
    <orderEntry type="library" name="Maven: dom4j:dom4j:1.6.1" level="project" />
    <orderEntry type="library" name="Maven: xml-apis:xml-apis:1.0.b2" level="project" />
    <orderEntry type="library" name="Maven: jaxen:jaxen:1.1.1" level="project" />
    <orderEntry type="library" name="Maven: jdom:jdom:1.0" level="project" />
    <orderEntry type="library" name="Maven: xerces:xercesImpl:2.6.2" level="project" />
    <orderEntry type="library" name="Maven: xom:xom:1.0" level="project" />
    <orderEntry type="library" name="Maven: xerces:xmlParserAPIs:2.6.2" level="project" />
    <orderEntry type="library" name="Maven: xalan:xalan:2.6.0" level="project" />
    <orderEntry type="library" name="Maven: com.ibm.icu:icu4j:2.6.1" level="project" />
    <orderEntry type="library" name="Maven: org.jsoup:jsoup:1.7.2" level="project" />
    <orderEntry type="library" name="Maven: org.apache.commons:commons-lang3:3.4" level="project" />
    <orderEntry type="library" name="Maven: org.apache.poi:poi:3.9" level="project" />
    <orderEntry type="library" name="Maven: commons-codec:commons-codec:1.5" level="project" />
    <orderEntry type="library" name="Maven: com.alibaba:fastjson:1.2.38" level="project" />
    <orderEntry type="library" name="Maven: org.yaml:snakeyaml:1.23" level="project" />
    <orderEntry type="library" name="Maven: com.ai.wade:wade-data:1.0" level="project" />
  </component>
</module>

ipu-zk-example/src/main/java/com/ai/ipu/example/zookeeper/PushServerPathWatcher.java

@@ -0,0 +1,61 @@
package com.ai.ipu.example.zookeeper;
import org.apache.curator.framework.recipes.cache.PathChildrenCacheEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.ai.ipu.zk.listener.AbstractEventWatcher;
/**
 * 节点操作类
 * 对于PushMgmt <--> zookeeper集群而言，是 1 v 1 的连接
 *
 *
 * @author kevin
 *
 */
public class PushServerPathWatcher extends AbstractEventWatcher<PathChildrenCacheEvent>{
    private static final Logger LOG = LoggerFactory.getLogger(PushServerPathWatcher.class);
    /**
     * 节点添加，适用于TreeNodeCache、PathParentCache
     *
     * @param event 触发事件
     */
    @Override
    public void addListener(PathChildrenCacheEvent event) {
        if(event.getData() == null)
            return;
        LOG.debug("PushServerPathWatcher-add:path=["+event.getData().getPath()+"],value=["+new String(event.getData().getData())+"]");
    }
    /**
     * 节点删除，适用于PathParentCache、TreeNodeCache
     *
     * @param event 触发事件
     */
    @Override
    public void removeListener(PathChildrenCacheEvent event) {
        if(event.getData() == null)
            return;
        LOG.debug("PushServerPathWatcher-remove:path=["+event.getData().getPath()+"],value=["+new String(event.getData().getData())+"]");
    }
    /**
     * 节点修改，适用于PathParentCache、TreeNodeCache
     *
     * 只有当 Push-Server 的 http 端口修改时，才会触发相应操作；
     *
     * @param event 触发事件
     */
    @Override
    public void changeListener(PathChildrenCacheEvent event) {
        //空数据不予处理
        if(event.getData() == null)
            return;
        LOG.debug("PushServerPathWatcher-modify:path=["+event.getData().getPath()+"],value=["+new String(event.getData().getData())+"]");
    }
}

ipu-zk-example/src/main/java/com/ai/ipu/example/zookeeper/PushServerTreeWatcher.java

@@ -0,0 +1,84 @@
package com.ai.ipu.example.zookeeper;
import com.ai.ipu.zk.listener.AbstractEventWatcher;
import org.apache.curator.framework.recipes.cache.TreeCacheEvent;
import org.apache.zookeeper.data.Stat;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
 * 节点操作类
 * 对于PushMgmt <--> zookeeper集群而言，是 1 v 1 的连接
 *
 *
 * @author kevin
 *
 */
public class PushServerTreeWatcher extends AbstractEventWatcher<TreeCacheEvent>{
    private static final Logger LOG = LoggerFactory.getLogger(PushServerTreeWatcher.class);
    /**
     * 节点添加，适用于TreeNodeCache、PathParentCache
     *
     * @param event 触发事件
     */
    @Override
    public void addListener(TreeCacheEvent event) {
        //不需要关注的目录跳过
        if(skip(event)) return;
        LOG.debug("PushServerTreeWatcher-add:path=["+event.getData().getPath()+"],value=["+ new String(event.getData().getData())+"]");
    }
    /**
     * 节点删除，适用于PathParentCache、TreeNodeCache
     *
     * @param event 触发事件
     */
    @Override
    public void removeListener(TreeCacheEvent event) {
        //空数据不予处理
        if(event.getData() == null)
            return;
        if (event.getData().getData() == null) {
            LOG.debug("PushServerTreeWatcher-remove:path=["+event.getData().getPath()+"],value=[]");
        } else {
            LOG.debug("PushServerTreeWatcher-remove:path=["+event.getData().getPath()+"],value=["+ new String(event.getData().getData())+"]");
        }
    }
    /**
     * 节点修改，适用于PathParentCache、TreeNodeCache
     *
     * 只有当 Push-Server 的 http 端口修改时，才会触发相应操作；
     *
     * @param event 触发事件
     */
    @Override
    public void changeListener(TreeCacheEvent event) {
        //空数据不予处理
        if(event.getData() == null)
            return;
        LOG.debug("PushServerTreeWatcher-modify:path=["+event.getData().getPath()+"],value=["+ new String(event.getData().getData())+"]");
    }
    /**
     * 不需要要处理的事件
     *
     * @param event
     * @return
     * @throws
     */
    private boolean skip(TreeCacheEvent event) {
        //空数据不予处理
        if(event.getData() == null)
            return true;
        //只处理叶子节点（PushServer的值）
        Stat stat = event.getData().getStat();
        if(stat.getNumChildren() != 0)
            return true;
        return false;
    }
}

ipu-zk-example/src/main/java/com/ai/ipu/example/zookeeper/ZookeeperExample.java

@@ -0,0 +1,57 @@
package com.ai.ipu.example.zookeeper;
import com.ai.ipu.zk.ICuratorClient;
import com.ai.ipu.zk.ZkClientFactory;
import com.ai.ipu.zk.util.OperType;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class ZookeeperExample {
    private static final Logger LOG = LoggerFactory.getLogger(ZookeeperExample.class);
    private static final String PATH = "test";
    private static final String PATH2 = "test/test2";
    private static final String TYPE = "/myzktest";
    private static final boolean VERSIONCTL = true;
    private static final boolean PERSISTENT = true;
    private ICuratorClient client;
    @Before
    public void setBefore(){
        try {
            client = (ICuratorClient) ZkClientFactory.getZkClient("push");
            setUserDefine();
            client.addPathCacheListener(OperType.Define, PATH, true, new PushServerPathWatcher());
            client.addTreeCacheListener(OperType.Define, PATH, true, new PushServerTreeWatcher());
        } catch (Exception e) {
            LOG.error(e.getMessage(), e);
        }
    }
    private void setUserDefine() {
        OperType.Define.setType(TYPE);
        OperType.Define.setPersistent(PERSISTENT);
        OperType.Define.setVersionCtrl(VERSIONCTL);
    }
    @Test
    public void testSetValue() throws Exception{
        LOG.debug(Boolean.toString(client.setValue(OperType.Define, PATH, "myData1")));
    }
    @Test
    public void testSetValue2() throws Exception{
        LOG.debug(Boolean.toString(client.setValue(OperType.Define, PATH2, "myData1")));
    }
    @Test
    public void testGetValue() throws Exception {
        LOG.debug(client.getValue(OperType.Define, PATH2));
    }
    @Test
    public void deleteNode() throws Exception {
        LOG.debug(Boolean.toString(client.delete(OperType.Define, PATH2)));
    }
}

ipu-zk-example/src/main/resources/ipu-zk.xml

@@ -0,0 +1,21 @@
<?xml version = '1.0' encoding = 'UTF-8'?>
<zookeepers>
    <zookeeper name="push">
        <!-- 连接主机 -->
        <config name="connHosts" value="47.105.160.21:2101,47.105.160.21:2102,47.105.160.21:2103"/>
        <!-- 认证名和密码 -->
        <config name="auth" value="ipu:ipu321"/>
        <!-- 身份验证方式,world,auth,digest,ip -->
        <config name="acl" value="digest"/>
        <!-- 版本信息 -->
        <config name="version" value="3.0"/>
        <!-- 会话超时时间-->
        <config name="sessionTimeout" value="30000"/>
        <!-- 连接超时时间-->
        <config name="connectTimeout" value="3000"/>
        <!-- 重连超时时间-->
        <config name="reconnectTimeout" value="1000"/>
        <!-- 重连次数-->
        <config name="reconnectCount" value="3"/>
    </zookeeper>
</zookeepers>

ipu-zk-example/src/main/resources/log4j.properties

@@ -0,0 +1,8 @@
log4j.rootLogger=error, console
log4j.logger.org.apache.com.ai.ipu.zookeeper=error
log4j.logger.com.ai.ipu.example=debug
log4j.appender.console=org.apache.log4j.ConsoleAppender
log4j.appender.console.target=System.out
log4j.appender.console.layout=org.apache.log4j.PatternLayout
log4j.appender.console.layout.ConversionPattern=%d{yy/MM/dd HH:mm:ss} %5p [%t] (%F:%L) - %m%n