package com.zook.util;

import org.apache.zookeeper.*;
import org.apache.zookeeper.data.Stat;

import java.io.IOException;

public class ZookeeperWatcher implements Watcher {

    //连接地址
    private static final String CONNECT_ADDDR="0.0.0.0:2181";

    //session超时时间
    private static final int SESSION_OUTTIMF=2000;

    //创建zk方法
    private ZooKeeper zk;

    private ZookeeperWatcher(){
        openConnection(CONNECT_ADDDR,SESSION_OUTTIMF);
    }

    @Override
    public void process(WatchedEvent watchedEvent) {
        //获取当前的状态
        Event.KeeperState keeperState=watchedEvent.getState();
        //获取通知类型
        Event.EventType eventType=watchedEvent.getType();
        //获取节点的路径
        String path = watchedEvent.getPath();
        System.out.println("当前的状态："+keeperState+"\t通知乐行："+eventType+"节点路径："+path);
        if (Event.KeeperState.SyncConnected==keeperState){
            if (Event.EventType.None==eventType){
                System.out.println("=================连接事件回调==================");
            }
            if (Event.EventType.NodeCreated==eventType){
                System.out.println("=================创建事件回调==================");
            }
            if (Event.EventType.NodeDataChanged==eventType){
                System.out.println("=================修改事件回调==================");
            }
            if (Event.EventType.NodeDeleted==eventType){
                System.out.println("=================删除事件回调==================");
            }
        }
    }

    /*连接zk方法*/
    public void openConnection(String zk_address,int zk_session){
        try {
            zk=new ZooKeeper(zk_address,zk_session,this);
            System.out.println("===============连接成功===============");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /*创建节点*/
    public void createNode(String path,String data){
        try {
            //监听器
            zk.exists(path,true);
            String s = zk.create(path, data.getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
            System.out.println("================创建成功===================");
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    /*修改节点*/
    public void setNode(String path,String data){
        try {
            //监听器
            zk.exists(path,true);
            Stat stat = zk.setData(path, data.getBytes(), -1);
            System.out.println("===================修改成功======================");
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    /*删除节点*/
    public void delNode(String path){
        try {
            //监听器
            zk.exists(path,true);
            zk.delete(path,-1);
            System.out.println("=====================删除成功过====================");
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (KeeperException e) {
            e.printStackTrace();
        }
    }

    /*关闭连接*/
    public void closeConnection(){
        try {
            zk.close();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args){
        ZookeeperWatcher zook=new ZookeeperWatcher();
        //创建节点
        //zook.createNode("/zk01","zkValue");
        //修改节点
        //zook.setNode("/zk01","zkValueSet");
        //删除节点
        zook.delNode("/zk01");
        zook.closeConnection();
    }


}