ネッティー+ ZooKeeperのシンプルなサービスの登録と発見

一. 背景

最近のプロジェクト:システムの単一のアップストリームを送信するタスクを受信した後、私たちのシステムは、関連機器の指定店、および対応するビジネスプロセスにプッシュされます。

二つ。ネッティー使用

網状で指定された、単を送信するタスクを受信した後、関連機器を格納するためにプッシュ。我々のシステムではネッティーメッセージは、長い接続とハートビートメカニズムを達成するためにプッシュします。

2.1ネッティーサーバー側:

網状のConcurrentHashMapにより、各サーバは、それが接続されているSocketChannelへのclientIdクライアントを保存します。

サーバがクライアントにメッセージを送信すると、限り対応のSocketChannelが取得CLIENTIDとして、のSocketChannel缶に書かれたメッセージに対応します。

        EventLoopGroup boss = new NioEventLoopGroup(1);
        EventLoopGroup worker = new NioEventLoopGroup();
        ServerBootstrap bootstrap = new ServerBootstrap();
        bootstrap.group(boss, worker)
                .channel(NioServerSocketChannel.class)
                .option(ChannelOption.SO_BACKLOG, 128)
                .option(ChannelOption.TCP_NODELAY, true)
                .childOption(ChannelOption.SO_KEEPALIVE, true)
                .childHandler(new ChannelInitializer() {
                    @Override
                    protected void initChannel(Channel channel) throws Exception {
                        ChannelPipeline p = channel.pipeline();
                        p.addLast(new MessageEncoder());
                        p.addLast(new MessageDecoder());
                        p.addLast(new PushServerHandler());
                    }
                });

        ChannelFuture future = bootstrap.bind(host,port).sync();
        if (future.isSuccess()) {
            logger.info("server start...");
        }
复制代码

2.2ネッティークライアント側:

その後、クライアント側のためのサービス・メッセージ、サービス処理を受けます。また、クライアントのハートビートメカニズムは、のSocketChannelが中断するかどうかを検出するためにIdleEventイベントを介してサーバーにPingメッセージを実行するように調節されて。

    public PushClientBootstrap(String host, int port) throws InterruptedException {

        this.host = host;
        this.port = port;

        start(host,port);
    }

    private void start(String host, int port) throws InterruptedException {

        bootstrap = new Bootstrap();
        bootstrap.channel(NioSocketChannel.class)
                .option(ChannelOption.SO_KEEPALIVE, true)
                .group(workGroup)
                .remoteAddress(host, port)
                .handler(new ChannelInitializer(){

                    @Override
                    protected void initChannel(Channel channel) throws Exception {
                        ChannelPipeline p = channel.pipeline();
                        p.addLast(new IdleStateHandler(20, 10, 0));  // IdleStateHandler 用于检测心跳
                        p.addLast(new MessageDecoder());
                        p.addLast(new MessageEncoder());
                        p.addLast(new PushClientHandler());
                    }
                });
        doConnect(port, host);
    }

    /**
     * 建立连接,并且可以实现自动重连.
     * @param port port.
     * @param host host.
     * @throws InterruptedException InterruptedException.
     */
    private void doConnect(int port, String host) throws InterruptedException {

        if (socketChannel != null && socketChannel.isActive()) {
            return;
        }

        final int portConnect = port;
        final String hostConnect = host;

        ChannelFuture future = bootstrap.connect(host, port);

        future.addListener(new ChannelFutureListener() {

            @Override
            public void operationComplete(ChannelFuture futureListener) throws Exception {
                if (futureListener.isSuccess()) {
                    socketChannel = (SocketChannel) futureListener.channel();
                    logger.info("Connect to server successfully!");
                } else {
                    logger.info("Failed to connect to server, try connect after 10s");

                    futureListener.channel().eventLoop().schedule(new Runnable() {
                        @Override
                        public void run() {
                            try {
                                doConnect(portConnect, hostConnect);
                            } catch (InterruptedException e) {
                                e.printStackTrace();
                            }
                        }
                    }, 10, TimeUnit.SECONDS);
                }
            }
        }).sync();
    }
复制代码

III。ZooKeeperのシンプルなサービスの登録と発見で

3.1登録サービス

サービスの性質に登録されたサービス・プロバイダーとサービス・コンシューマーを切り離すために。サービスは、強力な一貫性は、主にサービスAPIとアドレスの対応関係を格納するために使用される高可用性サービス発見保管倉庫であり、登録されました。高可用性のために、サービスレジストリは、通常、クラスタと一貫性を確保するために、分散型です。現在、その上のZooKeeper、Etcdとがあります。

私たちのプロジェクトでのZooKeeperを使用すると、サービスレジストリを実装しました。

public class ServiceRegistry {

    private static final Logger logger = LoggerFactory.getLogger(ServiceRegistry.class);

    private CountDownLatch latch = new CountDownLatch(1);

    private String registryAddress;

    public ServiceRegistry(String registryAddress) {
        this.registryAddress = registryAddress;
    }

    public void register(String data) {
        if (data != null) {
            ZooKeeper zk = connectServer();
            if (zk != null) {
                createNode(zk, data);
            }
        }
    }

    /**
     * 连接 zookeeper 服务器
     * @return
     */
    private ZooKeeper connectServer() {
        ZooKeeper zk = null;
        try {
            zk = new ZooKeeper(registryAddress, Constants.ZK_SESSION_TIMEOUT, new Watcher() {
                @Override
                public void process(WatchedEvent event) {
                    if (event.getState() == Event.KeeperState.SyncConnected) {
                        latch.countDown();
                    }
                }
            });
            latch.await();
        } catch (IOException | InterruptedException e) {
            logger.error("", e);
        }
        return zk;
    }

