一、Dubbo SPI原理

Dubbo采用微内核+插件体系,使得设计优雅,扩展性强。那所谓的微内核+插件体系是如何实现的呢!大家是否熟悉spi(service providerinterface)机制,即我们定义了服务接口标准,让厂商去实现(如果不了解spi的请谷歌百度下), jdk通过ServiceLoader类实现spi机制的服务查找功能 – Java 规范 SPI

Java的SPI机制

1、为什么不使用JDK SPI

在dubbo中它实现了一套自己的SPI机制。JDK标准的SPI会一次性实例化扩展点所有实现,如果有扩展实现初始化很耗时,但如果没用上也加载,会很浪费资源.

增加了对扩展点IoC和AOP的支持,一个扩展点可以直接setter注入其它扩展点。

2、Dubbo SPI 约定

SPI文件的存储路径在以下三个文件路径:

  • META-INF/dubbo/internal/ dubbo内部实现的各种扩展都放在了这个目录了
  • META-INF/dubbo/
  • META-INF/services/

spi 文件 存储路径在以上三个SPI目录下,并且文件名为接口的全路径名 就是=接口的包名+接口名也就是SPI接口的全路径类名:例如

  1. E:\Project\github\dubbo\dubbo-rpc\dubbo-rpc-default\src\main\resources
  2. \META-INF\dubbo\internal\
  3. com.alibaba.dubbo.rpc.Protocol

每个spi 文件里面的格式定义为: 扩展名=具体的类名,例如

  1. dubbo=com.alibaba.dubbo.rpc.protocol.dubbo.DubboProtoco

3、涉及到类与注解简单说明

在dubbo SPI中最关键的类是ExtensionLoader。每个定义的spi的接口都会构建一个ExtensionLoader实例,存储在ExtensionLoader对象的ConcurrentMap<Class<?>,ExtensionLoader<?>> EXTENSION_LOADERS这个map对象中。

获取SPI对象的典型方式为:

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

对于获取SPI对象的过程会在后面详细说明。

涉及到几个注解。下面我们就来简单的分析一下这些注解。

  • @SPI:标识在dubbo中需要使用SPI的接口上,指定的SPI里面指定的值为默认值。
  • @Adaptive:这个注解和@SPI注解配置使用,用于它可以标注在SPI接口扩展类上,也可以标注在SPI接口的方法上。如果这个注解标注在SPI接口实现的扩展类上时,获取的SPI实例对象就是标注了@Adaptive注册的类。例如:ExtensionFactory的SPI扩展对象为AdaptiveExtensionFactory。如果注解在标注在SPI接口的方法上说明就是一个动态代理类,它会通过dubbo里面的com.alibaba.dubbo.common.compiler.CompilerSPI接口通过字节码技术来创建对象。创建出来的对象名格式为SPI接口$Adaptive,例如Protocol接口创建的SPI对象为Protocol$Adaptive。
  • @Activate: 是一个 Duboo 框架提供的注解。在 Dubbo 官方文档上有记载:
    对于集合类扩展点,比如:Filter, InvokerListener, ExportListener, TelnetHandler, StatusChecker等, 可以同时加载多个实现,此时,可以用自动激活来简化配置。

4、ExtensionLoader

在上一个章节我们就说过对于每个定义的spi的接口都会构建一个ExtensionLoader实例,然后通过这个实例的getAdaptiveExtension就可以获取一个扩展。下面我们就来详细的解剖一下ExtensionLoader这个对象。

在这个对象里面有几个重要的方法:

  • getExtensionLoader(Class type) 就是为该接口new 一个-ExtensionLoader,然后缓存起来。
  • getAdaptiveExtension() 获取一个扩展类,如果@Adaptive注解在类上就是一个装饰类;如果注解在方法上就是一个动态代理类,例如Protocol$Adaptive对象。
  • getExtension(String name) 获取一个指定对象。
  • getActivateExtension(URL url, String[] values, String group):方法主要获取当前扩展的所有可自动激活的实现标注了@Activate注解
  1. Protocol protocol = ExtensionLoader.getExtensionLoader(Protocol.class).getAdaptiveExtension();

基于以上的dubbo SPI典型使用我们来分析一下这些方法。

4.0 Dubbo生产者Api直接调用方式的示例

  1. public class ProviderApi {
  2. public static void main(String[] args) throws IOException {
  3. ServiceConfig<UserService> config = new ServiceConfig<>();
  4. config.setApplication(new ApplicationConfig("api-provider"));
  5. config.setRegistry(new RegistryConfig("zookeeper://127.0.0.1:2181"));
  6. config.setInterface(UserService.class);
  7. config.setRef(new UserServiceImpl());
  8. config.export();
  9. System.out.println("first-dubbo-provider is running.");
  10. System.in.read();
  11. }
  12. }

