dubbo源码分析二：SPI扩展

概述

哈喽，咱们如约而至，今天咱一起看下dubbo的SPI扩展部分，说起这个不禁感慨当初开发dubbo的人，脑子怎么长得做出来这么优秀的产品。

我们可以基于SPI做过滤器、负载均衡器、日志打印、协议扩展等等操作，非常的方便，而且相对于jdk原生SPI做了不少优化。

咱们先实战一波，看下具体是如何使用的，然后再分析下源码。

应用实战

dubbo支持的SPI扩展有:协议扩展、调用拦截扩展、引用监听扩展、暴露监听扩展、集群扩展、路由扩展、负载均衡扩展、合并结果扩展、注册中心扩展、监控中心扩展、扩展点加载扩展、动态代理扩展、编译器扩展、配置中心扩展、消息派发扩展、线程池扩展、序列化扩展、网络传输扩展、信息交换扩展、组网扩展、Telnet 命令扩展、状态检查扩展、容器扩展、缓存扩展、验证扩展、日志适配扩展 等等。

咱们今天实现定义日志过滤器，记录rpc执行耗时，首先在消费者项目中创建LogMonitorFilter类实现Filter，代码如下：

public class LogMonitorFilter implements Filter {
   @Override
    public Result invoke(Invoker invoker, Invocation invocation) throws RpcException {
        Long startTime=System.currentTimeMillis();
        Result invoke = invoker.invoke(invocation);
        Long endTime=System.currentTimeMillis();
        System.out.println(String.format("----------------------------%s类的%s方法rpc耗时：%s",invocation.getServiceName(),invocation.getMethodName(),(endTime-startTime)));
        return invoke;
    }
}

创建 META-INF.dubbo.internal/org.apache.dubbo.rpc.Filter

文件内容如下：log=org.apache.dubbo.demo.consumer.comp.LogMonitorFilter

在rpc接口上通过注解指定用那个filter：

@Component("demoServiceComponent")
public class DemoServiceComponent implements DemoService {

    @DubboReference(interfaceClass = org.apache.dubbo.demo.DemoService.class,filter = "log")
    private DemoService demoService;
    @Override
    public String sayHello(String name) {
        return demoService.sayHello(name);
    }

    @Override
    public CompletableFuture sayHelloAsync(String name) {
        return null;
    }
}

运行程序在控制台完美打印了我们的过滤器：

接下来咱们从源码层面分析下它是如何加载我们的SPI过滤器的

源码分析

先看下包结构：

org.apache.dubbo.common.extension包下的都是关于SPI的实现，重点是ExtensionLoader这个类。

这个类有近1200行的代码，看起来不是很方便，所以我们还是采用debug的模式一步步的跟踪。

咱们先创建3个类（Car、Bmw、Test）,Car为接口 Bmw为具体实现通过test做测试，并在META-INF下创建文件如图：

Car代码如下：

@SPI
public interface Car {

    void toHome();
}

Bmw代码如下：​​​​​​​

public class Bmw implements Car{
    @Override
    public void toHome() {
        System.out.println("开着宝马回老家！");
    }
}

Test类代码：

public class Test {

    public static void main(String[] args) {
        Car bmw = ExtensionLoader.getExtensionLoader(Car.class).getExtension("bmw");
        bmw.toHome();
    }
}

直接在getExtension方法设置条件断点到了这里：

public T getExtension(String name, boolean wrap) {
        if (StringUtils.isEmpty(name)) {
            throw new IllegalArgumentException("Extension name == null");
        }
        if ("true".equals(name)) {
            return getDefaultExtension();
        }
        String cacheKey = name;
        if (!wrap) {
            cacheKey += "_origin";
        }
        final Holder