3.2 引用服务

本节我们要从 createProxy 开始看起。从字面意思上来看,createProxy 似乎只是用于创建代理对象的。但实际上并非如此,该方法还会调用其他方法构建以及合并 Invoker 实例。具体细节如下。

  1. private T createProxy(Map<String, String> map) {
  2. URL tmpUrl = new URL("temp", "localhost", 0, map);
  3. final boolean isJvmRefer;
  4. if (isInjvm() == null) {
  5. // url 配置被指定,则不做本地引用
  6. if (url != null && url.length() > 0) {
  7. isJvmRefer = false;
  8. // 根据 url 的协议、scope 以及 injvm 等参数检测是否需要本地引用
  9. // 比如如果用户显式配置了 scope=local,此时 isInjvmRefer 返回 true
  10. } else if (InjvmProtocol.getInjvmProtocol().isInjvmRefer(tmpUrl)) {
  11. isJvmRefer = true;
  12. } else {
  13. isJvmRefer = false;
  14. }
  15. } else {
  16. // 获取 injvm 配置值
  17. isJvmRefer = isInjvm().booleanValue();
  18. }
  19. // 本地引用
  20. if (isJvmRefer) {
  21. // 生成本地引用 URL,协议为 injvm
  22. URL url = new URL(Constants.LOCAL_PROTOCOL, NetUtils.LOCALHOST, 0, interfaceClass.getName()).addParameters(map);
  23. // 调用 refer 方法构建 InjvmInvoker 实例
  24. invoker = refprotocol.refer(interfaceClass, url);
  25. // 远程引用
  26. } else {
  27. // url 不为空,表明用户可能想进行点对点调用
  28. if (url != null && url.length() > 0) {
  29. // 当需要配置多个 url 时,可用分号进行分割,这里会进行切分
  30. String[] us = Constants.SEMICOLON_SPLIT_PATTERN.split(url);
  31. if (us != null && us.length > 0) {
  32. for (String u : us) {
  33. URL url = URL.valueOf(u);
  34. if (url.getPath() == null || url.getPath().length() == 0) {
  35. // 设置接口全限定名为 url 路径
  36. url = url.setPath(interfaceName);
  37. }
  38. // 检测 url 协议是否为 registry,若是,表明用户想使用指定的注册中心
  39. if (Constants.REGISTRY_PROTOCOL.equals(url.getProtocol())) {
  40. // 将 map 转换为查询字符串,并作为 refer 参数的值添加到 url 中
  41. urls.add(url.addParameterAndEncoded(Constants.REFER_KEY, StringUtils.toQueryString(map)));
  42. } else {
  43. // 合并 url,移除服务提供者的一些配置(这些配置来源于用户配置的 url 属性),
  44. // 比如线程池相关配置。并保留服务提供者的部分配置,比如版本,group,时间戳等
  45. // 最后将合并后的配置设置为 url 查询字符串中。
  46. urls.add(ClusterUtils.mergeUrl(url, map));
  47. }
  48. }
  49. }
  50. } else {
  51. // 加载注册中心 url
  52. List<URL> us = loadRegistries(false);
  53. if (us != null && !us.isEmpty()) {
  54. for (URL u : us) {
  55. URL monitorUrl = loadMonitor(u);
  56. if (monitorUrl != null) {
  57. map.put(Constants.MONITOR_KEY, URL.encode(monitorUrl.toFullString()));
  58. }
  59. // 添加 refer 参数到 url 中,并将 url 添加到 urls 中
  60. urls.add(u.addParameterAndEncoded(Constants.REFER_KEY, StringUtils.toQueryString(map)));
  61. }
  62. }
  63. // 未配置注册中心,抛出异常
  64. if (urls.isEmpty()) {
  65. throw new IllegalStateException("No such any registry to reference...");
  66. }
  67. }
  68. // 单个注册中心或服务提供者(服务直连,下同)
  69. if (urls.size() == 1) {
  70. // 调用 RegistryProtocol 的 refer 构建 Invoker 实例
  71. invoker = refprotocol.refer(interfaceClass, urls.get(0));
  72. // 多个注册中心或多个服务提供者,或者两者混合
  73. } else {
  74. List<Invoker<?>> invokers = new ArrayList<Invoker<?>>();
  75. URL registryURL = null;
  76. // 获取所有的 Invoker
  77. for (URL url : urls) {
  78. // 通过 refprotocol 调用 refer 构建 Invoker,refprotocol 会在运行时
  79. // 根据 url 协议头加载指定的 Protocol 实例,并调用实例的 refer 方法
  80. invokers.add(refprotocol.refer(interfaceClass, url));
  81. if (Constants.REGISTRY_PROTOCOL.equals(url.getProtocol())) {
  82. registryURL = url;
  83. }
  84. }
  85. if (registryURL != null) {
  86. // 如果注册中心链接不为空,则将使用 AvailableCluster
  87. URL u = registryURL.addParameter(Constants.CLUSTER_KEY, AvailableCluster.NAME);
  88. // 创建 StaticDirectory 实例,并由 Cluster 对多个 Invoker 进行合并
  89. invoker = cluster.join(new StaticDirectory(u, invokers));
  90. } else {
  91. invoker = cluster.join(new StaticDirectory(invokers));
  92. }
  93. }
  94. }
  95. Boolean c = check;
  96. if (c == null && consumer != null) {
  97. c = consumer.isCheck();
  98. }
  99. if (c == null) {
  100. c = true;
  101. }
  102. // invoker 可用性检查
  103. if (c && !invoker.isAvailable()) {
  104. throw new IllegalStateException("No provider available for the service...");
  105. }
  106. // 生成代理类
  107. return (T) proxyFactory.getProxy(invoker);
  108. }