上方生产使用的 ServiceConfig 类:

  1. public class ServiceConfig<T> extends AbstractServiceConfig {
  2. private static final Protocol protocol = ExtensionLoader.getExtensionLoader(Protocol.class).getAdaptiveExtension();
  3. private static final ProxyFactory proxyFactory = ExtensionLoader.getExtensionLoader(ProxyFactory.class).getAdaptiveExtension();
  4. .....
  5. }

4.1 getExtensionLoader

就是为该接口new 一个ExtensionLoader,然后缓存起来。并为非ExtensionFactory的对象创建一个objectFactory用来依赖注入。

com.alibaba.dubbo.common.extension.ExtensionLoader#getExtensionLoader

  1. public static <T> ExtensionLoader<T> getExtensionLoader(Class<T> type) {
  2. ExtensionLoader<T> loader = (ExtensionLoader<T>) EXTENSION_LOADERS.get(type);
  3. if (loader == null) {
  4. // 创建一个ExtensionLoader
  5. EXTENSION_LOADERS.putIfAbsent(type, new ExtensionLoader<T>(type));
  6. loader = (ExtensionLoader<T>) EXTENSION_LOADERS.get(type);
  7. }
  8. return loader;
  9. }

为SPI接口创建一个ExtensionLoader对象,用于获取扩展对象。

com.alibaba.dubbo.common.extension.ExtensionLoader#ExtensionLoader

  1. private ExtensionLoader(Class<?> type) {
  2. this.type = type;
  3. objectFactory = (type == ExtensionFactory.class ? null : ExtensionLoader.getExtensionLoader(ExtensionFactory.class).getAdaptiveExtension());
  4. }

在创建ExtensionLoader对象的时候,如果当前对象不是ExtensionFactory为当前SPI接口创建一个ExtensionFactory对象。当调用ExtensionLoader#injectExtension方法的时候进行依赖注入。

4.2 getAdaptiveExtension

这个方法是dubbo SPI里最核心的方法。dubbo通过这个方法来获取到SPI接口的对应扩展类。

com.alibaba.dubbo.common.extension.ExtensionLoader#getAdaptiveExtension

  1. public T getAdaptiveExtension() {
  2. Object instance = cachedAdaptiveInstance.get();
  3. if (instance == null) {
  4. if(createAdaptiveInstanceError == null) {
  5. synchronized (cachedAdaptiveInstance) {
  6. instance = cachedAdaptiveInstance.get();
  7. if (instance == null) {
  8. try {
  9. // 创建对应的扩展类
  10. instance = createAdaptiveExtension();
  11. cachedAdaptiveInstance.set(instance);
  12. }
  13. }
  14. }
  15. }
  16. }
  17. return (T) instance;
  18. }

当SPI接口首次调用这个方法的时候,扩展类还没有创建好,所以它就会直接访问createAdaptiveExtension方法。

com.alibaba.dubbo.common.extension.ExtensionLoader#createAdaptiveExtension

  1. private T createAdaptiveExtension() {
  2. try {
  3. return injectExtension((T) getAdaptiveExtensionClass().newInstance());
  4. } catch (Exception e) {
  5. throw new IllegalStateException("Can not create adaptive extenstion " + type + ", cause: " + e.getMessage(), e);
  6. }
  7. }

通过getAdaptiveExtensionClass获取到SPI扩展对象Class的实例,然后通过反射方法newInstance()创建这个对象。最后通过最开始介绍的ExtensionLoader#getExtensionLoader创建的objectFactory进行依赖注入。

com.alibaba.dubbo.common.extension.ExtensionLoader#getAdaptiveExtensionClass

  1. private Class<?> getAdaptiveExtensionClass() {
  2. // @Adaptive 注解在类上
  3. getExtensionClasses();
  4. if (cachedAdaptiveClass != null) {
  5. return cachedAdaptiveClass;
  6. }
  7. // @Adaptive注解在SPI接口方法上
  8. return cachedAdaptiveClass = createAdaptiveExtensionClass();
  9. }

这里就是上面说的,如果@Adaptive接口标注在@SPI接口的实现类上面就会直接返回这个对象的Class实例。如果标注在@SPI接口的方法上,就会通过dubbo中的字节码Compiler接口通过动态代理来创建SPI接口的实例。

4.2.1 @Adaptive在类上

下面我们就来分析一下@Adaptive标注在SPI接口的实现类上。SPI扩展的创建过程。

