Dubbo源码笔记-服务注册

今天来简单做一下Dubbo服务注册部分源码学习手记。

一、Dubbo配置解析

目前Dubbo最多的用法就是跟Spring集成，既然跟Spring集成，那么，Dubbo对象的实例化都将交由Spring统一处理。而Dubbo配置，对于Spring来说其实就是自定标签。这里Dubbo自定义标签解析类，在Dubbo配置模块（\dubbo-config\dubbo-config-spring\src\main\resources\META-INF/spring.handlers）进行了声明：

http\://dubbo.apache.org/schema/dubbo=org.apache.dubbo.config.spring.schema.DubboNamespaceHandler
http\://code.alibabatech.com/schema/dubbo=org.apache.dubbo.config.spring.schema.DubboNamespaceHandler

 public class DubboNamespaceHandler extends NamespaceHandlerSupport {
     static {
         Version.checkDuplicate(DubboNamespaceHandler.class);
     }
     @Override
     public void init() {
         registerBeanDefinitionParser("application", new DubboBeanDefinitionParser(ApplicationConfig.class, true));
         registerBeanDefinitionParser("module", new DubboBeanDefinitionParser(ModuleConfig.class, true));
         registerBeanDefinitionParser("registry", new DubboBeanDefinitionParser(RegistryConfig.class, true));
         registerBeanDefinitionParser("config-center", new DubboBeanDefinitionParser(ConfigCenterBean.class, true));
         registerBeanDefinitionParser("monitor", new DubboBeanDefinitionParser(MonitorConfig.class, true));
         registerBeanDefinitionParser("provider", new DubboBeanDefinitionParser(ProviderConfig.class, true));
         registerBeanDefinitionParser("consumer", new DubboBeanDefinitionParser(ConsumerConfig.class, true));
         registerBeanDefinitionParser("protocol", new DubboBeanDefinitionParser(ProtocolConfig.class, true));
         registerBeanDefinitionParser("service", new DubboBeanDefinitionParser(ServiceBean.class, true));
         registerBeanDefinitionParser("reference", new DubboBeanDefinitionParser(ReferenceBean.class, false));
         registerBeanDefinitionParser("annotation", new AnnotationBeanDefinitionParser());
     }
 }

对应一下Dubbo服务提供方的配置文件：

<dubbo:application name="demo-provider"/>
<dubbo:registry address="zookeeper://127.0.0.1:2181" />
<dubbo:protocol name="dubbo"/>
<bean id="demoService" class="org.apache.dubbo.demo.provider.DemoServiceImpl"/>
<dubbo:service interface="org.apache.dubbo.demo.DemoService" ref="demoService"/>

就是将这些标签解析成ApplicationConfig...ServiceBean等等对象。而服务注册部分最主要的桥梁就在于ServiceBean初始化解析完毕的时候。有Spring容器调用ServiceBean.afterPropertiesSet方法：

 public void afterPropertiesSet() throws Exception {
     // 此处省略配置检查
     export();
 }    

二、配置检查与缺省填充

进入export方法中，首先会对ServiceBean进行发布前的配置检查与填充。

 public synchronized void export() {
         checkAndUpdateSubConfigs(); // 加载并更新配置信息到Bean对象中，并检查
 }

这里会有一系列的配置校验与配置填充的逻辑：

 public void checkAndUpdateSubConfigs() {
     checkDefault();
     if (provider != null) {
         inheritIfAbsentFromProvider();
     }
     if (module != null) {
         inheritIfAbsentFromModule();
     }
     if (application != null) {
         inheritIfAbsentFromApplication();
     }
     checkApplication();
     checkRegistry();
     checkProtocol();
     this.refresh();
     checkMetadataReport();
     checkRegistryDataConfig();
     try {
         interfaceClass = Class.forName(interfaceName, true, Thread.currentThread().getContextClassLoader());
     } catch (ClassNotFoundException e) {
         throw new IllegalStateException(e.getMessage(), e);
     }
     checkInterfaceAndMethods(interfaceClass, methods);
     checkRef();
     checkStubAndLocal(interfaceClass);
     checkMock(interfaceClass);
 }

这里只列了部分配置检查与填充。总体配置的区分是否配置中心优先，优先级为：

 public void refresh() {
     // getPrefix为对应配置类的前缀，ProviderConfig->Provider, ServiceBean->Service,
     // getId为beanId,
     CompositeConfiguration compositeConfiguration = Environment.getInstance().getConfiguration(getPrefix(), getId());
     InmemoryConfiguration config = new InmemoryConfiguration(getPrefix(), getId());
     config.addProperties(getMetaData()); // Bean对象的初始值
     if (Environment.getInstance().isConfigCenterFirst()) {
         // -D系统属性 > 配置中心应用配置 > 配置中心全局配置 > Bean对象的初始值 > 属性文件中的信息
         compositeConfiguration.addConfiguration(3, config);
     } else {
         // -D > Bean对象的初始值 > 配置中心应用配置 > 配置中心全局配置 > 属性文件中的信息
         compositeConfiguration.addConfiguration(1, config);
     }
 }

