手写RPC---⑧实现server模块(下)

注册方法写完了，需要写查找方法

查找方法

    //查找方法 需要返回一个ServiceInstance(表示一个具体的服务) 传进来的参数是一个RPC的request
    public ServiceInstance lookup(Request request){
        //先拿到请求过来的服务 
        ServiceDescriptor sdp = request.getService();
        //把这个服务作为KEY 获取到ServiceInstance 返回
        
        return service.get(sdp);

    }

重写 hashCode()和equals()方法

ServiceDescriptor类需要重写 hashCode()和equals(Object obj)方法 因为这是自己定义的类，
map.get是通过equals来判断key是否相等的，而这里的key是ServiceDescriptor，所以需要重写equals方法

ServiceDescriptor类：

  @Override
    public int hashCode() {
        return toString().hashCode();
    }
    @Override
    public boolean equals(Object obj) {
        //如果对象一样
        if(this == obj) return true;
        //如果为空 或者类型不一样
        if(obj == null || getClass() != obj.getClass())
            return false;
        //最后一个情况 类型一样 那就比字符串是否相等
        //强转一下
        ServiceDescriptor  sd = (ServiceDescriptor) obj;
        return this.toString().equals(sd.toString());
    }

    @Override
    public String toString() {
        return "clazz = " + clazz +"method = "  + method +"returntype = " +returnType
                + "parameterTypes = " + Arrays.toString(parameterTypes);



    }

调用service实例的一个辅助类

//调用service实例的一个辅助类(调用具体的服务)
public class ServiceInvoker {
    //参数1：表示一个具体的服务   参数2：表示RPC的一个请求
    public Object invoke(ServiceInstance serviceInstance, Request request){
                                 //invoke()调用指定对象的指定方法
                         // 三个参数 分别是 1.被调用的对象 2、被调用的方法 3、方法的参数
        return ReflectionsUtils.invoke(
                serviceInstance.getTarget(),
                serviceInstance.getMethod(),
                request.getParameters()
        );
        
    }
}

RpcServer

RPC的服务类：代码有详细的注释 按照顺序看即可

@Slf4j
public class RpcServer {
    //配置类
    private RpcServerConfig config;
    //网络模块
    private TransportServer net;
    //序列化 反序列化
    private Encoder encoder;
    private Decoder decoder;
    //管理服务的
    private ServiceManager serviceManager;
    //调用服务的
    ServiceInvoker serviceInvoker;

    public RpcServer(RpcServerConfig config) {
        this.config = config;
        //net                       //根据参数返回对象，参数是config的transportClass类对象
        this.net = ReflectionsUtils.newInstance(config.getTransportClass());
        //调用初始化init方法
        this.net.init(config.getPort(),this.handler);
        //codec
        this.encoder = ReflectionsUtils.newInstance(config.getEncoderClass());
        this.decoder = ReflectionsUtils.newInstance(config.getDncoderClass());
        //service
        this.serviceManager = new ServiceManager();
        this.serviceInvoker = new ServiceInvoker();


    }
    //服务注册
    public <T> void register(Class<T> interfaceClass,T bean){
        serviceManager.register(interfaceClass,bean);
    }

    //启动方法
    public  void  start(){
        this.net.start();//网络监听
    }
    public  void  stop(){
        this.net.stop();
    }
    private RequestHandler handler = new RequestHandler() {
        @Override
        public void onRequest(InputStream recive, OutputStream toResp) {
                //第一步是从revice读取请求数据 即request序列化之后的二进制数据，
            // 读取之后通过serviceInvoker调用服务
            //第二步 拿到数据 toResp写回去
            Response response = new Response();
            try {
                //读取数据
                byte[] bytes = IOUtils.readFully(recive, recive.available());
                //反序列化后 成一个request对象 包含了请求数据
                Request request = decoder.decode(bytes, Request.class);
                log.info("get request: {}",request);
                //有了这个request之后 就能找到服务了
                ServiceInstance instance = serviceManager.lookup(request);
                //调用服务 参数1：具体的服务  2：一个请求
                Object invoke = serviceInvoker.invoke(instance, request);
                //调用服务后 服务器返回的invoke信息(数据)
                response.setData(invoke);
            } catch (Exception e) {
               log.warn(e.getMessage(),e);
               response.setCode(1);
               response.setMessage("发生异常" + e.getClass().getName()+
                       ":" + e.getMessage());
            }finally {
                //把response序列化  然后response信息返回给客户端
                try {
                    byte[] out = encoder.encode(response);
                    toResp.write(out);

                    log.info("返回response给客户端");
                } catch (IOException e) {
                    log.warn(e.getMessage(),e);
                    }
            }
        }
    };

}