com.alibaba.dubbo.common.extension.ExtensionLoader#getExtensionClasses

  1. private Map<String, Class<?>> getExtensionClasses() {
  2. Map<String, Class<?>> classes = cachedClasses.get();
  3. if (classes == null) {
  4. synchronized (cachedClasses) {
  5. classes = cachedClasses.get();
  6. if (classes == null) {
  7. // 加载扩展类
  8. classes = loadExtensionClasses();
  9. cachedClasses.set(classes);
  10. }
  11. }
  12. }
  13. return classes;
  14. }

加载扩展类,并把扩展类Class实例设置到cachedClasses中。

com.alibaba.dubbo.common.extension.ExtensionLoader#loadExtensionClasses

  1. private Map<String, Class<?>> loadExtensionClasses() {
  2. final SPI defaultAnnotation = type.getAnnotation(SPI.class);
  3. if(defaultAnnotation != null) {
  4. String value = defaultAnnotation.value();
  5. if(value != null && (value = value.trim()).length() > 0) {
  6. String[] names = NAME_SEPARATOR.split(value);
  7. if(names.length > 1) {
  8. throw new IllegalStateException("more than 1 default extension name on extension " + type.getName()
  9. + ": " + Arrays.toString(names));
  10. }
  11. // 设置扩展类的默认名称
  12. if(names.length == 1) cachedDefaultName = names[0];
  13. }
  14. }
  15. Map<String, Class<?>> extensionClasses = new HashMap<String, Class<?>>();
  16. // 加载上面所说的三个配置文件中的dubbo SPI文件
  17. loadFile(extensionClasses, DUBBO_INTERNAL_DIRECTORY);
  18. loadFile(extensionClasses, DUBBO_DIRECTORY);
  19. loadFile(extensionClasses, SERVICES_DIRECTORY);
  20. return extensionClasses;
  21. }

从代码里面可以看到,在loadExtensionClasses中首先会检测扩展点在@SPI注解中配置的默认扩展实现的名称,并将其赋值给cachedDefaultName属性进行缓存,后面想要获取该扩展点的默认实现名称就可以直接通过访问cachedDefaultName字段来完成,比如getDefaultExtensionName方法就是这么实现的。从这里的代码中又可以看到,具体的扩展实现类型,是通过调用loadFile方法来加载,分别从一下三个地方加载:

  • META-INF/dubbo/internal/
  • META-INF/dubbo/
  • META-INF/services/

