ZKClient使用

zkclient的使用记录

1.最近有一个需求是导出zk上的注册服务列表，zkclient做这个比较方便，所以就使用过程做一些记录

2.需求分析

本次这个需求需要实现2个功能：

a.取出所有分组下的服务列表，并计算提供者的个数,最终以  分组   接口   提供者个数这样的格式输出到表格

b.取出分组下的接口，截取包名，并计算同一分组下的同一包名的个数，最终以  分组     包名    个数这样 的格式输出到表格

3.代码实现

a.首先获得zk地址，设置session超时时间

    /** zookeeper地址 */
    static final String CONNECT_ADDR = "172.16.80.44:2181";

3     /** session超时时间 */
4     static final int SESSION_OUTTIME = 10000;//ms

b.获得与zk的连接

ZkClient zkc = new ZkClient(new ZkConnection(CONNECT_ADDR), SESSION_OUTTIME);

c.获得zk下的所有分组

List<String> children = zkc.getChildren("/");//  "/"代表获得zk下的根节点目录，即Dubbo的分组目录

d.对chlidren进行遍历，即对所有分组进行遍历，来获得服务的接口目录，然后对获得的接口列表进行遍历，获得所有接口名称

 for (String group : children) {
         List<String> interfaces = zkc.getChildren("/" + group);//这一步获得分组下的所有接口

           for (String interfaceName : interfaces) {//对接口列表进行遍历

                    System.out.println(interfaceName);
                   //List<String> msg = zkc.getChildren("/" + group + "/" + interfaceName + "/providers");//这样可以获得接口的提供者

           } 

}

 

这一部分主要介绍了如何获得zk连接，并获得列表的过程，主要是通过zkc.getChildren()实现的，通过group+interface+providers获得分组，接口，提供者等信息，providers.size()获得提供者个数

下面给出完整 的例子

4.完整例子

这个例子实现了获取接口列表，计算提供者个数，截取接口的包名等功能，最后输出到.csv文件。最后还进行了csv文件的合并

a.关于截取包名这个功能，想到了用substring这个方法，接口全路径名到接口名那出现大写字母，所以对于全路径名，截取大写字母之前的路径

public static String getString(String str){
        int index = -1;
        char[] chars = str.toCharArray();
        if(chars!=null){
            for(int i=0; i<chars.length; i++){
                if((chars[i] >= 'A') && (chars[i] <= 'Z')){
                    index = i;
                    break;
                }
            }

        }
        return str.substring(0, index-1);//考虑到大写字母之前还有一个点，所以index-1返回，zk上有的分组下面没有接口，会发生Index=-2越界问题，建议排除掉这些分组
    }

b.关于文件输出问题，采用了PrintWriter这个类

PrintWriter out = null;
        try {
            out = new PrintWriter(new BufferedWriter(new FileWriter("11.csv")));//11.csv是记录输出的位置
            out.println("group,prefix,number");//以","分隔目录在表格里这是三个单元格的开头

              out.println(group + "," + key + "," + map.get(key));//这是具体的输出内容，分组，包名，个数

}catch (Exception e){ 

         e.printStackTrace(); 

} 

c.关于截取包名之后如何计数的问题，在获得一个分组的所有接口列表后，进行包名截取，然后new一个map,key=包名，value=相同包名重复的次数

       Map<String, Integer> map = new HashMap<String, Integer>();//new HashMap用于存放包名和其在一个分组下出现的次数

       for (String interfaceName : interfaces) {//遍历接口列表
         try {
              String packageName = getString(interfaceName);//截取包名
              Integer times = map.get(packageName);//先从map里取值，看能不能取到值，能取到，times+1，不能取到就放入map里，times=1
              if (null != times) {
              times = times + 1;
              map.put(packageName, times);
              } else {
                     map.put(packageName, 1);
              }
         12               List<String> msg = zkc.getChildren("/" + group + "/" + interfaceName + "/providers");
              for (String s:msg) {
                     System.out.println(s);
                    }

18             } catch (Exception e) {
                       e.printStackTrace();
                        }
                    }//循环结束
          System.out.println(map); //这个map存放着对象的hascode值和重复次数
          Set<String> keys = map.keySet();//key值的集合，就是对象的 hascode的值
          for (String key : keys) {
                  out.println(group + "," + key + "," + map.get(key));//这里输出，分组，包名，同一包名的次数
                }

