packageorg.zookeeper.demo;
importjava.io.IOException;
importjava.util.concurrent.CountDownLatch;
importorg.apache.zookeeper.CreateMode;
importorg.apache.zookeeper.KeeperException;
importorg.apache.zookeeper.WatchedEvent;
importorg.apache.zookeeper.Watcher;
importorg.apache.zookeeper.Watcher.Event.KeeperState;
importorg.apache.zookeeper.ZooDefs.Ids;
importorg.apache.zookeeper.ZooKeeper;
publicclassZookeeperClientimplementsWatcher{
//連接超時時間,10s
privatestaticfinalintSESSION_TIMEOUT= 10000;
//連接的zookeeperserver
privatestaticfinalStringCONNECTION_STRING = "172.31.25.8:2181";
privatestaticfinalStringZK_PATH = "/qyktest";
privateZooKeeperzk = null;
privateCountDownLatchconnectedSemaphore = newCountDownLatch(1);
publicvoidcreateConnection(StringconnectString, intsessionTimeout){
this.releaseConnection();
try{
zk= newZooKeeper(connectString,sessionTimeout, this);
connectedSemaphore.await();
}catch(InterruptedExceptione) {
System.out.println("連接創(chuàng)建失敗，發(fā)生InterruptedException");
e.printStackTrace();
}catch(IOExceptione) {
System.out.println("連接創(chuàng)建失敗，發(fā)生IOException");
e.printStackTrace();
}
}
publicvoidreleaseConnection(){
if(this.zk!= null){
try{
this.zk.close();
}catch(InterruptedExceptione) {
e.printStackTrace();
}
}
}
publicbooleancreatePath(Stringpath, String data) {
try{
Stringresult = this.zk.create(path,data.getBytes(), Ids.OPEN_ACL_UNSAFE,CreateMode.PERSISTENT);
System.out.println("節(jié)點創(chuàng)建成功,Path: "+result + ", content: "+data);
}catch(KeeperExceptione) {
System.out.println("節(jié)點創(chuàng)建失敗，發(fā)生KeeperException");
e.printStackTrace();
}catch(InterruptedExceptione) {
System.out.println("節(jié)點創(chuàng)建失敗，發(fā)生InterruptedException");
e.printStackTrace();
}
returntrue;
}
publicStringreadData(Stringpath) {
try{
System.out.println("獲取數據成功，path："+path);
returnnewString(this.zk.getData(path,false,null));
}catch(KeeperExceptione) {
System.out.println("讀取數據失敗，發(fā)生KeeperException，path:"+path);
e.printStackTrace();
return"";
}catch(InterruptedExceptione) {
System.out.println("讀取數據失敗，發(fā)生InterruptedException，path: "+path);
e.printStackTrace();
return"";
}
}
publicbooleanwriteData(Stringpath, String data) {
try{
System.out.println("更新數據成功，path："+path + ", stat: "+this.zk.setData(path,data.getBytes(), -1));
}catch(KeeperExceptione) {
System.out.println("更新數據失敗，發(fā)生KeeperException，path:"+path);
e.printStackTrace();
}catch(InterruptedExceptione) {
System.out.println("更新數據失敗，發(fā)生InterruptedException，path: "+path);
e.printStackTrace();
}
returnfalse;
}
publicvoiddeleteNode(Stringpath) {
try{
this.zk.delete(path,-1);
System.out.println("刪除節(jié)點成功，path："+path);
}catch(KeeperExceptione) {
System.out.println("刪除節(jié)點失敗，發(fā)生KeeperException，path:"+path);
e.printStackTrace();
}catch(InterruptedExceptione) {
System.out.println("刪除節(jié)點失敗，發(fā)生InterruptedException，path: "+path);
e.printStackTrace();
}
}
publicstaticvoidmain(String[]args) {
ZookeeperClientsample = newZookeeperClient();
//獲取連接
sample.createConnection(CONNECTION_STRING,SESSION_TIMEOUT);
//讀數據
Stringqyk = sample.readData("/qyktest");
System.out.println("qyk:"+qyk);
Stringurl = sample.readData("/qyk/db/url");
System.out.println("url"+url);
Stringdriver = sample.readData("/qyk/db/driver");
System.out.println("driver"+driver);
StringuserName = sample.readData("/qyk/db/userName");
System.out.println("userName"+userName);
Stringpassword = sample.readData("/qyk/db/password");
System.out.println("password"+password);
//創(chuàng)建節(jié)點
sample.createPath(ZK_PATH,"我是節(jié)點初始內容");
System.out.println("數據內容:"+sample.readData(ZK_PATH) + "\n");
//更新節(jié)點
sample.writeData(ZK_PATH,"更新后的數據");
System.out.println("數據內容:"+sample.readData(ZK_PATH) + "\n");
//刪除節(jié)點
sample.deleteNode(ZK_PATH);
//釋放連接
sample.releaseConnection();
}
@Override
publicvoidprocess(WatchedEventevent) {
System.out.println("收到事件通知："+event.getState() + "\n");
if(KeeperState.SyncConnected== event.getState()) {
connectedSemaphore.countDown();
}
}
}