那么这个loadFile方法则至关重要了,看看其源代码: 【后面的源码改为了loadDirectory()】

  1. private void loadFile(Map<String, Class<?>> extensionClasses, String dir) {
  2. // SPI目录与SPI接口
  3. String fileName = dir + type.getName();
  4. try {
  5. Enumeration<java.net.URL> urls;
  6. ClassLoader classLoader = findClassLoader();
  7. // 扫描classpath下面的当前SPI接口的扩展
  8. if (classLoader != null) {
  9. urls = classLoader.getResources(fileName);
  10. } else {
  11. urls = ClassLoader.getSystemResources(fileName);
  12. }
  13. if (urls != null) {
  14. while (urls.hasMoreElements()) {
  15. java.net.URL url = urls.nextElement();
  16. try {
  17. BufferedReader reader = new BufferedReader(new InputStreamReader(url.openStream(), "utf-8"));
  18. try {
  19. String line = null;
  20. while ((line = reader.readLine()) != null) {
  21. final int ci = line.indexOf('#');
  22. // 读取SPI文件中的SPI扩展
  23. if (ci >= 0) line = line.substring(0, ci);
  24. line = line.trim();
  25. if (line.length() > 0) {
  26. try {
  27. String name = null;
  28. int i = line.indexOf('=');
  29. if (i > 0) {
  30. name = line.substring(0, i).trim();
  31. line = line.substring(i + 1).trim();
  32. }
  33. if (line.length() > 0) {
  34. Class<?> clazz = Class.forName(line, true, classLoader);
  35. if (! type.isAssignableFrom(clazz)) {
  36. throw new IllegalStateException("Error when load extension class(interface: " +
  37. type + ", class line: " + clazz.getName() + "), class "
  38. + clazz.getName() + "is not subtype of interface.");
  39. }
  40. // 如果Adaptive注解标注在这个对象上,设置cachedAdaptiveClass值为当前对象
  41. if (clazz.isAnnotationPresent(Adaptive.class)) {
  42. if(cachedAdaptiveClass == null) {
  43. cachedAdaptiveClass = clazz;
  44. } else if (! cachedAdaptiveClass.equals(clazz)) {
  45. throw new IllegalStateException("More than 1 adaptive class found: "
  46. + cachedAdaptiveClass.getClass().getName()
  47. + ", " + clazz.getClass().getName());
  48. }
  49. } else {
  50. try {
  51. // 判断这个SPI扩展是否以当前SPI接口为构造器,使用装饰器模式增强这个类
  52. clazz.getConstructor(type);
  53. Set<Class<?>> wrappers = cachedWrapperClasses;
  54. if (wrappers == null) {
  55. cachedWrapperClasses = new ConcurrentHashSet<Class<?>>();
  56. wrappers = cachedWrapperClasses;
  57. }
  58. wrappers.add(clazz);
  59. } catch (NoSuchMethodException e) {
  60. // 没有以当前SPI接口为构造器
  61. clazz.getConstructor();
  62. if (name == null || name.length() == 0) {
  63. name = findAnnotationName(clazz);
  64. if (name == null || name.length() == 0) {
  65. if (clazz.getSimpleName().length() > type.getSimpleName().length()
  66. && clazz.getSimpleName().endsWith(type.getSimpleName())) {
  67. name = clazz.getSimpleName().substring(0, clazz.getSimpleName().length() - type.getSimpleName().length()).toLowerCase();
  68. } else {
  69. throw new IllegalStateException("No such extension name for the class " + clazz.getName() + " in the config " + url);
  70. }
  71. }
  72. }
  73. String[] names = NAME_SEPARATOR.split(name);
  74. if (names != null && names.length > 0) {
  75. // 标注了Activate注解,保存在cachedActivates属性中
  76. Activate activate = clazz.getAnnotation(Activate.class);
  77. if (activate != null) {
  78. cachedActivates.put(names[0], activate);
  79. }
  80. for (String n : names) {
  81. // 把扩展名称与扩展类的映射关系保存在cachedNames中
  82. if (! cachedNames.containsKey(clazz)) {
  83. cachedNames.put(clazz, n);
  84. }
  85. Class<?> c = extensionClasses.get(n);
  86. if (c == null) {
  87. // 把扩展名称与扩展类的映射关系保存在extensionClasses中,用于返回
  88. extensionClasses.put(n, clazz);
  89. } else if (c != clazz) {
  90. throw new IllegalStateException("Duplicate extension " + type.getName() + " name " + n + " on " + c.getName() + " and " + clazz.getName());
  91. }
  92. }
  93. }
  94. }
  95. }
  96. }
  97. } catch (Throwable t) {
  98. IllegalStateException e = new IllegalStateException("Failed to load extension class(interface: " + type + ", class line: " + line + ") in " + url + ", cause: " + t.getMessage(), t);
  99. exceptions.put(line, e);
  100. }
  101. }
  102. } // end of while read lines
  103. } finally {
  104. reader.close();
  105. }
  106. } catch (Throwable t) {
  107. logger.error("Exception when load extension class(interface: " +
  108. type + ", class file: " + url + ") in " + url, t);
  109. }
  110. } // end of while urls
  111. }
  112. } catch (Throwable t) {
  113. logger.error("Exception when load extension class(interface: " +
  114. type + ", description file: " + fileName + ").", t);
  115. }
  116. }

这里就把@Adaptive的对象放置到cachedAdaptiveClass属性中,把有SPI接口为构造的包装对象放置在cachedWrapperClasses属性中,把没有SPI接口为构造的对象放置在cachedNames属性中。SPI接口作为构造包装对象如:ProtocolFilterWrapper

  1. public class ProtocolFilterWrapper implements Protocol {
  2. private final Protocol protocol;
  3. public ProtocolFilterWrapper(Protocol protocol) {
  4. if (protocol == null) {
  5. throw new IllegalArgumentException("protocol == null");
  6. }
  7. this.protocol = protocol;
  8. }
  9. ....
  10. }

4.2.2 @Adaptive在方法上

如果@Adaptive标注在SPI接口的方法上,那么dubbo就会通过SPI接口Compiler进行字节码操作生成代理对象。默认使用Javassist字节码框架生成代理对象。

com.alibaba.dubbo.common.extension.ExtensionLoader#createAdaptiveExtensionClass

  1. private Class<?> createAdaptiveExtensionClass() {
  2. String code = createAdaptiveExtensionClassCode();
  3. ClassLoader classLoader = findClassLoader();
  4. com.alibaba.dubbo.common.compiler.Compiler compiler = ExtensionLoader.getExtensionLoader(com.alibaba.dubbo.common.compiler.Compiler.class).getAdaptiveExtension();
  5. return compiler.compile(code, classLoader);
  6. }

