packagecom.toov5.zkDubbo;importjava.io.BufferedReader;importjava.io.IOException;importjava.io.InputStreamReader;importjava.io.PrintWriter;importjava.net.Socket;importjava.util.ArrayList;importjava.util.List;importorg.I0Itec.zkclient.IZkChildListener;importorg.I0Itec.zkclient.ZkClient;public classZkServerClient {public static List listServer = new ArrayList();public static voidmain(String[] args) {
initServer();
ZkServerClient client= newZkServerClient();
BufferedReader console= new BufferedReader(newInputStreamReader(System.in));while (true) {
String name;try{
name=console.readLine();if ("exit".equals(name)) {
System.exit(0);
}
client.send(name);
}catch(IOException e) {
e.printStackTrace();
}
}
}//注册所有server
public static voidinitServer() {
listServer.clear();//listServer.add("127.0.0.1:18080");//连接服务 放到list中 存放集群地址 做负载均衡用的//建立zk连接
final ZkClient zkClient = new ZkClient("192.168.91.5",5000,10000);//读取父节点信息
final String root = "/toov5";
List children=zkClient.getChildren(root);for(String pString : children){//父+子 =完整的路径
String path = root+"/"+pString;
String nodeValue=zkClient.readData(path);//把值放到集合中去
listServer.add(nodeValue);
}//服务发现成功
System.out.println("服务发现:"+listServer.toString());//然后给 serverCount赋值
serverCount=listServer.size();//使用zk事件监听,如果服务发生宕机情况,重新读取新的节点
zkClient.subscribeChildChanges(root, newIZkChildListener() {public void handleChildChange(String arg0, List children) throws Exception { //children移除之后 剩下的
System.out.println("有服务宕机了,重新读取新的节点信息"+listServer.toString());
listServer.clear();//清除之前的 然后读取最新的
for(String pString : children){//父+子 =完整的路径
String path = root+"/"+pString;
String nodeValue=zkClient.readData(path);//把值放到集合中去
listServer.add(nodeValue);
}
serverCount=children.size();
System.out.println("服务发现"+listServer.toString());
}
});
}//请求总数
private static int reqCount = 1;//服务个数
private static int serverCount = 0; //初始值是0//获取当前server信息
public staticString getServer() {//return listServer.get(0); 不能写死 本地负载均衡轮询算法
String serverName = listServer.get(reqCount%serverCount );
System.out.println("客户端请求次数:"+reqCount+"对应服务器"+serverName);
reqCount++;returnserverName;
}public voidsend(String name) {
String server=ZkServerClient.getServer();
String[] cfg= server.split(":");
Socket socket= null;
BufferedReader in= null;
PrintWriter out= null;try{
socket= new Socket(cfg[0], Integer.parseInt(cfg[1]));
in= new BufferedReader(newInputStreamReader(socket.getInputStream()));
out= new PrintWriter(socket.getOutputStream(), true);
out.println(name);while (true) {
String resp=in.readLine();if (resp == null)break;else if (resp.length() > 0) {
System.out.println("Receive : " +resp);break;
}
}
}catch(Exception e) {
e.printStackTrace();
}finally{if (out != null) {
out.close();
}if (in != null) {try{
in.close();
}catch(IOException e) {
e.printStackTrace();
}
}if (socket != null) {try{
socket.close();
}catch(IOException e) {
e.printStackTrace();
}
}
}
}
}