DUBBO源码分析(一)---内核SPI扩展实现

阅读dubbo代码的困惑

在阅读dubbo源码的过程中,最初带给我最大的困惑就是在多实现的接口中,如何查找当前上下文中的接口实现。例如:

在dubbo源码中这样格式的代码大量存在,所以在学习dubbo代码前,一定要了解dubbo spi机制。这样对于dubbo的学习,会有事半功倍的效果。

dubbo SPI简介

dubbo采用微内核+插件的设计原则,即dubbo的所有的功能点都可以被用户自定义的插件覆盖。这种设计原是通过service provider interface(SPI)的方式实现的,就是厂商定义了服务标准,用户去实现。jdk中也实现了这点,至于dubbo为什么没有采用jdk标准的spi,而是自己重新实现了,官方给出的即解释是jdk会一次实例化所有的插件,即使有的插件用不到。dubbo的开发者认为这是一种资源浪费,dubbo想要的是只有在用到某种扩展插件的时候,再去实例化。

例如,dubbo默认支持了ConsistentHashLoadBalance(一致性hash)、LeastActiveLoadBalance(最小活跃调用数)、RandomLoadBalance(按权重设置随机概率)、RoundRobinLoadBalance(按公约后的轮训设置权重比例)四种负载均衡策略。用户可以再此基础上扩展自己的负载均衡策略,并在consumer端指定自定义的负载策略。

dubbo SPI文件声明

SPI扩展文件存储在以下三个路径

dubbo支持的各种扩展实现都存储在第一个目录中了,但是dubbo在加载插件时仍然会扫描这三个目录。在每个spi文件中,格式定义为:扩展名=扩展实现类路径。如下:

spi文件名称格式:接口的全路径名称,如下:

com.alibaba.dubbo.rpc.cluster.LoadBalance

Protocol加载过程

下面以Protocol为实例,具体分析下dubbo加载Protocol实例的过程

Protocol protocol = ExtensionLoader.getExtensionLoader(Protocol.class).getAdaptiveExtension();

看下com.alibaba.dubbo.rpc.Protocol这个接口声明,发现了两个注解@SPI、@Adaptive。

1、@SPI: 指定接口的默认实现策略名称,这里LoadBalance默认的实现策略是RandomLoadBalance 2、@Adaptive: 可以标记在SPI扩展实现类上,也可以标记在SPI接口的方法上。如果标记在扩展类上,SPI获取的实例是Adaptive+接口名称.class,如AdaptiveExtensionFactory、AdaptiveCompiler;如果标记在接口方法上,SPI对应的实例是动态代理类, 如LoadBalance$Adpative.class。

ExtensionLoader初始化

在getExtensionLoader的过程中,这里使用单例的模式,每个加载的type类型返回一个new ExtensionLoader(type)实例。看一下这个构造函数:

  private ExtensionLoader(Class type) {

        this.type = type;

        objectFactory = (type == ExtensionFactory.class ? null :       

            ExtensionLoader.getExtensionLoader(ExtensionFactory.class).getAdaptiveExtension());

  }

objectFactory这个对象工厂是用于加载Protocol实例中的成员变量,例如RegistryProtocol实例中的cluster、protocol、registryFactory等,如果type类型等于ExtensionFactory则返回空,具体加载过程下文中会说到。另外,注意ExtensionFactory这个扩展工厂接口,就是上文说到的@Adaptive标记在扩展实现类上的,默认返回实例AdaptiveExtensionFactory。

getAdaptiveExtension()

这里获取一个Protocol接口的实现,由于Protocol接口中的@Adaptive注解是标记在方法上的,所有这里应该返回一个Protocol$Adpative.class的动态实例,具体过程如下:

  private T createAdaptiveExtension() {

        try {

            return injectExtension((T) getAdaptiveExtensionClass().newInstance());

        } catch (Exception e) {

            throw new IllegalStateException("Can not create adaptive extenstion " + type + ", cause: " + e.getMessage(), e);

        }

  }


