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.LdubboProtocol
com.lipenglong.ldubbo.rpc.protocol.RmiProtocol
把两个实现类添加到配置中。
ProtocolFactory写完了,接下来就是两个协议类的实现了,rmi协议比较简单,下一篇我们写rmi协议的服务发布实现。