RPC框架（1 - 实现服务端注册一个服务）

Posted 2023-02-26

参考技术A

基于这样一个假设，那就是客户端已经知道了服务端的地址，这部分会由后续的服务发现机制完善。通用接口

hello方法需要传递一个对象，HelloObject对象，定义如下：

注意这个对象需要实现 Serializable 接口，因为它需要在调用过程中从客户端传递给服务端。

接着我们在服务端对这个接口进行实现，实现的方式也很简单，返回一个字符串就行：

服务端需要哪些信息，才能唯一确定服务端需要调用的接口的方法呢？

那么服务端知道以上四个条件，就可以找到这个方法并且调用了。我们把这四个条件写到一个对象里，到时候传输时传输这个对象就行了。即 RpcRequest 对象：

那么服务器调用完这个方法后，需要给客户端返回哪些信息呢？

如果调用成功的话，显然需要返回值，如果调用失败了，就需要失败的信息，这里封装成一个 RpcResponse 对象：

这里还多写了两个静态方法，用于快速生成成功与失败的响应对象。其中，statusCode属性可以自行定义，客户端服务端一致即可。

客户端方面，由于在 客户端这一侧我们并没有接口的具体实现类，就没有办法直接生成实例对象 。这时，我们可以 通过动态代理的方式生成实例，并且调用方法时生成需要的RpcRequest对象并且发送给服务端 。

这里我们采用JDK动态代理，代理类是需要实现 InvocationHandler 接口的。

我们需要传递host和port来指明服务端的位置。并且使用getProxy()方法来生成代理对象。

InvocationHandler 接口需要实现invoke()方法，来指明代理对象的方法被调用时的动作。 在这里，我们显然就需要生成一个RpcRequest对象，发送出去，然后返回从服务端接收到的结果即可：

生成RpcRequest很简单，我 使用Builder模式来生成这个对象 。发送的逻辑我使用了一个RpcClient对象来实现，这个对象的作用， 就是将一个对象发过去，并且接收返回的对象。

我的实现很简单，直接使用Java的序列化方式，通过Socket传输。 创建一个Socket，获取ObjectOutputStream对象，然后把需要发送的对象传进去即可，接收时获取ObjectInputStream对象，readObject()方法就可以获得一个返回的对象。

服务端的实现就简单多了， 使用一个ServerSocket监听某个端口，循环接收连接请求，如果发来了请求就创建一个线程，在新线程中处理调用。 这里创建线程采用线程池：

这里简化了一下， RpcServer暂时只能注册一个接口，即对外提供一个接口的调用服务，添加register方法，在注册完一个服务后立刻开始监听：

这里向工作线程WorkerThread传入了socket和用于服务端实例service。

WorkerThread实现了Runnable接口，用于接收RpcRequest对象，解析并且调用，生成RpcResponse对象并传输回去。 run方法如下：

其中，通过class.getMethod方法，传入方法名和方法参数类型即可获得Method对象。如果你上面RpcRequest中使用String数组来存储方法参数类型的话，这里你就需要通过反射生成对应的Class数组了。通过method.invoke方法，传入对象实例和参数，即可调用并且获得返回值。

服务端侧，我们已经在上面实现了一个HelloService的实现类HelloServiceImpl，我们只需要创建一个RpcServer并且把这个实现类注册进去就行了：

服务端开放在9000端口。

客户端方面，我们需要通过动态代理，生成代理对象，并且调用，动态代理会自动帮我们向服务端发送请求的：

我们这里生成了一个HelloObject对象作为方法的参数。

首先启动服务端，再启动客户端,测试结果：