上面代码很多,不过逻辑比较清晰。首先根据配置检查是否为本地调用,若是,则调用 InjvmProtocol 的 refer 方法生成 InjvmInvoker 实例。若不是,则读取直连配置项,或注册中心 url,并将读取到的 url 存储到 urls 中。然后根据 urls 元素数量进行后续操作。若 urls 元素数量为1,则直接通过 Protocol 自适应拓展类构建 Invoker 实例接口。若 urls 元素数量大于1,即存在多个注册中心或服务直连 url,此时先根据 url 构建 Invoker。然后再通过 Cluster 合并多个 Invoker,最后调用 ProxyFactory 生成代理类。Invoker 的构建过程以及代理类的过程比较重要,因此接下来将分两小节对这两个过程进行分析。

3.2.1 创建 Invoker

Invoker 是 Dubbo 的核心模型,代表一个可执行体。在服务提供方,Invoker 用于调用服务提供类。在服务消费方,Invoker 用于执行远程调用。Invoker 是由 Protocol 实现类构建而来。Protocol 实现类有很多,本节会分析最常用的两个,分别是 RegistryProtocol 和 DubboProtocol,其他的大家自行分析。下面先来分析 DubboProtocol 的 refer 方法源码。如下:

  1. public <T> Invoker<T> refer(Class<T> serviceType, URL url) throws RpcException {
  2. optimizeSerialization(url);
  3. // 创建 DubboInvoker
  4. DubboInvoker<T> invoker = new DubboInvoker<T>(serviceType, url, getClients(url), invokers);
  5. invokers.add(invoker);
  6. return invoker;
  7. }