生成的代码对象的模板如下:

  1. package <扩展点接口所在包>;
  2. public class <扩展点接口名>$Adpative implements <扩展点接口> {
  3. public <有@Adaptive注解的接口方法>(<方法参数>) {
  4. if(是否有URL类型方法参数?) 使用该URL参数
  5. else if(是否有方法类型上有URL属性) 使用该URL属性
  6. # <else 在加载扩展点生成自适应扩展点类时抛异常,即加载扩展点失败!>
  7. if(获取的URL == null) {
  8. throw new IllegalArgumentException("url == null");
  9. }
  10. 根据@Adaptive注解上声明的Key的顺序,从URL获致Value,作为实际扩展点名。
  11. URL没有Value,则使用缺省扩展点实现。如没有扩展点, throw new IllegalStateException("Fail to get extension");
  12. 在扩展点实现调用该方法,并返回结果。
  13. }
  14. public <有@Adaptive注解的接口方法>(<方法参数>) {
  15. throw new UnsupportedOperationException("is not adaptive method!");
  16. }
  17. }

例如Protocol生成的代理对象Protocol$Adpative如下:

  1. package com.alibaba.dubbo.rpc;
  2. import com.alibaba.dubbo.common.extension.ExtensionLoader;
  3. public class Protocol$Adpative implements com.alibaba.dubbo.rpc.Protocol {
  4. public void destroy() {
  5. throw new UnsupportedOperationException("method public abstract void com.alibaba.dubbo.rpc.Protocol.destroy() of interface com.alibaba.dubbo.rpc.Protocol is not adaptive method!");
  6. }
  7. public int getDefaultPort() {
  8. throw new UnsupportedOperationException("method public abstract int com.alibaba.dubbo.rpc.Protocol.getDefaultPort() of interface com.alibaba.dubbo.rpc.Protocol is not adaptive method!");
  9. }
  10. public com.alibaba.dubbo.rpc.Exporter export(com.alibaba.dubbo.rpc.Invoker arg0) throws com.alibaba.dubbo.rpc.RpcException {
  11. if (arg0 == null) throw new IllegalArgumentException("com.alibaba.dubbo.rpc.Invoker argument == null");
  12. if (arg0.getUrl() == null)
  13. throw new IllegalArgumentException("com.alibaba.dubbo.rpc.Invoker argument getUrl() == null");
  14. com.alibaba.dubbo.common.URL url = arg0.getUrl();
  15. String extName = (url.getProtocol() == null ? "dubbo" : url.getProtocol());
  16. if (extName == null)
  17. throw new IllegalStateException("Fail to get extension(com.alibaba.dubbo.rpc.Protocol) name from url(" + url.toString() + ") use keys([protocol])");
  18. com.alibaba.dubbo.rpc.Protocol extension = (com.alibaba.dubbo.rpc.Protocol) ExtensionLoader.getExtensionLoader(com.alibaba.dubbo.rpc.Protocol.class).getExtension(extName);
  19. return extension.export(arg0);
  20. }
  21. public com.alibaba.dubbo.rpc.Invoker refer(java.lang.Class arg0, com.alibaba.dubbo.common.URL arg1) throws com.alibaba.dubbo.rpc.RpcException {
  22. if (arg1 == null) throw new IllegalArgumentException("url == null");
  23. com.alibaba.dubbo.common.URL url = arg1;
  24. String extName = (url.getProtocol() == null ? "dubbo" : url.getProtocol());
  25. if (extName == null)
  26. throw new IllegalStateException("Fail to get extension(com.alibaba.dubbo.rpc.Protocol) name from url(" + url.toString() + ") use keys([protocol])");
  27. com.alibaba.dubbo.rpc.Protocol extension = (com.alibaba.dubbo.rpc.Protocol) ExtensionLoader.getExtensionLoader(com.alibaba.dubbo.rpc.Protocol.class).getExtension(extName);
  28. return extension.refer(arg0, arg1);
  29. }
  30. }

创建自适应扩展点实现类型和实例化就已经完成了,下面就来看下扩展点自动注入的实现injectExtension:

com.alibaba.dubbo.common.extension.ExtensionLoader#injectExtension

  1. private T injectExtension(T instance) {
  2. try {
  3. if (objectFactory != null) {
  4. for (Method method : instance.getClass().getMethods()) {
  5. // 处理所有set方法
  6. if (method.getName().startsWith("set")
  7. && method.getParameterTypes().length == 1
  8. && Modifier.isPublic(method.getModifiers())) {
  9. // 获取set方法参数类型
  10. Class<?> pt = method.getParameterTypes()[0];
  11. try {
  12. // 获取setter对应的property名称
  13. String property = method.getName().length() > 3 ? method.getName().substring(3, 4).toLowerCase() + method.getName().substring(4) : "";
  14. // 根据类型,名称信息从ExtensionFactory获取
  15. Object object = objectFactory.getExtension(pt, property);
  16. if (object != null) {
  17. // 如果不为空,说set方法的参数是扩展点类型,那么进行注入
  18. method.invoke(instance, object);
  19. }
  20. } catch (Exception e) {
  21. logger.error("fail to inject via method " + method.getName()
  22. + " of interface " + type.getName() + ": " + e.getMessage(), e);
  23. }
  24. }
  25. }
  26. }
  27. } catch (Exception e) {
  28. logger.error(e.getMessage(), e);
  29. }
  30. return instance;
  31. }

