用zookeeper实现简单的发布订阅功能

发布方Provider的代码

package com.zhuyun.release.subscribe;
 
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs.Ids;
import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.data.Stat;
 
public class Provider {
    public static void main(String[] args) throws Exception {
        Watcher watcher = new Watcher() {
            @Override
            public void process(WatchedEvent event) {
                System.out.println("触发了" + event.getPath() + "的" + event.getType() + "事件！");                
            }
        };
        
        ZooKeeper zk = new ZooKeeper("192.168.10.203:2181", 20, watcher);
        
        Stat stat = zk.exists("/message", watcher);    
        if (stat == null) {                //假如节点不存在，则先创建节点
            zk.create("/message", "hello".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
        }
        
        //向该节点发送消息
        zk.setData("/message", "hello world".getBytes(), -1);
        
        
        zk.close();
    }
}

订阅方Consumer的代码：

package com.zhuyun.release.subscribe;
 
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.Watcher.Event.EventType;
import org.apache.zookeeper.ZooKeeper;
 
public class Consumer implements Watcher{
    ZooKeeper zk;
    String hostPort;
    String znode;
    
    public Consumer(String hostPort,String znode) throws Exception{
        this.hostPort = hostPort;
        this.znode = znode;
        
        zk = new ZooKeeper(hostPort, 3000, this);
        //第一次获取节点消息，同时添加watcher
        System.out.println("消息内容：" + new String(zk.getData(znode, true, null)));
    }
    
    @Override
    public void process(WatchedEvent event) {
        if (event.getType() == EventType.NodeDataChanged) {
            try {
                //当节点消息变化时，触发该操作：获取变化后的消息，同时再添加watcher
                System.out.println("你有新的消息:" + new String(zk.getData("/message", true, null)));
                
            } catch (KeeperException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }
    
    public static void main(String[] args) throws Exception {
        new Consumer("192.168.10.203:2181","/message");
 
        System.in.read();
    }
 
 
    
}

 

 

 

首先启动订阅方监听，再启动发布方发布消息，每次发布一条消息，订阅方都能收到该消息，结果如下：
消息内容：我是谁？
你有新的消息:hello
你有新的消息:你好
你有新的消息:hello world