上面方法看起来比较简单,不过这里有一个调用需要我们注意一下,即 getClients。这个方法用于获取客户端实例,实例类型为 ExchangeClient。ExchangeClient 实际上并不具备通信能力,它需要基于更底层的客户端实例进行通信。比如 NettyClient、MinaClient 等,默认情况下,Dubbo 使用 NettyClient 进行通信。接下来,我们简单看一下 getClients 方法的逻辑。

  1. private ExchangeClient[] getClients(URL url) {
  2. // 是否共享连接
  3. boolean service_share_connect = false;
  4. // 获取连接数,默认为0,表示未配置
  5. int connections = url.getParameter(Constants.CONNECTIONS_KEY, 0);
  6. // 如果未配置 connections,则共享连接
  7. if (connections == 0) {
  8. service_share_connect = true;
  9. connections = 1;
  10. }
  11. ExchangeClient[] clients = new ExchangeClient[connections];
  12. for (int i = 0; i < clients.length; i++) {
  13. if (service_share_connect) {
  14. // 获取共享客户端
  15. clients[i] = getSharedClient(url);
  16. } else {
  17. // 初始化新的客户端
  18. clients[i] = initClient(url);
  19. }
  20. }
  21. return clients;
  22. }

这里根据 connections 数量决定是获取共享客户端还是创建新的客户端实例,默认情况下,使用共享客户端实例。getSharedClient 方法中也会调用 initClient 方法,因此下面我们一起看一下这两个方法。

  1. private ExchangeClient getSharedClient(URL url) {
  2. String key = url.getAddress();
  3. // 获取带有“引用计数”功能的 ExchangeClient
  4. ReferenceCountExchangeClient client = referenceClientMap.get(key);
  5. if (client != null) {
  6. if (!client.isClosed()) {
  7. // 增加引用计数
  8. client.incrementAndGetCount();
  9. return client;
  10. } else {
  11. referenceClientMap.remove(key);
  12. }
  13. }
  14. locks.putIfAbsent(key, new Object());
  15. synchronized (locks.get(key)) {
  16. if (referenceClientMap.containsKey(key)) {
  17. return referenceClientMap.get(key);
  18. }
  19. // 创建 ExchangeClient 客户端
  20. ExchangeClient exchangeClient = initClient(url);
  21. // 将 ExchangeClient 实例传给 ReferenceCountExchangeClient,这里使用了装饰模式
  22. client = new ReferenceCountExchangeClient(exchangeClient, ghostClientMap);
  23. referenceClientMap.put(key, client);
  24. ghostClientMap.remove(key);
  25. locks.remove(key);
  26. return client;
  27. }
  28. }

上面方法先访问缓存,若缓存未命中,则通过 initClient 方法创建新的 ExchangeClient 实例,并将该实例传给 ReferenceCountExchangeClient 构造方法创建一个带有引用计数功能的 ExchangeClient 实例。ReferenceCountExchangeClient 内部实现比较简单,就不分析了。下面我们再来看一下 initClient 方法的代码。

  1. private ExchangeClient initClient(URL url) {
  2. // 获取客户端类型,默认为 netty
  3. String str = url.getParameter(Constants.CLIENT_KEY, url.getParameter(Constants.SERVER_KEY, Constants.DEFAULT_REMOTING_CLIENT));
  4. // 添加编解码和心跳包参数到 url 中
  5. url = url.addParameter(Constants.CODEC_KEY, DubboCodec.NAME);
  6. url = url.addParameterIfAbsent(Constants.HEARTBEAT_KEY, String.valueOf(Constants.DEFAULT_HEARTBEAT));
  7. // 检测客户端类型是否存在,不存在则抛出异常
  8. if (str != null && str.length() > 0 && !ExtensionLoader.getExtensionLoader(Transporter.class).hasExtension(str)) {
  9. throw new RpcException("Unsupported client type: ...");
  10. }
  11. ExchangeClient client;
  12. try {
  13. // 获取 lazy 配置,并根据配置值决定创建的客户端类型
  14. if (url.getParameter(Constants.LAZY_CONNECT_KEY, false)) {
  15. // 创建懒加载 ExchangeClient 实例
  16. client = new LazyConnectExchangeClient(url, requestHandler);
  17. } else {
  18. // 创建普通 ExchangeClient 实例
  19. client = Exchangers.connect(url, requestHandler);
  20. }
  21. } catch (RemotingException e) {
  22. throw new RpcException("Fail to create remoting client for service...");
  23. }
  24. return client;
  25. }