这里可以看到,扩展点自动注入的一句就是根据setter方法对应的参数类型和property名称从ExtensionFactory中查询,如果有返回扩展点实例,那么就进行注入操作。到这里getAdaptiveExtension方法就分析完毕了。

4.3 getExtension

这个方法的主要作用是用来获取ExtensionLoader实例代表的扩展的指定实现。已扩展实现的名字作为参数,结合前面学习getAdaptiveExtension的代码,我们可以推测,这方法中也使用了在调用getExtensionClasses方法的时候收集并缓存的数据,其中涉及到名字和具体实现类型对应关系的缓存属性是cachedClasses。具体是是否如我们猜想的那样呢,学习一下相关代码就知道了:

  1. public T getExtension(String name) {
  2. if (name == null || name.length() == 0)
  3. throw new IllegalArgumentException("Extension name == null");
  4. // 判断是否是获取默认实现
  5. if ("true".equals(name)) {
  6. return getDefaultExtension();
  7. }
  8. Holder<Object> holder = cachedInstances.get(name);// 缓存
  9. if (holder == null) {
  10. cachedInstances.putIfAbsent(name, new Holder<Object>());
  11. holder = cachedInstances.get(name);
  12. }
  13. Object instance = holder.get();
  14. if (instance == null) {
  15. synchronized (holder) {
  16. instance = holder.get();
  17. if (instance == null) {
  18. // 没有缓存实例则创建
  19. instance = createExtension(name);
  20. holder.set(instance);// 缓存起来
  21. }
  22. }
  23. }
  24. return (T) instance;
  25. }

判断是使用默认的扩展 还是根据名称来创建相应的SPI扩展。

com.alibaba.dubbo.common.extension.ExtensionLoader#createExtension

  1. private T createExtension(String name) {
  2. // 获取解析Adaptive标注了@Adaptive的对象缓存在属性cachedNames
  3. Class<?> clazz = getExtensionClasses().get(name);
  4. if (clazz == null) {
  5. throw findException(name);
  6. }
  7. try {
  8. // 从已创建Extension实例缓存中获取
  9. T instance = (T) EXTENSION_INSTANCES.get(clazz);
  10. if (instance == null) {
  11. EXTENSION_INSTANCES.putIfAbsent(clazz, (T) clazz.newInstance());
  12. instance = (T) EXTENSION_INSTANCES.get(clazz);
  13. }
  14. // 注入AdaptiveExtension
  15. injectExtension(instance);
  16. Set<Class<?>> wrapperClasses = cachedWrapperClasses;
  17. // Wrapper类型进行包装,层层包裹
  18. if (wrapperClasses != null && wrapperClasses.size() > 0) {
  19. for (Class<?> wrapperClass : wrapperClasses) {
  20. instance = injectExtension((T) wrapperClass.getConstructor(type).newInstance(instance));
  21. }
  22. }
  23. return instance;
  24. } catch (Throwable t) {
  25. throw new IllegalStateException("Extension instance(name: " + name + ", class: " +
  26. type + ") could not be instantiated: " + t.getMessage(), t);
  27. }
  28. }

这里或许有一个疑问: 从代码中看,不论instance是否存在于EXTENSION_INSTANCE,都会进行扩展点注入和Wrap操作。那么如果对于同一个扩展点,调用了两次createExtension方法的话,那不就进行了两次Wrap操作么?

如果外部能够直接调用createExtension方法,那么确实可能出现这个问题。但是由于createExtension方法是private的,因此外部无法直接调用。而在ExtensionLoader类中调用它的getExtension方法(只有它这一处调用),内部自己做了缓存(cachedInstances),因此当getExtension方法内部调用了一次createExtension方法之后,后面对getExtension方法执行同样的调用时,会直接使用cachedInstances缓存而不会再去调用createExtension方法了。

4.4 getActivateExtension

getActivateExtension方法主要获取当前扩展的所有可自动激活的实现。可根据入参(values)调整指定实现的顺序,在这个方法里面也使用到getExtensionClasses方法中收集的缓存数据。

