1. 程式人生 > 其它 >dubbo實戰之“動態呼叫服務”(實為“泛化呼叫”)探索

dubbo實戰之“動態呼叫服務”(實為“泛化呼叫”)探索

一、簡介

需求:Dubbo的呼叫方,在不引入服務介面類的情況下,遠端呼叫其他Dubbo服務。

二、專案依賴

SpringBoot整合Dubbo3.x關於curator和zookeeper版本選型的思考

經嘗試,使用高版本組合(Curator 5.2.0 + ZooKeeper 3.6.3)

三、核心程式碼

參考自 Apache Dubbo 高階用法 使用泛化呼叫

如果引用的是 Dubbo 3.0.5

import cn.hutool.core.util.StrUtil;
import org.apache.dubbo.config.ApplicationConfig;
import org.apache.dubbo.config.ReferenceConfig;
import org.apache.dubbo.config.RegistryConfig;
import org.apache.dubbo.config.utils.SimpleReferenceCache;
import org.apache.dubbo.rpc.model.ApplicationModel;
import org.apache.dubbo.rpc.service.GenericService;
import org.springframework.lang.Nullable;

/**
 * 功能描述: 基於 Dubbo 泛化呼叫特性的遠端呼叫
 *
 * @author geekziyu
 * @version 1.0.0
 */
class DubboRpc {

  private RegistryConfig registry;

  DubboRpc(ApplicationConfig application, RegistryConfig registry) {
    this.registry = registry;
    ApplicationModel.defaultModel().getApplicationConfigManager().setApplication(application);
  }

  Object genericInvoke(String interfaceClass, String methodName, String[] parameterTypes, Object[] args) {
    return genericInvoke(interfaceClass, methodName, parameterTypes, args, null, null);
  }

  Object genericInvoke(String interfaceClass, String methodName, String[] parameterTypes, Object[] args, @Nullable String group, @Nullable String version) {
    ReferenceConfig<GenericService> reference = new ReferenceConfig<>();
    reference.setRegistry(registry);
    reference.setInterface(interfaceClass); // 介面名
    reference.setGeneric("true"); // 宣告為泛化介面
    if (StrUtil.isNotBlank(group)) { // NOSONAR
      reference.setGroup(group);
    }
    if (StrUtil.isNotBlank(version)) {// NOSONAR
      reference.setVersion(version);
    }

    // ReferenceConfig例項很重,封裝了與註冊中心的連線以及與提供者的連線,
    // 需要快取,否則重複生成ReferenceConfig可能造成效能問題並且會有記憶體和連線洩漏。
    // API方式程式設計時,容易忽略此問題。
    // 這裡使用dubbo內建的簡單快取工具類進行快取
    SimpleReferenceCache cache = SimpleReferenceCache.getCache();
    GenericService genericService = cache.get(reference);
    // 用com.alibaba.dubbo.rpc.service.GenericService可以替代所有介面引用
    return genericService.$invoke(methodName, parameterTypes, args);
  }
}

如果引用的是 Dubbo 3.0.1

import cn.hutool.core.util.StrUtil;
import org.apache.dubbo.config.ApplicationConfig;
import org.apache.dubbo.config.ReferenceConfig;
import org.apache.dubbo.config.RegistryConfig;
import org.apache.dubbo.config.utils.ReferenceConfigCache;
import org.apache.dubbo.rpc.model.ApplicationModel;
import org.apache.dubbo.rpc.service.GenericService;

import javax.annotation.Nullable;

/**
 * 功能描述: 基於 Dubbo 泛化呼叫特性的遠端呼叫
 *
 * @author [email protected]
 * @version 1.0.0
 */
class DubboRpc {

  private RegistryConfig registry;

  DubboRpc(ApplicationConfig application, RegistryConfig registry) {
    this.registry = registry;
    ApplicationModel.getConfigManager().setApplication(application);
  }

  Object genericInvoke(String interfaceClass, String methodName, String[] parameterTypes, Object[] args) {
    return genericInvoke(interfaceClass, methodName, parameterTypes, args, null, null);
  }

  Object genericInvoke(String interfaceClass, String methodName, String[] parameterTypes, Object[] args, @Nullable String group, @Nullable String version) {
    referenceConfig<GenericService> reference = new ReferenceConfig<>();
    reference.setRegistry(registry);
    reference.setInterface(interfaceClass); // 介面名
    reference.setGeneric("true"); // 宣告為泛化介面
    if (StrUtil.isNotBlank(group)) { // NOSONAR
      reference.setGroup(group);
    }
    if (StrUtil.isNotBlank(version)) {// NOSONAR
      reference.setVersion(version);
    }

    // ReferenceConfig例項很重,封裝了與註冊中心的連線以及與提供者的連線,
    // 需要快取,否則重複生成ReferenceConfig可能造成效能問題並且會有記憶體和連線洩漏。
    // API方式程式設計時,容易忽略此問題。
    // 這裡使用dubbo內建的簡單快取工具類進行快取
    ReferenceConfigCache cache = ReferenceConfigCache.getCache();
    GenericService genericService = cache.get(reference);
    // 用com.alibaba.dubbo.rpc.service.GenericService可以替代所有介面引用

    return genericService.$invoke(methodName, parameterTypes, args);
  }
}

至於這個 RegistryConfig 可以這樣:

@Service
public class RpcInvokerService implements InitializingBean {
  @Value("${dubbo.application.name}")
  private String applicationName;

  @Value("${dubbo.registry.protocol}")
  private String protocol;

  @Value("${dubbo.registry.address}")
  private String address;

  private DubboRpc rpc;

  @Override
  public void afterPropertiesSet() {
    ApplicationConfig application = new ApplicationConfig(applicationName);
    RegistryConfig registry = new RegistryConfig();
    registry.setProtocol(protocol);
    registry.setAddress(address);
    rpc = new DubboRpc(application, registry);
  }

  // ...
}