initClient 方法首先获取用户配置的客户端类型,默认为 netty。然后检测用户配置的客户端类型是否存在,不存在则抛出异常。最后根据 lazy 配置决定创建什么类型的客户端。这里的 LazyConnectExchangeClient 代码并不是很复杂,该类会在 request 方法被调用时通过 Exchangers 的 connect 方法创建 ExchangeClient 客户端,该类的代码本节就不分析了。下面我们分析一下 Exchangers 的 connect 方法。

  1. public static ExchangeClient connect(URL url, ExchangeHandler handler) throws RemotingException {
  2. if (url == null) {
  3. throw new IllegalArgumentException("url == null");
  4. }
  5. if (handler == null) {
  6. throw new IllegalArgumentException("handler == null");
  7. }
  8. url = url.addParameterIfAbsent(Constants.CODEC_KEY, "exchange");
  9. // 获取 Exchanger 实例,默认为 HeaderExchangeClient
  10. return getExchanger(url).connect(url, handler);
  11. }

如上,getExchanger 会通过 SPI 加载 HeaderExchangeClient 实例,这个方法比较简单,大家自己看一下吧。接下来分析 HeaderExchangeClient 的实现。

  1. public ExchangeClient connect(URL url, ExchangeHandler handler) throws RemotingException {
  2. // 这里包含了多个调用,分别如下:
  3. // 1. 创建 HeaderExchangeHandler 对象
  4. // 2. 创建 DecodeHandler 对象
  5. // 3. 通过 Transporters 构建 Client 实例
  6. // 4. 创建 HeaderExchangeClient 对象
  7. return new HeaderExchangeClient(Transporters.connect(url, new DecodeHandler(new HeaderExchangeHandler(handler))), true);
  8. }

这里的调用比较多,我们这里重点看一下 Transporters 的 connect 方法。如下:

  1. public static Client connect(URL url, ChannelHandler... handlers) throws RemotingException {
  2. if (url == null) {
  3. throw new IllegalArgumentException("url == null");
  4. }
  5. ChannelHandler handler;
  6. if (handlers == null || handlers.length == 0) {
  7. handler = new ChannelHandlerAdapter();
  8. } else if (handlers.length == 1) {
  9. handler = handlers[0];
  10. } else {
  11. // 如果 handler 数量大于1,则创建一个 ChannelHandler 分发器
  12. handler = new ChannelHandlerDispatcher(handlers);
  13. }
  14. // 获取 Transporter 自适应拓展类,并调用 connect 方法生成 Client 实例
  15. return getTransporter().connect(url, handler);
  16. }

如上,getTransporter 方法返回的是自适应拓展类,该类会在运行时根据客户端类型加载指定的 Transporter 实现类。若用户未配置客户端类型,则默认加载 NettyTransporter,并调用该类的 connect 方法。如下:

  1. public Client connect(URL url, ChannelHandler listener) throws RemotingException {
  2. // 创建 NettyClient 对象
  3. return new NettyClient(url, listener);
  4. }

到这里就不继续跟下去了,在往下就是通过 Netty 提供的 API 构建 Netty 客户端了,大家有兴趣自己看看。到这里,关于 DubboProtocol 的 refer 方法就分析完了。接下来,继续分析 RegistryProtocol 的 refer 方法逻辑。

  1. public <T> Invoker<T> refer(Class<T> type, URL url) throws RpcException {
  2. // 取 registry 参数值,并将其设置为协议头
  3. url = url.setProtocol(url.getParameter(Constants.REGISTRY_KEY, Constants.DEFAULT_REGISTRY)).removeParameter(Constants.REGISTRY_KEY);
  4. // 获取注册中心实例
  5. Registry registry = registryFactory.getRegistry(url);
  6. if (RegistryService.class.equals(type)) {
  7. return proxyFactory.getInvoker((T) registry, type, url);
  8. }
  9. // 将 url 查询字符串转为 Map
  10. Map<String, String> qs = StringUtils.parseQueryString(url.getParameterAndDecoded(Constants.REFER_KEY));
  11. // 获取 group 配置
  12. String group = qs.get(Constants.GROUP_KEY);
  13. if (group != null && group.length() > 0) {
  14. if ((Constants.COMMA_SPLIT_PATTERN.split(group)).length > 1
  15. || "*".equals(group)) {
  16. // 通过 SPI 加载 MergeableCluster 实例,并调用 doRefer 继续执行服务引用逻辑
  17. return doRefer(getMergeableCluster(), registry, type, url);
  18. }
  19. }
  20. // 调用 doRefer 继续执行服务引用逻辑
  21. return doRefer(cluster, registry, type, url);
  22. }

