自定义 RPC框架4——RMI+Zookeeper实现RPC框架

准备工作

这次我们用RMI+Zookeeper实现一个远程调用的RPC框架，RMI实现远程调用，Zookeeper作为注册中心，具体的操作之前的文章都提到过，这里不再做过多赘述。

自定义 RPC框架2——RMI实现RPC

https://blog.csdn.net/qq_45587153/article/details/124211478?spm=1001.2014.3001.5502

自定义 RPC框架3——JAVA实现Zookeeper节点增删改查

https://blog.csdn.net/qq_45587153/article/details/124225572?spm=1001.2014.3001.5502

代码实现

项目结构

• ZkConnection负责创建一个zookeeper对象并返回

• ShenRpcRegistry

  • 提供注册服务的方法，将服务节点保存到Zookeeper并注册到Registry中，
  • 提供一个获取服务的方法，从Zookeeper中获取节点，根据查询的结果创建一个代理对象返回

• ShenRpcFactory框架入口

  • 使用该框架需要两个配置文件:shen.properties和shen-services.properties

  • 提供快速注册服务方法和快速获取代理对象方法，以及基本的连接和批量注册

  • shen.properties里面写响应的的配置:

    • registry.ip=服务器IP地址，默认为localhost

    • registry.port=服务端端口号，默认为9090

    • zk.server=Zookeeper访问地址，默认为localhost:2181

    • zk.sessionTimeout=Zookeeper连接会话超时，默认为10000

    • 例如：

    • registry.port=9999
      zk.server=129.211.65.241:2181
      zk.sessionTimeout=20000
      

  • shen-service.properties里面写需要批量注册的服务，例如：

    com.shen.service.UserService=com.shen.service.impl.UserServiceImpl
    com.shen.service.CustomService=com.shen.service.impl.CustomServiceImpl
    

导入POM依赖

导入zookeeper依赖的同时，要排除其中的logback依赖。

<dependency>
    <groupId>org.apache.zookeeper</groupId>
    <artifactId>zookeeper</artifactId>
    <version>3.4.10</version>
    <exclusions>
        <exclusion>
            <groupId>ch.qos.logback</groupId>
            <artifactId>logback-classic</artifactId>
        </exclusion>
        <exclusion>
            <groupId>ch.qos.logback</groupId>
            <artifactId>logback-core</artifactId>
        </exclusion>
    </exclusions>
</dependency>

ZkConnection

package com.shen.rpc.connection;

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

import java.io.IOException;

//专门提供zookeeper连接的自定义类型
public class ZkConnection {
   
    //保存ZK的地址，格式是ip:port,如：129.211.65.241:2181
    private String zkServer;
    //保存会话超时时间
    private int sessionTimeout;

    public ZkConnection(){
   
        super();
        this.zkServer = "localhost:2181";
        this.sessionTimeout = 10000;
    }

    public ZkConnection(String zkServer,int sessionTimeout){
   
        this.zkServer = zkServer;
        this.sessionTimeout = sessionTimeout;
    }

    public ZooKeeper getConnection() throws IOException {
   
        return new ZooKeeper(this.zkServer, this.sessionTimeout, new Watcher() {
   
            @Override
            public void process(WatchedEvent watchedEvent) {
   
                System.out.