com.alibaba.dubbo.common.extension.ExtensionLoader#getActivateExtension(com.alibaba.dubbo.common.URL, java.lang.String[], java.lang.String)

  1. public List<T> getActivateExtension(URL url, String[] values, String group) {
  2. List<T> exts = new ArrayList<T>();
  3. List<String> names = values == null ? new ArrayList<String>(0) : Arrays.asList(values);
  4. // 如果未配置"-default",则加载所有Activates扩展(names指定的扩展)
  5. if (! names.contains(Constants.REMOVE_VALUE_PREFIX + Constants.DEFAULT_KEY)) {
  6. // 加载当前Extension所有实现,会获取到当前Extension中所有@Active实现,赋值给cachedActivates变量
  7. getExtensionClasses();
  8. // 遍历当前扩展所有的@Activate扩展
  9. for (Map.Entry<String, Activate> entry : cachedActivates.entrySet()) {
  10. String name = entry.getKey();
  11. Activate activate = entry.getValue();
  12. // 判断group是否满足,group为null则直接返回true
  13. if (isMatchGroup(group, activate.group())) {
  14. // 获取扩展示例
  15. T ext = getExtension(name);
  16. // 排除names指定的扩展;并且如果names中没有指定移除该扩展(-name),且当前url匹配结果显示可激活才进行使用
  17. if (! names.contains(name)
  18. && ! names.contains(Constants.REMOVE_VALUE_PREFIX + name)
  19. && isActive(activate, url)) {
  20. exts.add(ext);
  21. }
  22. }
  23. }
  24. Collections.sort(exts, ActivateComparator.COMPARATOR); // 默认排序
  25. }
  26. // 对names指定的扩展进行专门的处理
  27. List<T> usrs = new ArrayList<T>();
  28. // 遍历names指定的扩展名
  29. for (int i = 0; i < names.size(); i ++) {
  30. String name = names.get(i);
  31. if (! name.startsWith(Constants.REMOVE_VALUE_PREFIX)
  32. // 未设置移除该扩展
  33. && ! names.contains(Constants.REMOVE_VALUE_PREFIX + name)) {
  34. if (Constants.DEFAULT_KEY.equals(name)) {
  35. // default表示上面已经加载并且排序的exts,将排在default之前的Activate扩展放置到default组之前,例如:ext1,default,ext2
  36. if (usrs.size() > 0) { // 如果此时user不为空,则user中存放的是配置在default之前的Activate扩展
  37. exts.addAll(0, usrs); // 注意index是0,放在default前面
  38. usrs.clear(); // 放到default之前,然后清空
  39. }
  40. } else {
  41. T ext = getExtension(name);
  42. usrs.add(ext);
  43. }
  44. }
  45. }
  46. if (usrs.size() > 0) { // 这里留下的都是配置在default之后的
  47. exts.addAll(usrs); // 添加到default排序之后
  48. }
  49. return exts;
  50. }

@Activate注解示例

  1. @Activate(group = "provider", value = Constants.ACCESS_LOG_KEY)
  2. public class AccessLogFilter implements Filter {
  3. @Override
  4. public Result invoke(Invoker<?> invoker, Invocation inv) throws RpcException {
  5. ...
  6. }
  7. }

5、总结

上面把SPI的思路以及源代码分析了一遍,有几点可能需要注意的地方:

  • 每个ExtensionLoader实例只负责加载一个特定扩展点实现
  • 每个扩展点对应最多只有一个ExtensionLoader实例
  • 对于每个扩展点实现,最多只会有一个实例
  • 一个扩展点实现可以对应多个名称(逗号分隔)
  • 对于需要等到运行时才能决定使用哪一个具体实现的扩展点,应获取其自使用扩展点实现(AdaptiveExtension)
  • @Adaptive注解要么注释在扩展点@SPI的方法上,要么注释在其实现类的类定义上
  • 如果@Adaptive注解注释在@SPI接口的方法上,那么原则上该接口所- - 有方法都应该加@Adaptive注解(自动生成的实现中默认为注解的方法抛异常)
  • 每个扩展点最多只能有一个被AdaptiveExtension
  • 每个扩展点可以有多个可自动激活的扩展点实现(使用@Activate注解)
  • 由于每个扩展点实现最多只有一个实例,因此扩展点实现应保证线程安全
  • 如果扩展点有多个Wrapper,那么最终其执行的顺序不确定(内部使用ConcurrentHashSet存储)

参考文章:
1、Dubbo原理解析-Dubbo内核实现之基于SPI思想Dubbo内核实现
2、Spring Boot 中如何使用 Dubbo Activate 扩展点
3、Dubbo扩展点加载机制 - ExtensionLoader

二、Dubbo SPI Protocol 应用实例

Protocol 属于 dubbo 十层结构 中的远程调用层, 它封装了RPC调用。以Invocation 和 Result 为中心, 其它的扩展接口还包括 Invoker 和 Exporter。 Protocol 是服务域, 它是 Invoker 暴露和引用的主功能入口, 它负责 Invoker 的生命周期管理。(以下内容可以结构Dubbo源码二、三一起阅读)