上面代码首先为 url 设置协议头,然后根据 url 参数加载注册中心实例。然后获取 group 配置,根据 group 配置决定 doRefer 第一个参数的类型。这里的重点是 doRefer 方法,如下:

  1. private <T> Invoker<T> doRefer(Cluster cluster, Registry registry, Class<T> type, URL url) {
  2. // 创建 RegistryDirectory 实例
  3. RegistryDirectory<T> directory = new RegistryDirectory<T>(type, url);
  4. // 设置注册中心和协议
  5. directory.setRegistry(registry);
  6. directory.setProtocol(protocol);
  7. Map<String, String> parameters = new HashMap<String, String>(directory.getUrl().getParameters());
  8. // 生成服务消费者链接
  9. URL subscribeUrl = new URL(Constants.CONSUMER_PROTOCOL, parameters.remove(Constants.REGISTER_IP_KEY), 0, type.getName(), parameters);
  10. // 注册服务消费者,在 consumers 目录下新节点
  11. if (!Constants.ANY_VALUE.equals(url.getServiceInterface())
  12. && url.getParameter(Constants.REGISTER_KEY, true)) {
  13. registry.register(subscribeUrl.addParameters(Constants.CATEGORY_KEY, Constants.CONSUMERS_CATEGORY,
  14. Constants.CHECK_KEY, String.valueOf(false)));
  15. }
  16. // 订阅 providers、configurators、routers 等节点数据
  17. directory.subscribe(subscribeUrl.addParameter(Constants.CATEGORY_KEY,
  18. Constants.PROVIDERS_CATEGORY
  19. + "," + Constants.CONFIGURATORS_CATEGORY
  20. + "," + Constants.ROUTERS_CATEGORY));
  21. // 一个注册中心可能有多个服务提供者,因此这里需要将多个服务提供者合并为一个
  22. Invoker invoker = cluster.join(directory);
  23. ProviderConsumerRegTable.registerConsumer(invoker, url, subscribeUrl, directory);
  24. return invoker;
  25. }

如上,doRefer 方法创建一个 RegistryDirectory 实例,然后生成服务者消费者链接,并向注册中心进行注册。注册完毕后,紧接着订阅 providers、configurators、routers 等节点下的数据。完成订阅后,RegistryDirectory 会收到这几个节点下的子节点信息。由于一个服务可能部署在多台服务器上,这样就会在 providers 产生多个节点,这个时候就需要 Cluster 将多个服务节点合并为一个,并生成一个 Invoker。关于 RegistryDirectory 和 Cluster,本文不打算进行分析,相关分析将会在随后的文章中展开。

3.2.2 创建代理

