| public class ZookeeperIpList {
private String dataId = "com.books.dubbo.demo.api.GreetingService/providers:1.0.0";
private URL CONSUMER_URL;
private static final Joiner j = Joiner.on("|").useForNull("nil");
public final List getIpList() {
return ipList;
}
private volatile List ipList = new ArrayList();
//对获取的列表内容进行过滤
private static List toUrlsWithoutEmpty(URL consumer, List providers) {
List urls = new ArrayList();
if (providers != null && providers.size() > 0) {
urls = providers.stream().map(provider -> URL.decode(provider)).filter(provider -> provider.contains("://"))
.map(provider -> URL.valueOf(provider)).filter(url -> UrlUtils.isMatch(consumer, url))
.collect(Collectors.toList());
}
return urls;
}
// 解析服务提供者地址列表为ip:port格式
private void parseIpList(List ipSet) {
List urlList = toUrlsWithoutEmpty(CONSUMER_URL, ipSet);
final List ipListTemp = urlList.stream().map(url -> url.getAddress()).collect(Collectors.toList());
this.ipList = ipListTemp;
}
public void init(String zkServerAddr, String zkGroup, String dataId, String serviceGroup) {
// 1.参数校验
Assert.notNull(zkServerAddr, "zkServerAddr is null.");
Assert.notNull(dataId, "dataId is null.");
Assert.notNull(dataId, "zkGroup is null.");
Assert.notNull(dataId, "serviceGroup is null.");
// 2.拼接订阅的path
String[] temp = dataId.split(":");
if (temp.length != 2) {
throw new RuntimeException("dataId is illegal");
}
this.dataId = "/" + zkGroup + "/" + temp[0] + "/providers";
String consumeUrl = "consumer://127.0.0.1/?group=" + serviceGroup + "&interface=" + temp[0] + "&version="
+ temp[1];
CONSUMER_URL = URL.valueOf(consumeUrl);
// 3.开启zk，订阅path路径下服务提供者信息，并添加监听器
System.out.println(j.join("init zk ", zkServerAddr, this.dataId, consumeUrl));
ZkClient zkClient = new ZkClient(zkServerAddr);
List list = zkClient.subscribeChildChanges(this.dataId, new IZkChildListener() {
@Override
public void handleChildChange(String parentPath, List currentChilds) throws Exception {
// 3.1解析服务提供者地址列表
parseIpList(currentChilds);
try {
System.out.println((j.join("ipList changed:", JSON.json(ipList))));
} catch (IOException e) {
}
}
});
//4. 解析服务提供者ip列表
parseIpList(list);
}
public static void main(String[] a) throws InterruptedException {
ZookeeperIpList zk = new ZookeeperIpList();
zk.init("127.0.0.1:2181", "dubbo", "com.books.dubbo.demo.api.GreetingService:1.0.0", "dubbo");
try {
System.out.println((j.join("parseIpList", JSON.json(zk.getIpList()))));
} catch (IOException e) {
}
Thread.currentThread().join();
}
}
|
