从 dubbo zookeeper 注册地址提取 zookeeper 地址

叁歲伎倆 2022-04-17 06:37 220阅读 0赞

本文仅为个人记录，方便以后查阅。

## 用途 ##

项目中使用了 dubbo，注册中心使用的 zookeeper，使用 zookeeper 实现了一个简单的分布式锁（依赖 curator），因为配置文件存在 dubbo.registry 配置，为了直接使用这个地址来创建分布式锁，写了一个简单的方法来提取 zookeeper 地址。

## 效果 ##

dubbo.registry 有多种配置方式，支持所有情况，下面是常见的例子和提取结果：

> zookeeper://localhost:2181  
> zookeeper://localhost:2181?client=zkclient  
> zookeeper://localhost:2181?backup=localhost:2182,localhost:2183  
> zookeeper://localhost:2181?client=zkclient&backup=localhost:2182,localhost:2183  
> \------------结果------------  
> Optional\[localhost:2181\]  
> Optional\[localhost:2181\]  
> Optional\[localhost:2181,localhost:2182,localhost:2182\]  
> Optional\[localhost:2181,localhost:2183,localhost:2183\]

## 代码 ##

import java.util.Optional;
    
    public class ZookeeperURL { 
        public static final String PREFIX = "zookeeper://";
        public static final String BACKUP = "backup=";
    
        public static Optional<String> convertDubboRegistryToZookeeperURL(String dubboRegistry){ 
            StringBuilder zookeeperURL = new StringBuilder();
            if(dubboRegistry != null && dubboRegistry.startsWith(PREFIX)){ 
                dubboRegistry = dubboRegistry.substring(PREFIX.length());
                int index = dubboRegistry.indexOf("?");
                if(index > 0){ 
                    zookeeperURL.append(dubboRegistry.substring(0, index));
                    dubboRegistry = dubboRegistry.substring(index + 1);
                    String[] dubboRegistries = dubboRegistry.split("&");
                    for (int i = 0; i < dubboRegistries.length; i++) { 
                        if(dubboRegistries[i].startsWith(BACKUP)){ 
                            String[] backups = dubboRegistries[i].substring(BACKUP.length()).split(",");
                            for (int j = 0; j < backups.length; j++) { 
                                zookeeperURL.append(",").append(backups[i]);
                            }
                        }
                    }
                } else { 
                    zookeeperURL.append(dubboRegistry);
                }
                return Optional.of(zookeeperURL.toString());
            }
            return Optional.empty();
        }
    
        public static void main(String[] args) { 
            System.out.println(convertDubboRegistryToZookeeperURL("zookeeper://localhost:2181"));
            System.out.println(convertDubboRegistryToZookeeperURL("zookeeper://localhost:2181?client=zkclient"));
            System.out.println(convertDubboRegistryToZookeeperURL("zookeeper://localhost:2181?backup=localhost:2182,localhost:2183"));
            System.out.println(convertDubboRegistryToZookeeperURL("zookeeper://localhost:2181?client=zkclient&backup=localhost:2182,localhost:2183"));
        }
    
    }