Invoker 创建完毕后,接下来要做的事情是为服务接口生成代理对象。有了代理对象,即可进行远程调用。代理对象生成的入口方法为 ProxyFactory 的 getProxy,接下来进行分析。

  1. public <T> T getProxy(Invoker<T> invoker) throws RpcException {
  2. // 调用重载方法
  3. return getProxy(invoker, false);
  4. }
  5. public <T> T getProxy(Invoker<T> invoker, boolean generic) throws RpcException {
  6. Class<?>[] interfaces = null;
  7. // 获取接口列表
  8. String config = invoker.getUrl().getParameter("interfaces");
  9. if (config != null && config.length() > 0) {
  10. // 切分接口列表
  11. String[] types = Constants.COMMA_SPLIT_PATTERN.split(config);
  12. if (types != null && types.length > 0) {
  13. interfaces = new Class<?>[types.length + 2];
  14. // 设置服务接口类和 EchoService.class 到 interfaces 中
  15. interfaces[0] = invoker.getInterface();
  16. interfaces[1] = EchoService.class;
  17. for (int i = 0; i < types.length; i++) {
  18. // 加载接口类
  19. interfaces[i + 1] = ReflectUtils.forName(types[i]);
  20. }
  21. }
  22. }
  23. if (interfaces == null) {
  24. interfaces = new Class<?>[]{invoker.getInterface(), EchoService.class};
  25. }
  26. // 为 http 和 hessian 协议提供泛化调用支持,参考 pull request #1827
  27. if (!invoker.getInterface().equals(GenericService.class) && generic) {
  28. int len = interfaces.length;
  29. Class<?>[] temp = interfaces;
  30. // 创建新的 interfaces 数组
  31. interfaces = new Class<?>[len + 1];
  32. System.arraycopy(temp, 0, interfaces, 0, len);
  33. // 设置 GenericService.class 到数组中
  34. interfaces[len] = GenericService.class;
  35. }
  36. // 调用重载方法
  37. return getProxy(invoker, interfaces);
  38. }
  39. public abstract <T> T getProxy(Invoker<T> invoker, Class<?>[] types);

如上,上面大段代码都是用来获取 interfaces 数组的,我们继续往下看。getProxy(Invoker, Class<?>[]) 这个方法是一个抽象方法,下面我们到 JavassistProxyFactory 类中看一下该方法的实现代码。

  1. public <T> T getProxy(Invoker<T> invoker, Class<?>[] interfaces) {
  2. // 生成 Proxy 子类(Proxy 是抽象类)。并调用 Proxy 子类的 newInstance 方法创建 Proxy 实例
  3. return (T) Proxy.getProxy(interfaces).newInstance(new InvokerInvocationHandler(invoker));
  4. }

