造一个轮子，实现RPC调用

在写了一个Netty实现通信的简单例子后，萌发了自己实现RPC调用的想法，于是就开始进行了Netty-Rpc的工作，实现了一个简单的RPC调用工程。

如果也有兴趣动手造轮子的同学，可以先看看之前写的使用Java实现Netty通信这篇博客。

本文源地址:造一个RPC的轮子

准备

首先，你需要明白下列知识。

Netty

处理服务之间的通信。

Zookeeper

服务注册与发现。

SpringBoot

目前单单只是作为启动项目。

Cglib Proxy & Jdk Reflect

使用代理实例化暴露的接口，通过反射调用接口方法的实现。

RPC处理流程

服务提供者通过Netty进行对应的端口暴露。

同时提供者将需要暴露的服务信息注册到Zookeeper，Zookeeper注册的节点信息为接口的类路径，注册的Data为暴露的端口，并且需要将对应的接口实现类在ApplicationContext上下文中进行保存。

此时消费者启动后，会根据对应接口的类路径在Zookeeper进行Discover。

根据对应接口的类路径在Zookeeper中通过ReadData获取到对应暴露的端口信息。

拿到了端口信息，通过Netty发起请求。

Netty发起请求接收后，通过之前的ApplicationContext上下文调用对应的接口方法。

ApplicationContext调用成功后，Netty将响应的结果返回。

服务消费者得到响应结果，RPC流程结束。

动手

在使用Java实现Netty通信这篇文章的基础上，我们新建下列工程信息。

rpc-sample-api

依赖信息

<dependency>

    <groupId>com.yanzhenyidai</groupId>

    <artifactId>netty-rpc-common</artifactId>

    <version>1.0-SNAPSHOT</version>

</dependency>

代码

只需要定义一个HiService接口。

public interface HiService {

    public String hi(String msg);

}

rpc-sample-server

依赖信息

<dependency>

    <groupId>org.springframework.boot</groupId>

    <artifactId>spring-boot-starter-web</artifactId>

    <version>2.0.3.RELEASE</version>

</dependency>

<dependency>

    <groupId>com.yanzhenyidai</groupId>

    <artifactId>rpc-sample-api</artifactId>

    <version>1.0-SNAPSHOT</version>

</dependency>

<dependency>

    <groupId>com.yanzhenyidai</groupId>

    <artifactId>netty-rpc-server</artifactId>

    <version>1.0-SNAPSHOT</version>

</dependency>

application.yaml

register.address: 127.0.0.1:3000

代码

首先实现HiService接口。

@RpcServer(cls = HiService.class)

public class HiServiceImpl implements HiService {

    public String hi(String msg) {

        return "hello, I'm Rpc, I want say : " + msg;

    }

}

Server类。

@Component

public class Server implements ApplicationContextAware {

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

    @Value("${register.address}")

    private String registerAddress;

    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {

        Map<String, Object> serviceBean = new HashMap<String, Object>();

        Map<String, Object> objectMap = applicationContext.getBeansWithAnnotation(RpcServer.class);

        for (Object object : objectMap.values()) {

            try {

                RpcServer annotation = object.getClass().getAnnotation(RpcServer.class);

                serviceBean.put("/yanzhenyidai/" + annotation.cls().getName(), object);

                String[] split = registerAddress.split(":");

                new NettyServer(split[0], Integer.valueOf(split[1])).server(serviceBean);

            } catch (Exception e) {

                logger.error("[server-start] fail ", e);

            }

        }

    }

}

实现 ApplicationContextAware 以获取到Spring上下文，通过扫描RpcServer注解，得到本次需要暴露的服务信息，并且开启NettyServer的端口服务暴露及Zookeeper注册。

Application启动类

@SpringBootApplication

public class RpcServerApplication {

    public static void main(String[] args) {

        new SpringApplicationBuilder(RpcServerApplication.class)

                .web(WebApplicationType.NONE)

                .run(args);

    }

}

开启SpringBoot无Web启动。

rpc-sample-client

依赖

<dependency>

    <groupId>com.yanzhenyidai</groupId>

    <artifactId>netty-rpc-client</artifactId>

    <version>1.0-SNAPSHOT</version>

</dependency>

<dependency>

    <groupId>com.yanzhenyidai</groupId>

    <artifactId>rpc-sample-api</artifactId>

    <version>1.0-SNAPSHOT</version>

</dependency>

<dependency>

    <groupId>cglib</groupId>

    <artifactId>cglib</artifactId>

    <version>3.2.2</version>

</dependency>

<dependency>

    <groupId>asm</groupId>

    <artifactId>asm</artifactId>

    <version>3.3.1</version>

</dependency>

代码

Client

public class Client {

    public <T> T create(final Class<?> cls) {

        return (T) Proxy.newProxyInstance(cls.getClassLoader(), new Class<?>[]{cls}, new InvocationHandler() {

            public Object invoke(Object o, Method method, Object[] objects) throws Throwable {

                Request request = new Request();

                request.setInterfaceName("/yanzhenyidai/" + cls.getName());

                request.setRequestId(UUID.randomUUID().toString());

                request.setParameter(objects);

                request.setMethodName(method.getName());

                request.setParameterTypes(method.getParameterTypes());

                Response response = new NettyClient().client(request);

                return response.getResult();

            }

        });

    }

}

使用Cglib动态代理先实例化接口信息，在调用的时候，通过NettyClient发送请求到NettyServer，由NettyServer处理发现Zookeeper节点以及反射调用接口实现方法。

context.xml

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

       xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd">

    <bean class="com.yanzhenyidai.config.Client"></bean>

</beans>

注入Client的bean对象信息。

Application

public class RpcClientApplication {

    public static void main(String[] args) {

        ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext("context.xml");

        Client client = context.getBean(Client.class);

        HiService hiService = client.create(HiService.class);

        String msg = hiService.hi("msg");

        System.out.println(msg);

    }

}

运行结果

总结

以上只是一个简单的RPC过程，相对于Dubbo RPC会更加直观的看明白，希望能对大家有所帮助作用，也欢迎大家和我一起造轮子。

本项目Github地址：Netty-RPC

巴特西

Java使用Netty实现简单的RPC

造一个轮子，实现RPC调用

准备

RPC处理流程

动手

rpc-sample-api

rpc-sample-server

rpc-sample-client

总结

最新文章

热门文章