手写 RPC (二) 集成 Spring

最新推荐文章于 2022-10-14 13:20:43 发布

原创

最新推荐文章于 2022-10-14 13:20:43 发布 · 572 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#手写RPC #rpc集成spring #远程调用框架

本文介绍了如何将自定义的RPC框架与Spring集成，包括改造服务端的`rpc-server-provider`，客户端`rpc-client`的修改，以及实现多版本支持。通过自定义`@RpcServiceAnno`注解，配合`ApplicationContextAware`和`InitializingBean`接口，实现服务的自动发布。客户端只需将`RpcProxyClient`交给Spring管理即可。此外，文章还展示了如何在服务端和客户端处理接口的版本迭代。

【诗90:10】我们一生的年日是七十岁，若是强壮可到八十岁；但其中所矜夸的不过是劳苦愁烦，转眼成空，我们便如飞而去。

背景

在第一篇「手写RPC（一）基于BIO」我们已经完成了 RPC 的实现，那么服务的发布和调用怎样交给 spring 管理呢？

这正是下面要探讨完成的事情 ?

背景

改造服务端的 rpc-server-provider

客户端 rpc-client 改造

踩蛋 -> 实现多版本支持

rpc-server-api 服务端 api 迭代

rpc-server-provider 服务端 provicer 迭代

rpc-client 客户端修改

test

源码

改造服务端的 rpc-server-provider

在 pom.xml 文件中添加 spring 依赖

        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-context</artifactId>
            <version>4.3.23.RELEASE</version>
        </dependency>

自定义Rpc注解 @RpcServiceAnno

用以向 spring 容器中注入实例的同时，对外发布接口服务

package com.ddup.rpc.spring;
/**
 * 自定义服务发布注解：
 *     凡是加此注解的，都被发布成对外服务接口
 */
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Component // 使用此注解的类，能被spring扫描
public @interface RpcServiceAnno {
    Class<?> value(); //定义使用此注解的服务接口名称
}

业务实现类 HelloServerImpl

添加 @RpcServiceAnno：[1]交给 spring 管理 [2]容器启动时可动态处理此注解，对外发布接口

@RpcServiceAnno(IHelloSevice.class)
public class HelloServerImpl implements IHelloSevice {

    @Override
    public String sayHello(String name) {
        System.out.println("reqeust in content : " + name);
        return "hello " + name;
    }
    @Override
    public void saveUser(User user) {
        System.out.println("request in saveUser : "+ user.toString());

    }

现在 HelloServerImpl 已经 spring 容器管理了，那么接下来此服务应该怎么发布，我们应该怎样结合 spring 处理此注解呢？

【1】首先创建 RpcServerPublisher 对 IOC 容器启动时做额外处理

实现 ApplicationContextAware 接口重写 #setApplicationContext 处理添加了 @RpcServiceAnno 的 bean
实现 InitializingBean 接口重写 #afterPropertiesSet 实现启动服务的过程

package com.ddup.rpc.spring;
/**
 * RPC服务端 集成 spring
 */
public class RpcServerPublisher implements ApplicationContextAware, InitializingBean {

    private int port;
    private Map<String, Object> handlerMap  =  new HashMap<String, Object>();
    /**
     * 缓存线程池
     */
    private ExecutorService executorService = Executors.newCachedThreadPool();

    public RpcServerPublisher(int port) {
        this.port = port;
    }

    /**
     * 启动服务
     */
    @Override
    public void afterPropertiesSet() throws Exception {
        ServerSocket serverSocket = null;
        try {
            serverSocket = new ServerSocket(port);
            while (true) {
                Socket socket = serverSocket.accept();
                // 利用多线程处理任务
                executorService.execute(new ProcessorHandler(socket, handlerMap));
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (serverSocket != null) {
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 利用容器处理自定义的RPC注解
     */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(RpcServiceAnno.class);
        if (beansWithAnnotation != null) {
            for (Object bean : beansWithAnnotation.values()) {
                // 处理注解
                RpcServiceAnno rpcServiceAnno = bean.getClass().getAnnotation(RpcServiceAnno.class);
                String serviceName = rpcServiceAnno.value().getName();
                System.out.println("IOC publish service ---> "+serviceName);
                handlerMap.put(serviceName, bean);
            }
        }
    }

    public int getPort() {
        return port;
    }

    public void setPort(int port) {
        this.port = port;
    }
}