上面代码并不多,首先是通过 Proxy 的 getProxy 方法获取 Proxy 子类,然后创建 InvokerInvocationHandler 对象,并将该对象传给 newInstance 生成 Proxy 实例。InvokerInvocationHandler 实现 JDK 的 InvocationHandler 接口,具体的用途是拦截接口类调用。该类逻辑比较简单,这里就不分析了。下面我们重点关注一下 Proxy 的 getProxy 方法,如下。

  1. public static Proxy getProxy(Class<?>... ics) {
  2. // 调用重载方法
  3. return getProxy(ClassHelper.getClassLoader(Proxy.class), ics);
  4. }
  5. public static Proxy getProxy(ClassLoader cl, Class<?>... ics) {
  6. if (ics.length > 65535)
  7. throw new IllegalArgumentException("interface limit exceeded");
  8. StringBuilder sb = new StringBuilder();
  9. // 遍历接口列表
  10. for (int i = 0; i < ics.length; i++) {
  11. String itf = ics[i].getName();
  12. // 检测类型是否为接口
  13. if (!ics[i].isInterface())
  14. throw new RuntimeException(itf + " is not a interface.");
  15. Class<?> tmp = null;
  16. try {
  17. // 重新加载接口类
  18. tmp = Class.forName(itf, false, cl);
  19. } catch (ClassNotFoundException e) {
  20. }
  21. // 检测接口是否相同,这里 tmp 有可能为空
  22. if (tmp != ics[i])
  23. throw new IllegalArgumentException(ics[i] + " is not visible from class loader");
  24. // 拼接接口全限定名,分隔符为 ;
  25. sb.append(itf).append(';');
  26. }
  27. // 使用拼接后的接口名作为 key
  28. String key = sb.toString();
  29. Map<String, Object> cache;
  30. synchronized (ProxyCacheMap) {
  31. cache = ProxyCacheMap.get(cl);
  32. if (cache == null) {
  33. cache = new HashMap<String, Object>();
  34. ProxyCacheMap.put(cl, cache);
  35. }
  36. }
  37. Proxy proxy = null;
  38. synchronized (cache) {
  39. do {
  40. // 从缓存中获取 Reference<Proxy> 实例
  41. Object value = cache.get(key);
  42. if (value instanceof Reference<?>) {
  43. proxy = (Proxy) ((Reference<?>) value).get();
  44. if (proxy != null) {
  45. return proxy;
  46. }
  47. }
  48. // 并发控制,保证只有一个线程可以进行后续操作
  49. if (value == PendingGenerationMarker) {
  50. try {
  51. // 其他线程在此处进行等待
  52. cache.wait();
  53. } catch (InterruptedException e) {
  54. }
  55. } else {
  56. // 放置标志位到缓存中,并跳出 while 循环进行后续操作
  57. cache.put(key, PendingGenerationMarker);
  58. break;
  59. }
  60. }
  61. while (true);
  62. }
  63. long id = PROXY_CLASS_COUNTER.getAndIncrement();
  64. String pkg = null;
  65. ClassGenerator ccp = null, ccm = null;
  66. try {
  67. // 创建 ClassGenerator 对象
  68. ccp = ClassGenerator.newInstance(cl);
  69. Set<String> worked = new HashSet<String>();
  70. List<Method> methods = new ArrayList<Method>();
  71. for (int i = 0; i < ics.length; i++) {
  72. // 检测接口访问级别是否为 protected 或 privete
  73. if (!Modifier.isPublic(ics[i].getModifiers())) {
  74. // 获取接口包名
  75. String npkg = ics[i].getPackage().getName();
  76. if (pkg == null) {
  77. pkg = npkg;
  78. } else {
  79. if (!pkg.equals(npkg))
  80. // 非 public 级别的接口必须在同一个包下,否者抛出异常
  81. throw new IllegalArgumentException("non-public interfaces from different packages");
  82. }
  83. }
  84. // 添加接口到 ClassGenerator 中
  85. ccp.addInterface(ics[i]);
  86. // 遍历接口方法
  87. for (Method method : ics[i].getMethods()) {
  88. // 获取方法描述,可理解为方法签名
  89. String desc = ReflectUtils.getDesc(method);
  90. // 如果方法描述字符串已在 worked 中,则忽略。考虑这种情况,
  91. // A 接口和 B 接口中包含一个完全相同的方法
  92. if (worked.contains(desc))
  93. continue;
  94. worked.add(desc);
  95. int ix = methods.size();
  96. // 获取方法返回值类型
  97. Class<?> rt = method.getReturnType();
  98. // 获取参数列表
  99. Class<?>[] pts = method.getParameterTypes();
  100. // 生成 Object[] args = new Object[1...N]
  101. StringBuilder code = new StringBuilder("Object[] args = new Object[").append(pts.length).append("];");
  102. for (int j = 0; j < pts.length; j++)
  103. // 生成 args[1...N] = ($w)$1...N;
  104. code.append(" args[").append(j).append("] = ($w)$").append(j + 1).append(";");
  105. // 生成 InvokerHandler 接口的 invoker 方法调用语句,如下:
  106. // Object ret = handler.invoke(this, methods[1...N], args);
  107. code.append(" Object ret = handler.invoke(this, methods[" + ix + "], args);");
  108. // 返回值不为 void
  109. if (!Void.TYPE.equals(rt))
  110. // 生成返回语句,形如 return (java.lang.String) ret;
  111. code.append(" return ").append(asArgument(rt, "ret")).append(";");
  112. methods.add(method);
  113. // 添加方法名、访问控制符、参数列表、方法代码等信息到 ClassGenerator 中
  114. ccp.addMethod(method.getName(), method.getModifiers(), rt, pts, method.getExceptionTypes(), code.toString());
  115. }
  116. }
  117. if (pkg == null)
  118. pkg = PACKAGE_NAME;
  119. // 构建接口代理类名称:pkg + ".proxy" + id,比如 org.apache.dubbo.proxy0
  120. String pcn = pkg + ".proxy" + id;
  121. ccp.setClassName(pcn);
  122. ccp.addField("public static java.lang.reflect.Method[] methods;");
  123. // 生成 private java.lang.reflect.InvocationHandler handler;
  124. ccp.addField("private " + InvocationHandler.class.getName() + " handler;");
  125. // 为接口代理类添加带有 InvocationHandler 参数的构造方法,比如:
  126. // porxy0(java.lang.reflect.InvocationHandler arg0) {
  127. // handler=$1;
  128. // }
  129. ccp.addConstructor(Modifier.PUBLIC, new Class<?>[]{InvocationHandler.class}, new Class<?>[0], "handler=$1;");
  130. // 为接口代理类添加默认构造方法
  131. ccp.addDefaultConstructor();
  132. // 生成接口代理类
  133. Class<?> clazz = ccp.toClass();
  134. clazz.getField("methods").set(null, methods.toArray(new Method[0]));
  135. // 构建 Proxy 子类名称,比如 Proxy1,Proxy2 等
  136. String fcn = Proxy.class.getName() + id;
  137. ccm = ClassGenerator.newInstance(cl);
  138. ccm.setClassName(fcn);
  139. ccm.addDefaultConstructor();
  140. ccm.setSuperClass(Proxy.class);
  141. // 为 Proxy 的抽象方法 newInstance 生成实现代码,形如:
  142. // public Object newInstance(java.lang.reflect.InvocationHandler h) {
  143. // return new org.apache.dubbo.proxy0($1);
  144. // }
  145. ccm.addMethod("public Object newInstance(" + InvocationHandler.class.getName() + " h){ return new " + pcn + "($1); }");
  146. // 生成 Proxy 实现类
  147. Class<?> pc = ccm.toClass();
  148. // 通过反射创建 Proxy 实例
  149. proxy = (Proxy) pc.newInstance();
  150. } catch (RuntimeException e) {
  151. throw e;
  152. } catch (Exception e) {
  153. throw new RuntimeException(e.getMessage(), e);
  154. } finally {
  155. if (ccp != null)
  156. // 释放资源
  157. ccp.release();
  158. if (ccm != null)
  159. ccm.release();
  160. synchronized (cache) {
  161. if (proxy == null)
  162. cache.remove(key);
  163. else
  164. // 写缓存
  165. cache.put(key, new WeakReference<Proxy>(proxy));
  166. // 唤醒其他等待线程
  167. cache.notifyAll();
  168. }
  169. }
  170. return proxy;
  171. }