Dubbo框架设计 (官网)

Invoker是实体域, 它是Dubbo的核心模型, 其它模型都向它靠扰, 或转换成它. 它代表一个可执行体,可向它发起invoke调用,它有可能是一个本地的实现,也可能是一个远程的实现,也可能一个集群实现。

PS: 大家可以把 dubbo 的日志级别设置为 debug 级别。这样 dubbo 由 Javassist 字节码框架生成代理对象就会打印到控制台。大家可以把创建这个类, 然后就可以 debug 这个类了。

在服务端配置类中 ServiceConfig 中我们可以看到它聚合了 Protocol。 而且是通过 dubbo 自己实现的 SPI 机制获取的。当我们启动 dubbo 源码中的 dubbo-demo 中的 com.alibaba.dubbo.demo.provider.Provider 类的时候就会进行服务发布。我们可以在控制台中看到生成的 Protocol 接口的动态代理类 Protocol$Adaptive:

Dubbo源码(一) 内核SPI实现 - 图1

我们只需要把这个类按生成的规则, 也就是包名与类名一致的放在 dubbo 源码中就可以 debug 这个类了。

Dubbo源码(一) 内核SPI实现 - 图2

在之前讲 dubbo 的 SPI 机制中, 我们知道 dubbo 在通过 ExtensionLoader#createExtension 方法来创建相应的 SPI 的扩展。在这个方法中 dubbo SPI 实现了自己的 IOC 与 AOP。

  1. private T createExtension(String name) {
  2. // 获取解析Adaptive标注了@Adaptive的对象缓存在属性cachedNames
  3. Class<?> clazz = getExtensionClasses().get(name);
  4. if (clazz == null) {
  5. throw findException(name);
  6. }
  7. try {
  8. // 从已创建Extension实例缓存中获取
  9. T instance = (T) EXTENSION_INSTANCES.get(clazz);
  10. if (instance == null) {
  11. EXTENSION_INSTANCES.putIfAbsent(clazz, (T) clazz.newInstance());
  12. instance = (T) EXTENSION_INSTANCES.get(clazz);
  13. }
  14. // dubbo ioc
  15. injectExtension(instance);
  16. Set<Class<?>> wrapperClasses = cachedWrapperClasses;
  17. // dubbo aop
  18. if (wrapperClasses != null && wrapperClasses.size() > 0) {
  19. for (Class<?> wrapperClass : wrapperClasses) {
  20. instance = injectExtension((T) wrapperClass.getConstructor(type).newInstance(instance));
  21. }
  22. }
  23. return instance;
  24. } catch (Throwable t) {
  25. throw new IllegalStateException("Extension instance(name: " + name + ", class: " +
  26. type + ") could not be instantiated: " + t.getMessage(), t);
  27. }
  28. }

如ProtocolFilterWrapper类:构造方法中包含Protocol这个类,是包装类
image.png

injectExtension 方法中是 dubbo 实现自己的依赖注入。 dubbo 通过 ExtensionFactory 来获取需要依赖注入的对象。然后对当前实例通过反射调用 setter 方法来进行依赖注入。

当dubbo进行服务暴露的时候。首先是通过 Protocol 的 SPI 接口 与 获取到 RegistryProtocol 实例。然后通过 dubbo 的 aop把这个对象先代理到 ProtocolListenerWrapper, 然后再将代理后的对象代理到 ProtocolFilterWrapper中。

1、调用SPI生成的代理对象

调用SPI生成的代理对象, 对象 dubbo 的SPI 机制通过 ExtensionLoader#createExtension 获取到 RegistryProtocol实例。

Dubbo源码(一) 内核SPI实现 - 图4

2、调用SPI的AOP生成最终对象

获取到需要代理的对象,这里的 cachedWrapperClasses 包装对象是在解析 dubbo SPI 扩展文件生成的。 这个对象的满足的条件是实现了 SPI 接口, 并且有只有 SPI 接口为构造器, 且配置在 SPI 扩展文件的对象。我们可以看到针对 Protocol 接口,以下2个类满足条件:

  • ProtocolListenerWrapper
  • ProtocolFilterWrapper

Dubbo源码(一) 内核SPI实现 - 图5

3、它的执行过程如下

  1. 通过dubbo的SPI获取 RegistryProtocol 实例。
  2. 然后通过调用ProtocolListenerWrapper的 ProtocolListenerWrapper(Protocol protocol)生成代理对象, 然后再将代理后的对象传入到ProtocolFilterWrapper的ProtocolFilterWrapper(Protocol protocol)中生成代理对象。
  3. 生成的最终对象返回到我们最开始说的ServiceConfig这个类中。