Dubbo源码解析-核心SPI扩展

Dubbo版本：2.7.5
源码下载地址：https://github.com/apache/dubbo/releases/tag/dubbo-2.7.5
编译工具：IDEA

下载好源码后，导入idea

• SPI机制（spi机制的思想提供一种更加灵活的，可插拔式的机制）
• 自适应扩展点
• IOC和 AOP
• Dubbo与Spring 如何集成

Dubbo核心之SPI

1.Dubbo源码很多地方中存在这三种代码：分别是自适应扩展点，指定名称扩展点，激活扩展点
ExtensionLoader.getExtensionLoader(xxx.class).getAdaptiveExtension() 自适应扩展点
ExtensionLoader.getExtensionLoader(xxx.class).getExtension(name); 指定名称扩展点
ExtensionLoader.getExtensionLoader(xxx.class).getActivateExtension(url,key) 激活扩展点

这种扩展点实际上就是Dubbo中的SPI机制。在Springboot中的SpringFactoriesLoader就是一种SPI机制。

java中的SPI机制的实现
SPI机制原本是JDK内置的一种服务提供发现机制，主要用来做服务的扩展实现。例如JDK中提供了java.sql.Driver接口，这个驱动类在JDK中并没有实现，而是由不同的数据库厂商来实现，比如mysql、oracle驱动包都会去实现这个接口，然后JDK利用SPI机制从classpath中找到对应的驱动来获取指定数据库连接，这种插拔式的扩展加载方式，也同样遵循一定的协议规定。比如所有的扩展点都要放在resource/META-INF/services目录下，SPI机制会默认扫描这个路径下的属性文件完成加载

Dubbo自定义协议扩展点

• Dubbo和SpringFactoriesLoader 都没有使用java的SPI机制，而是利用了SPI的思想。
• Dubbo的SPI机制都被封装在 ExtensionLoader 类中，通过ExtensionLoader 来加载指定的实现类

Dubbo的SPI机制扩展有两个规则：

• 和JDK内置SPI一样，需要在resouce目录下创建任一目录结构：META-INF/dubbo、META-INF/dubbo/internal、META-INF/services,在对应的目录下创建接口全路径名称文件，Dubbo会去这三个目录下加载相应扩展点
• 文件内容和JDK SPI内容不同，是一种Key和Value形式的数据。Key：字符串 value:对应扩展点的实现。这样的方式可以按需加载指定的实现类

实现步骤：

• 在一个依赖了Dubbo框架的工程中，创建一个扩展点及实现，其中扩展点需要声明@SPI注解

创建Driver 接口

import org.apache.dubbo.common.extension.SPI;

@SPI
public interface Driver {
    String connect();
}

MysqlDriver 实现接口

public class MysqlDriver implements Driver {

    @Override
    public String connect() {
        return "连接MYSQL数据库";
    }
}

在resouces文件夹下新建,以接口全路径名作为文件名

image.png

文件中内容：KEY： 字符串 value：具体扩展点实现

image.png

测试：

public class MysqlDriverTest {

    @Test
    public void connect() {
        ExtensionLoader extensionLoader = ExtensionLoader.getExtensionLoader(Driver.class);
        Driver mysqlDriver = extensionLoader.getExtension("mysqlDriver");
        System.out.println(mysqlDriver.connect());
    }
}

输出结果：

image.png

Dubbo SPI扩展点源码分析

整体逻辑：先根据ExtensionLoader.getExtensionLoader获取到ExtensionLoader实例，然后再通过getExtension（）方法获得指定名称的扩展点。

ExtensionLoader.getExtensionLoader

用来返回一个ExtensionLoader实例

    public static  ExtensionLoader getExtensionLoader(Class type) {
        if (type == null) {
            throw new IllegalArgumentException("Extension type == null");
        } else if (!type.isInterface()) {
            throw new IllegalArgumentException("Extension type (" + type + ") is not an interface!");
        } else if (!withExtensionAnnotation(type)) {
            throw new IllegalArgumentException("Extension type (" + type + ") is not an extension, because it is NOT annotated with @" + SPI.class.getSimpleName() + "!");
        } else {
            //先从缓存中获取和扩展类对应的ExtensionLoader
            ExtensionLoader loader = (ExtensionLoader)EXTENSION_LOADERS.get(type);
            //缓存未命中
            if (loader == null) {
                //创建一个新实例放入缓存，
                EXTENSION_LOADERS.putIfAbsent(type, new ExtensionLoader(type));
                loader = (ExtensionLoader)EXTENSION_LOADERS.get(type);
            }

            return loader;
        }
    }

ExtensionLoader的构造方法，初始化一个objectFactory

    private ExtensionLoader(Class type) {
        this.type = type;
        this.objectFactory = type == ExtensionFactory.class ? null : (ExtensionFactory)getExtensionLoader(ExtensionFactory.class).getAdaptiveExtension();
    }

getExtension（）

这个方法用于根据指定名称获得对象的扩展点并返回。在前面的例子中，name就是mysqlDriver，返回的实例就是MysqlDriver

• 如果name="true" 则返回一个默认的扩展类实现
• 创建一个 Holder对象，缓存该扩展点实例
• 如果该扩展点不存在，则createExtension（）创建扩展点

   public T getExtension(String name) {
        if (StringUtils.isEmpty(name)) {
            throw new IllegalArgumentException("Extension name == null");
        // 如果name="true" 则返回一个默认的扩展类实现
        } else if ("true".equals(name)) {
            return this.getDefaultExtension();
        } else {
            //创建一个 Holder对象，缓存该扩展点实例
            Holder