8.zookeeper之生成分布式唯一ID

生成分布式唯一ID案例

在过去的单库单表型系统中，通常可以使用数据库字段自带的auto_increment 属性来自动为每条记录生成一个唯一的ID。但是分库分表后，就无法在依靠数据库的auto_increment属性来唯一标识一条记录了。此时我们就可以用zookeeper在分布式环境下生成全局唯一ID。

zookeeper在创建有序节点时，会生成一个序列号，而这个序列号在zookeeper中（即使在整个集群环境中）是唯一的，所以我们可以直接或者间接（比如拿原生的序列号经过运算转换成另外一种符合业务格式的ID）利用该序列号作为我们分布式系统的唯一ID。而在zookeeper中，有序节点分为临时有序节点和永久有序节点，利用zookeeper生成序列号，目的只是取一个唯一的序列号，我们并不关心节点的数据，以及节点创建后以后可能不再使用了，所以创建临时有序节点就是最佳的选择。

例如：create -e -s /uniqueId “”，最终生成的节点路径为/uniqueId0000000001，而0000000001就是我们需要的唯一ID。

设计思路：

1. 连接zookeeper服务器。
2. 指定路径生成临时有序节点。
3. 取序列号为分布式环境下的唯一ID。

package com.huazai.zookeeper.zkexample.config;

import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooKeeper;

import java.util.concurrent.CountDownLatch;

/**
 * @author pyh
 * @date 2021/5/19 0:20
 */
public class ZookeeperConnection {
    public static void main(String[] args) {
        connect();
    }

    public static ZooKeeper zooKeeper;

    public static ZooKeeper connect() {
        try {
            // 由于连接zookeeper服务器是异步连接，需要CountDownLatch阻塞主线程，等待子线程连接结果后反馈给主线程
            CountDownLatch countDownLatch = new CountDownLatch(1);
            /*
                connectString:服务器的ip和端口
                sessionTimeout:客户端与服务器之间的会话超时时间，以毫秒为单位的
                watcher：监视器对象
            */
            zooKeeper = new ZooKeeper("192.168.64.132:2181", 50000, new Watcher() {
                @Override
                public void process(WatchedEvent watchedEvent) {
                    if (watchedEvent.getState() == Event.KeeperState.SyncConnected) {
                        System.out.println("zookeeper异步连接成功");
                    } else if (watchedEvent.getState() == Event.KeeperState.Disconnected) {
                        System.out.println("断开连接");
                    } else if (watchedEvent.getState() == Event.KeeperState.Expired) {
                        System.out.println("会话超时");
                    } else if (watchedEvent.getState() == Event.KeeperState.AuthFailed) {
                        System.out.println("认证失败");
                    } else if (watchedEvent.getState() == Event.KeeperState.Closed) {
                        System.out.println("连接关闭");
                    }
                    countDownLatch.countDown();
                    System.out.println("使用构造函数默认的watcher");
                    System.out.println("path=" + watchedEvent.getPath());
                    System.out.println("eventType=" + watchedEvent.getType());
                }
            });
            // 主线程阻塞等待连接对象的创建成功
            countDownLatch.await();
            // 会话编号
            System.out.println("客户端sessionId：" + zooKeeper.getSessionId());
            return zooKeeper;
        } catch (Exception e) {
            System.out.println("zookeeper连接异常");
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 关闭zookeeper会话
     */
    public static void close() {
        if (zooKeeper != null) {
            try {
                zooKeeper.close();
                System.out.println("zookeeper关闭成功");
            } catch (InterruptedException e) {
                System.out.println("zookeeper关闭失败");
                e.printStackTrace();
            }
        }
    }
}

package com.huazai.zookeeper.zkexample.utils;

import com.huazai.zookeeper.zkexample.config.ZookeeperConnection;
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.ZooKeeper;
import org.junit.Test;

/**
 * @author pyh
 * @date 2021/6/7 23:12
 */
public class GloballyUniqueId {
    private ZooKeeper zooKeeper;

    // 用户生成序号的节点
    private String defaultPath = "/uniqueId";

    public GloballyUniqueId() {
        // 1.初始化连接对象
        zooKeeper = ZookeeperConnection.connect();
    }

    @Test
    public void parseUniqueId() throws Exception {
        System.out.println("开始获取分布式系统唯一ID......");
        for (int i = 0; i < 5; i++) {
            // 2.创建临时有序节点，成功创建的节点路径格式为/uniqueIdxxxxxxxxxx
            String nodePath = zooKeeper.create(defaultPath, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
            // 3.解析节点路径的xxxxxxxxxx，就是我们需要的序列号（唯一ID）
            String uniqueId = nodePath.substring(defaultPath.length());
            System.out.println("分布式唯一ID：" + uniqueId);
        }
    }
}