前言
RPC 框架是后端攻城狮永远都绕不开的知识点,目前业界比较知名有 Dubbo
、Spring Cloud
等。很多人都停留在了只会用的阶段,作为程序猿,拥有好奇心深入学习,才能有效提高自己的竞争力。再进一层的同学,会去翻源码,看功能是如何实现的,这是很好的开始。看源码过一段时间容易忘记,我觉得看完源码之后,更好的做法是自己动手开发一个出来,这样你对框架的理解会更深。我认为,“会用”、“会读源码”、"会写出来"是完全不一样的水平。
本系列 “造轮子系列之RPC”,手把手教大家如何打造自己的RPC框架。
以下是我个人写的简单版 RPC
框架 ccx-rpc
的源码,欢迎 Star、Fork。水平有限,大家有更好的想法可以提出来。
Github:https://github.com/chenchuxin/ccx-rpc
Gitee:https://gitee.com/imccx/ccx-rpc
RPC 框架的结构
一个最简单的 RPC 框架分成三个部分:注册中心、服务端、客户端。以下是一个最简单的结构流程图。
组成部分:
- 注册中心:用于注册和获取服务。
- 服务端:指提供服务的一方,也叫服务提供方
Provider
- 客户端:指调用服务的一方,也叫服务消费者
Consumer
流程:
- 服务端把服务信息注册到注册中心,通常包含服务端地址、接口类和方法
- 客户端从注册中心获取对应服务的信息
- 客户端根据服务的信息,通过网络调用到服务端的接口
RPC 框架的设计
上面的流程有很多细节没有画出来,例如:
- 服务端以什么形式注册到注册中心?
- 客户端是怎么做到像调用接口一样调用服务?
- 调用服务的网络协议是怎样的?
一个基本的 RPC 框架,需要包含以下部分:
- 注册中心:注册中心负责服务信息的注册与查找。服务端在启动的时候,扫描所有的服务,然后将自己的服务地址和服务名注册到注册中心。客户端在调用服务之前,通过注册中心查找到服务的地址,就可以通过服务的地址调用到服务啦。常见的注册中心有
Zookeeper
、Eureka
等。 - 动态代理:客户端调用接口,需要框架能自己根据接口去远程调用服务,这一步是用户无感知的。这样一来,就需要使用到动态代理,用户调用接口,实际上是在调用动态生成的代理类。常见的动态代理有:
JDK Proxy
,CGLib
,Javassist
等。 - 网络传输:RPC 远程调用实际上就是网络传输,所以网络传输是 RPC 框架中必不可少的部分。网络框架有
Java NIO
、Netty
框架等。 - 自定义协议:网络传输需要制定好协议,一个良好的协议能提高传输的效率。
- 序列化:网络传输肯定会涉及到序列化,常见的序列化有
Json
、Protostuff
、Kyro
等。 - 负载均衡:当请求调用量大的时候,需要增加服务端的数量,一旦增加,就会涉及到符合选择服务的问题,这就是负载均衡。常见的负载均衡策略有:轮询、随机、加权轮询、加权随机、一致性哈希等等。
- 集群容错:当请求服务异常的时候,我们是应该直接报错呢?还是重试?还是请求其他服务?这个就是集群容错策略啦。
代码实现概览
下面我们从代码的角度上,来看看以上几部分是如何组织的:
服务注册、监听
1. 扫描服务
服务要注册到注册中心,第一步是需要扫描到需要注册的接口。
我们通过 Spring
的 BeanPostProcessor#postProcessBeforeInitialization
,将带有 @RpcService
注解的接口进行发布。
@Component
public class ServiceBeanPostProcessor implements BeanPostProcessor {
@Override
public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
RpcService rpcService = bean.getClass().getAnnotation(RpcService.class);
// rpc 服务发布到注册中心
if (rpcService != null) {
RegistryFactory registryFactory = ExtensionLoader.getLoader(RegistryFactory.class).getAdaptiveExtension();
RegistryConfig registryConfig = ConfigManager.getInstant().getRegistryConfig();
Registry registry = registryFactory.getRegistry(registryConfig.toURL());
registry.register(buildServiceURL(bean, rpcService));
}
return bean;
}
}
2. 注册中心
服务注册最终的表现就是:把服务信息注册到注册中心中。
根据注册中心的特性,可以抽出一个接口 Registry
,包含了注册、取消注册、查找服务的方法。
通过实现 Registry
接口,可以扩展出多种类型的注册中心。
public interface Registry {
/**
* 向注册中心注册服务