d.由于是要实现2个需求，所以最终文件输入到了2个csv文件中，现在需要将2个csv整合到一起，将2个csv文件放到同一个文件夹中，新建一个excel文件，打开excel文件,在sheet上右击，选择检查代码

Private Sub copy_csvfile_to_excel()

Dim MyPath$, myFile$, AK As Workbook
   Application.ScreenUpdating = False
   MyPath = ThisWorkbook.Path & "\"
   myFile = Dir(MyPath & "*.csv")
   Do While myFile <> ""
      If myFile <> ThisWorkbook.Name Then
         Set AK = Workbooks.Open(MyPath & myFile)
         AK.Sheets(1).Copy After:=ThisWorkbook.Sheets(ThisWorkbook.Sheets.Count)
         Workbooks(myFile).Close
         End If
      myFile = Dir
      Set AK = Nothing
    Loop
   Application.ScreenUpdating = True
   ActiveWorkbook.Save
   MsgBox "汇总完成，请查看！", 64, "提示"
End Sub

e.下面是这个例子的完整代码

 

package com.alibaba.dubbo.zkc.zkc;

import com.alibaba.dubbo.common.utils.StringUtils;
import org.I0Itec.zkclient.ZkClient;
import org.I0Itec.zkclient.ZkConnection;

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.*;


/**



 */
public class ZKClientBase {
    /** zookeeper地址 */
    static final String CONNECT_ADDR = "127.0.0.1:2181";//zk地址选择自己需要的,多个zk以逗号分隔，"192.168.1.171:2181,192.168.1.172:2181,192.168.1.173:2181"
    /** session超时时间 */
    static final int SESSION_OUTTIME = 10000;//ms

    public static String getString(String str){
        int index = -1;
        char[] chars = str.toCharArray();
        if(chars!=null){
            for(int i=0; i<chars.length; i++){
                if((chars[i] >= 'A') && (chars[i] <= 'Z')){
                    index = i;
                    break;
                }
            }

        }
        return str.substring(0, index-1);
    }



    public static void main(String[] args) throws Exception {
        ZkClient zkc = new ZkClient(new ZkConnection(CONNECT_ADDR), SESSION_OUTTIME);

        PrintWriter out = null;
        try {
            out = new PrintWriter(new BufferedWriter(new FileWriter("packageNumber44.csv")));
            out.println("group,prefix,number");
            List<String> children = zkc.getChildren("/");
            Map<String, Integer> map = new HashMap<String, Integer>();
            for (String group : children) {
                if (!group.equals("BSSTask") && !group.equals("zookeeper") && !group.equals("ocs_root")&&!group.equals("ZMQ")) {
                    List<String> interfaces = zkc.getChildren("/" + group);
                    //System.out.println(group);
                    for (String interfaceName : interfaces) {
                        try {
                            String packageName = getString(interfaceName);
                            Integer times = map.get(packageName);
                            if (null != times) {
                                times = times + 1;
                                map.put(packageName, times);
                            } else {
                                map.put(packageName, 1);
                            }
                            if (interfaceName.equals("com.ztesoft.inf.hsf.service.ReceiveHBResult")||interfaceName.equals("com.ztesoft.bss.svcability")){
                                List<String> msg = zkc.getChildren("/" + group + "/" + interfaceName + "/providers");
                                for (String s:msg) {
                                    System.out.println(s);
                                }

                            }
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }//循环结束
                    System.out.println(map); //这个map存放着对象的hascode值和重复次数
                    Set<String> keys = map.keySet();//key值的集合，就是对象的 hascode的值
                    for (String key : keys) {
                        out.println(group + "," + key + "," + map.get(key));
                    }
                }
            }

        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (out != null) {
                out.close();
            }
        }


    }
}

5.简单总结

网上的例子大多是自己在zk上增加和删除节点，监听节点的变化等，本次我的需求是自己去zk上获取数据，参考了网上给出的例子，实现了自己的需求，第一次写随笔，很多细节处理的都不好，如带来困扰，敬请谅解

6.参考链接

https://blog.csdn.net/t1dmzks/article/details/78440717   //zkclient用法

https://blog.csdn.net/u011089523/article/details/53421333?locationNum=14&fps=1     //将多个csv文件写入一个excel的不同sheet中

https://www.cnblogs.com/wang-zai/p/7799778.html       //截取第一个大写字母前的字符串

https://blog.csdn.net/jie1991liu/article/details/50424304  //计算一个list中某一个元素重复的参数