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

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

用途

项目中使用了 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"));
    }

}