    /**
     * 创建节点
     * @param zk
     * @param data
     */
    private void createNode(ZooKeeper zk, String data) {
        try {
            byte[] bytes = data.getBytes();
            String path = zk.create(Constants.ZK_DATA_PATH, bytes, ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
            logger.debug("create zookeeper node ({} => {})", path, data);
        } catch (KeeperException | InterruptedException e) {
            logger.error("", e);
        }
    }
}
复制代码

サービス登録では、ネッティーサーバが起動し、IPネッティーサーバとポートがのZooKeeperに登録した後。

        EventLoopGroup boss = new NioEventLoopGroup(1);
        EventLoopGroup worker = new NioEventLoopGroup();
        ServerBootstrap bootstrap = new ServerBootstrap();
        bootstrap.group(boss, worker)
                .channel(NioServerSocketChannel.class)
                .option(ChannelOption.SO_BACKLOG, 128)
                .option(ChannelOption.TCP_NODELAY, true)
                .childOption(ChannelOption.SO_KEEPALIVE, true)
                .childHandler(new ChannelInitializer() {
                    @Override
                    protected void initChannel(Channel channel) throws Exception {
                        ChannelPipeline p = channel.pipeline();
                        p.addLast(new MessageEncoder());
                        p.addLast(new MessageDecoder());
                        p.addLast(new PushServerHandler());
                    }
                });

        ChannelFuture future = bootstrap.bind(host,port).sync();
        if (future.isSuccess()) {
            logger.info("server start...");
        }

        if (serviceRegistry != null) {
            serviceRegistry.register(host + ":" + port);
        }
复制代码

3.2サービスの検出

ここでは、クライアント・サービス・ディスカバリ、クライアントによって実現されるサービス発見メカニズムを使用しています。

クライアントとサーバーの前にレジストリを照会することによって、サーバのアドレスを取得するための接続を確立します。複数ネッティーサーバがある場合は、ロード・バランシング・サービスを行うことができます。私たちのプロジェクトでは、単純なランダム方式のロードを使用しています。

public class ServiceDiscovery {

    private static final Logger logger = LoggerFactory.getLogger(ServiceDiscovery.class);

    private CountDownLatch latch = new CountDownLatch(1);

    private volatile List<String> serviceAddressList = new ArrayList<>();

    private String registryAddress; // 注册中心的地址

    public ServiceDiscovery(String registryAddress) {
        this.registryAddress = registryAddress;

        ZooKeeper zk = connectServer();
        if (zk != null) {
            watchNode(zk);
        }
    }

    /**
     * 通过服务发现,获取服务提供方的地址
     * @return
     */
    public String discover() {
        String data = null;
        int size = serviceAddressList.size();
        if (size > 0) {
            if (size == 1) {  //只有一个服务提供方
                data = serviceAddressList.get(0);
                logger.info("unique service address : {}", data);
            } else {          //使用随机分配法。简单的负载均衡法
                data = serviceAddressList.get(ThreadLocalRandom.current().nextInt(size));
                logger.info("choose an address : {}", data);
            }
        }
        return data;
    }

    /**
     * 连接 zookeeper
     * @return
     */
    private ZooKeeper connectServer() {

        ZooKeeper zk = null;
        try {
            zk = new ZooKeeper(registryAddress, Constants.ZK_SESSION_TIMEOUT, new Watcher() {
                @Override
                public void process(WatchedEvent event) {
                    if (event.getState() == Watcher.Event.KeeperState.SyncConnected) {
                        latch.countDown();
                    }
                }
            });
            latch.await();
        } catch (IOException | InterruptedException e) {
            logger.error("", e);
        }
        return zk;
    }

    /**
     * 获取服务地址列表
     * @param zk
     */
    private void watchNode(final ZooKeeper zk) {

        try {
            //获取子节点列表
            List<String> nodeList = zk.getChildren(Constants.ZK_REGISTRY_PATH, new Watcher() {
                @Override
                public void process(WatchedEvent event) {
                    if (event.getType() == Event.EventType.NodeChildrenChanged) {
                        //发生子节点变化时再次调用此方法更新服务地址
                        watchNode(zk);
                    }
                }
            });
            List<String> dataList = new ArrayList<>();
            for (String node : nodeList) {
                byte[] bytes = zk.getData(Constants.ZK_REGISTRY_PATH + "/" + node, false, null);
                dataList.add(new String(bytes));
            }
            logger.debug("node data: {}", dataList);
            this.serviceAddressList = dataList;
        } catch (KeeperException | InterruptedException e) {
            logger.error("", e);
        }
    }
}
复制代码

クライアントは、IPアドレスとポートを取得するためにネッティーサービス発見サーバによってネッティーを開始した後。

    /**
     * 支持通过服务发现来获取 Socket 服务端的 host、port
     * @param discoveryAddress
     * @throws InterruptedException
     */
    public PushClientBootstrap(String discoveryAddress) throws InterruptedException {

        serviceDiscovery = new ServiceDiscovery(discoveryAddress);
        serverAddress = serviceDiscovery.discover();

        if (serverAddress!=null) {
            String[] array = serverAddress.split(":");
            if (array!=null && array.length==2) {

                String host = array[0];
                int port = Integer.parseInt(array[1]);

                start(host,port);
            }
        }
    }
复制代码

IV。まとめ

サービスの登録と発見は、コアコンポーネントを配布されています。この資料では、レジストリは、単純なサービスの登録と発見を実装する方法を、ZooKeeperのが何を意味について説明します。実際には、その上の例Etcd、ユーレカ、および選択のためのレジストリの多くは、あります。私たちのビジネスニーズを持つ行を選択し、最も重要です。

ます。https://juejin.im/post/5d083ab46fb9a07ee85c2b3eで再現

おすすめ

転載: blog.csdn.net/weixin_33805557/article/details/93165393