5.RPC框架的简单实现(SPI机制-ProtocolFactory)

来源:互联网 发布:倍增时代软件好用吗 编辑:程序博客网 时间:2024/06/08 04:44

dubbo框架中广泛使用了SPI的思想,很多dubbo源码解析的相关资料也是把基于SPI机制的dubbo内核实现作为了解dubbo源码的第一讲,dubbo源码中的ExtensionFactory、Protocol、Complier等大多数接口类都是基于SPI思想的。

ldubbo这个自己实现的rpc框架也加入SPI的简单实现,在上一篇提到的Protocol接口,我们实现rmi和ldubbo两个协议,那这一个接口就有了两种实现,具体该用哪个实现呢?jdk自带的ServiceLoader就实现了SPI机制的服务查找功能。

看一下ProtocolFactory的代码:

package com.lipenglong.ldubbo.rpc;import com.lipenglong.ldubbo.rpc.protocol.LdubboProtocol;import java.util.ServiceLoader;/** * com.lipenglong.ldubbo.rpc.ProtocolFactory * </p> * Created by lipenglong on 2017/9/7. */public class ProtocolFactory {    private static final ServiceLoader<Protocol> loader = ServiceLoader.load(Protocol.class);    public static Protocol getProtocol(String name) {        for (Protocol protocol : loader) {            if (protocol.getClass().getSimpleName().toLowerCase().startsWith(name)) {                return protocol;            }        }        return new LdubboProtocol();    }}

通过jdk自带的ServiceLoader.load()方法加载Protocol接口的实现类,判断name是否实现类开始的字母并返回实现类,如果没有找到相关的实现类返回LdubboProtocol,看一下Protocol接口的子类关系:
这里写图片描述


在dubbo源码中,定义了SPI注解,接口添加上@SPI注解就会从META-INF/dubbo/internal/下读取扩展点实现,实现类里面会有一个设配类,通过设配类查找具体的实现类。我们自己实现的ProtocolFactory就简单一些,通过ServiceLoader读取一下实现类,这里需要在META-INF目录下创建services目录(jdk默认读取的目录),然后创建com.lipenglong.ldubbo.rpc.Protocol文件,添加以下文本:

com.lipenglong.ldubbo.rpc.protocol.LdubboProtocolcom.lipenglong.ldubbo.rpc.protocol.RmiProtocol

把两个实现类添加到配置中。


ProtocolFactory写完了,接下来就是两个协议类的实现了,rmi协议比较简单,下一篇我们写rmi协议的服务发布实现。