上面代码比较复杂,我们写了大量的注释。大家在阅读这段代码时,要搞清楚 ccp 和 ccm 的用途,不然会被搞晕。ccp 用于为服务接口生成代理类,比如我们有一个 DemoService 接口,这个接口代理类就是由 ccp 生成的。ccm 则是用于为 org.apache.dubbo.common.bytecode.Proxy 抽象类生成子类,主要是实现 Proxy 类的抽象方法。下面以 org.apache.dubbo.demo.DemoService 这个接口为例,来看一下该接口代理类代码大致是怎样的(忽略 EchoService 接口)。

  1. package org.apache.dubbo.common.bytecode;
  2. public class proxy0 implements org.apache.dubbo.demo.DemoService {
  3. public static java.lang.reflect.Method[] methods;
  4. private java.lang.reflect.InvocationHandler handler;
  5. public proxy0() {
  6. }
  7. public proxy0(java.lang.reflect.InvocationHandler arg0) {
  8. handler = $1;
  9. }
  10. public java.lang.String sayHello(java.lang.String arg0) {
  11. Object[] args = new Object[1];
  12. args[0] = ($w) $1;
  13. Object ret = handler.invoke(this, methods[0], args);
  14. return (java.lang.String) ret;
  15. }
  16. }

好了,到这里代理类生成逻辑就分析完了。整个过程比较复杂,大家需要耐心看一下。