yhzhu的随手记
0%

基于配置中心属性,Dubbo Consumer动态切换调用RPC/本地方法

发表于

需求缘由

最近部分只读服务已经切换到了Dubbo提供的服务化接口；经过一段时间的生产验证, 各种监控指标显示非常稳定，所以打算开始切换写服务接口; 但大家对写操作跑服务化没有信心，于是提了一个需求能随时将远程rpc调用切换打本地jar方法调用；

dubbo的stub

听到这个需求，马上想到dubbo的stub属性；但需要在原来代码中做部分改造,具体步骤是：

  1. 原来的jar包实现方法中增加带自身接口类型参数的构造函数;
  2. 在所有的接口实现方法中判断开关状态决定调用RPC/本地方法;
    但是，这样对原代码有极强入侵性；我们认为这不是最好的方案，于是有了下面的方案；

扩展dubbo:reference标签属性

首先需要说明，我们在实际使用dubbo的过程中并没有直接使用dubbo这个命名空间，而是自定义了自己的命名空间，自定义命名空间兼容了dubbo命名空间的所有属性，而且扩展了自己的属性；于是针对该需求我们有了新的方案，具体流程如下 ：

  1. 增加reference的属性localbean，这个属性值为一个本地bean的id；类似的xml配置如下：
    1
    2
    3
    <bean id="localDemoService" class="com.example.Service.impl.DemoServiceImpl" />
    <mydubbo:reference interface="com.example.Service.DemoService" id="remoteDemoService"
                     localbean ="localDemoService" version="1.0"/>
  2. 继承ReferenceBean实现自定义属性locbean；
  3. 继承DubboNamespaceHandler重写init方法,如下:
    1
    2
    3
    4
    public void init() {
            super.init();
            this.registerBeanDefinitionParser("reference", new MyBeanDefinitionParser(LocalReferenceBean.class, false));
        }
  4. 自定义filter
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    public class RPCLocalSwitchFilter<T> implements Filter {
        private static final Logger logger = LoggerFactory.getLogger(RPCLocalSwitchFilter.class);
        private static final ProxyFactory PROXY_FACTORY = (ProxyFactory)ExtensionLoader.getExtensionLoader(ProxyFactory.class).getAdaptiveExtension();

        public RPCLocalSwitchFilter() {
        }

        public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
            if(cfg.getBooleanProperty("")) {  //判断配置中心开工状态
                Object bean = this.getLocalBean(invoker);
                return bean != null?this.executeLocalMethod(bean, invoker, invocation):invoker.invoke(invocation);
            } else {
                return invoker.invoke(invocation);
            }
        }

        private Result executeLocalMethod(T bean, Invoker<?> invoker, Invocation invocation) {
            if(logger.isDebugEnabled()) {
                logger.debug("execute local method instead of rpc " + bean.getClass().getCanonicalName() + "." + invocation.getMethodName());
            }

            Invoker localInvoker = PROXY_FACTORY.getInvoker(bean, invoker.getInterface(), invoker.getUrl());
            return localInvoker.invoke(invocation);
        }

        private <T> T getLocalBean(Invoker<?> invoker) {
            String refLocal = invoker.getUrl().getParameter("localBean");
            if(refLocal != null) {
                try {
                    ApplicationContext e = LocalReferenceBean.getSpringContext();
                    return e.getBean(refLocal, invoker.getInterface());
                } catch (BeansException e) {
                    logger.error(var4.getMessage() + " couldn\'t find the bean[" + localBean + "], please check", e);
                }
            }

            return null;
        }
    }
  5. 配置filter
    在resources/META-INF/dubbo/com.alibaba.dubbo.rpc.Filter文件中增加如下内容:
    1
    reflocal=com.example.filter. RPCLocalSwitchFilter
    配置consumer的filter属性
    1
    <mydubbo:consumer filter="reflocal" />
    至此可以通过配置中的开关状态来控制远程RPC/本地jar方法调用 ;