三、获取和构建注册中心统一资源定位器

 private void doExportUrls() {
     // 获取注册中心统一资源列表，如果注册中心没有初始化，则先初始化
     List<URL> registryURLs = loadRegistries(true);
     // 将目标服务所有协议模式注册到每一个注册中心上
     for (ProtocolConfig protocolConfig : protocols) {
         doExportUrlsFor1Protocol(protocolConfig, registryURLs);
     }
 }

四、服务启动与服务注册

在doExportUrlsForProtocal方法里边其实就做了以下几件事：

 private void doExportUrlsFor1Protocol(ProtocolConfig protocolConfig, List<URL> registryURLs) {
      Map map = builderUrl();
      // 代表一个服务
      URL url = new URL(name, host, port, (StringUtils.isEmpty(contextPath) ? "" : contextPath + "/") + path, map);
      // 通过代理工厂将ref对象转化成invoker对象
      Invoker<?> invoker = proxyFactory.getInvoker(ref, (Class) interfaceClass, url);
      //代理invoker对象
      DelegateProviderMetaDataInvoker wrapperInvoker = new DelegateProviderMetaDataInvoker(invoker, this);
      // 使用配置协议启动、暴露服务
      Exporter<?> exporter = protocol.export(wrapperInvoker);
      //一个服务可能有多个提供者，保存在一起
      exporters.add(exporter);
 }

这里，我们着重看第10行服务暴露部分，这里调用export方法，如果是Dubbo协议的话，分别列举一下DubboProtocal、HttpProtocol的export：

 public class DubboProtocol extends AbstractProtocol {
     public <T> Exporter<T> export(Invoker<T> invoker) throws RpcException {
         URL url = invoker.getUrl();
         //忽略若干代码
         //打开服务
         openServer(url);
         optimizeSerialization(url);
         return exporter;
     }
 }

 public class HttpProtocol extends AbstractProxyProtocol {
     protected <T> Runnable doExport(final T impl, Class<T> type, URL url) throws RpcException {
         String addr = getAddr(url);
         HttpServer server = serverMap.get(addr);
         server = httpBinder.bind(url, new InternalHandler());
     }
 }

其实这块就是，根据不同协议分别做了服务启动，如果是Dubbo协议则启动NettyServer，如果是Http协议则启动一个Tomcat。

那么现在服务启动了，但是在哪里注册了呢？其实我们要关注这个protocol对象的构建：

 /**
  * The {@link Protocol} implementation with adaptive functionality,it will be different in different scenarios.
  * A particular {@link Protocol} implementation is determined by the protocol attribute in the {@link URL}.
  * For example:
  *
  * <li>when the url is registry://224.5.6.7:1234/org.apache.dubbo.registry.RegistryService?application=dubbo-sample,
  * then the protocol is <b>RegistryProtocol</b></li>
  *
  * <li>when the url is dubbo://224.5.6.7:1234/org.apache.dubbo.config.api.DemoService?application=dubbo-sample, then
  * the protocol is <b>DubboProtocol</b></li>
  * <p>
  * Actually，when the {@link ExtensionLoader} init the {@link Protocol} instants,it will automatically wraps two
  * layers, and eventually will get a <b>ProtocolFilterWrapper</b> or <b>ProtocolListenerWrapper</b>
  */
 private static final Protocol protocol = ExtensionLoader.getExtensionLoader(Protocol.class).getAdaptiveExtension();

这段源码的意思是，在获取Protocol实例的时候，Dubbo框架自动给包装了两个切面，其实服务注册就是在这个切面里边完成的：

 public class ProtocolListenerWrapper implements Protocol {
     private final Protocol protocol;
     @Override
     public <T> Exporter<T> export(Invoker<T> invoker) throws RpcException {
         //如果是registerProtocol,则调用RegisterProtocol.export方法
         if (Constants.REGISTRY_PROTOCOL.equals(invoker.getUrl().getProtocol())) {
             return protocol.export(invoker);
         }
         return new ListenerExporterWrapper<T>(protocol.export(invoker),
                 Collections.unmodifiableList(ExtensionLoader.getExtensionLoader(ExporterListener.class)
                         .getActivateExtension(invoker.getUrl(), Constants.EXPORTER_LISTENER_KEY)));
     }
 }

再看看RegisterProtocol的export：

 public class RegistryProtocol implements Protocol {
     @Override
     public <T> Exporter<T> export(final Invoker<T> originInvoker) throws RpcException {
         register(registryUrl, registeredProviderUrl); // 服务注册
     }
 }

如果注册中心是ZK的话其实就是给ZK写数据：

 @Override
 public void register(URL url) {
          //忽略很多代码
         doRegister(url);
         //忽略很多代码
 }
 protected void doRegister(URL url) {
     try {
         zkClient.create(toUrlPath(url), url.getParameter(Constants.DYNAMIC_KEY, true));
     } catch (Throwable e) {
         throw new RpcException("Failed to register " + url + " to zookeeper " + getUrl() + ", cause: " + e.getMessage(), e);
     }
 }

相关文章：https://www.jianshu.com/p/7f3871492c71