getAdaptiveExtensionClass().newInstance()返回Protocol$Adpative.class的动态实例。注意下injectExtension这个函数,用上文说到的objectFactory对象工厂,在入参Protocol$Adpative.class的情况时,objectFactory并不会working,因为没有set开头的method。接下来我们看下Protocol$Adpative.class的创建过程,揭开Protocol$Adpative.class的神秘面纱。

# getAdaptiveExtensionClass()

  private Class getAdaptiveExtensionClass() {

        getExtensionClasses();

        if (cachedAdaptiveClass != null) {

            return cachedAdaptiveClass;

        }

        return cachedAdaptiveClass = createAdaptiveExtensionClass();

    }


1、首先getExtensionClasses(),顾名思义是加载所有的扩展类。翻代码跟进去loadFile函数,就是遍历上文说到的SPI扩展文件存储的三个路径,利用class.forName将文件中的扩展类加载到内存中,并以缓存起来。在加载的过程中,会判断当前class是否有标记在类上的@Adaptive注解,如果有赋值给cachedAdaptiveClass

  if (clazz.isAnnotationPresent(Adaptive.class)) {

        if(cachedAdaptiveClass == null) {

              cachedAdaptiveClass = clazz;

        } else if (! cachedAdaptiveClass.equals(clazz)) {

              throw new IllegalStateException("More than 1 adaptive class found: "

                      + cachedAdaptiveClass.getClass().getName()

                      + ", " + clazz.getClass().getName());

        }

  }


这里要强调下的是,大家是不还记得前文提到的dubbo之所以没有采用jdk标准的spi,而是自己重写了。只有在用到的时候才去加载扩展资源类。 

2、能走进这个if分支的,应该是@Adaptive注解标记在扩展类上的,目前有AdaptiveCompiler、AdaptiveExtensionFactory等,Protocol接口没有@Adaptive注解标记在扩展类上的实现。

  if (cachedAdaptiveClass != null) {

      return cachedAdaptiveClass;

  }

3、 createAdaptiveExtensionClass()动态创建Protocol$Adpative.class实例

  private Class createAdaptiveExtensionClass() {

        String code = createAdaptiveExtensionClassCode();

        ClassLoader classLoader = findClassLoader();

        com.alibaba.dubbo.common.compiler.Compiler compiler =                ExtensionLoader.getExtensionLoader(com.alibaba.dubbo.common.compiler.Compiler.class).getAdaptiveExtension();

        return compiler.compile(code, classLoader);

    }


分析第一行代码createAdaptiveExtensionClassCode(),这个函数很长有200多行,大家读到这里不要慌,函数的目的很简单,就是字符串拼接,并且校验Protocol接口中是否有标记在method上的@Adaptive注解。如果没有则抛出异常。拼接的结果如下:

DUBBO源码分析(一)---内核SPI扩展实现_第1张图片

这个就是Protocol$Adpative代理类的样子,这么拼接好处是性能很高,缺点是拼接过程很繁琐。接下来的编译过程,dubbo默认交给了javassist来完成。这样就完成了在当前classloader中创建一个Protocol$Adpative的动态代理类。

Protocol实例调用

在完成Protocol$Adpative的动态代理类在classloader中的创建后,我们看下这个代理类是如何完成具体接口调用的。例如:

    Exporter exporter = protocol.export(invoker);

这里看下上文Protocol$Adpative中的export函数,通过url参数中的getProtocol()方法,给extName变量赋值,extName默认等于dubbo。然后再次调用ExtensionLoader中的getExtension方法,返回具体的Protocol接口实现类。介绍下这里的extName变量,其实就是spi配置文件中的key。extName默认等于dubbo,那么Protocol接口默认的实现就是DubboProtocol。

原文地址:DUBBO源码分析 - 秦晓熙的博客 | Follow

你可能感兴趣的:(DUBBO源码分析(一)---内核SPI扩展实现)