dubbo微服务之间的隐式数据传递

利用dubbo的Attachment做隐式传递#

在做全链路压测插件时,如何让流量标识在dubbo远程RPC调用时进行传递是一个需要解决的问题,就可以用dubbo的Attachment做隐式传递,服务A） —> （服务B —> ）（服务C）等等.

• 上下文类

  1
  public class WormholeContextHolder {
  2
  
  3
      private static final InheritableThreadLocal<WormholeContext> wormholeContextThreadLocal = new InheritableThreadLocal<>();
  4
  
  5
      /**
  6
       * 上下文流量标识
  7
       */
  8
      public static final String WORMHOLE_REQUEST_MARK = "X-Wormhole-Header-Request";
  9
  
  10
      /**
  11
       * @return com.hoolai.wormhole.core.context.WormholeContext
  12
       * @description 获取上下文
  13
       */
  14
      public static WormholeContext getContext() {
  15
          return wormholeContextThreadLocal.get();
  16
      }
  17
  
  18
      /**
  19
       * @param key key
  20
       * @return java.lang.String
  21
       * @description 获取上下文中的属性
  22
       */
  23
      public static String getProperty(String key) {
  24
          WormholeContext wormholeContext = getContext();
  25
          if (wormholeContext != null) {
  26
              return wormholeContext.toString();
  27
          }
  28
          return null;
  29
      }
  30
  
  31
      /**
  32
       * @param context context
  33
       * @description 设置上下文
  34
       */
  35
      public static void setContext(WormholeContext context) {
  36
          wormholeContextThreadLocal.set(context);
  37
      }
  38
  
  39
      public static void setContext(String contextStr) {
  40
          if (StringUtils.isNotEmpty(contextStr)) {
  41
              setContext(new WormholeContext(contextStr));
  42
          }
  43
      }
  44
  
  45
      /**
  46
       * @description 将上下文进行失效
  47
       */
  48
      public static void invalidContext() {
  49
          wormholeContextThreadLocal.remove();
  50
      }
  51
  
  52
  }

  展开收起

• 消费者

  1
  import org.apache.dubbo.common.constants.CommonConstants;
  2
  import org.apache.dubbo.common.extension.Activate;
  3
  import org.apache.dubbo.rpc.*;
  4
  
  5
  @Activate(group = CommonConstants.CONSUMER)
  6
  public class ParamConsumerFilter implements Filter {
  7
  
  8
      @Override
  9
      public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
  10
          WormholeContext wormholeContext = WormholeContextHolder.getContext();
  11
          if (wormholeContext != null) {
  12
              RpcContext.getContext().setAttachment(WormholeContextHolder.WORMHOLE_REQUEST_MARK, wormholeContext.getWormholeValue());
  13
          }
  14
          return invoker.invoke(invocation);
  15
      }
  16
  }

  展开收起

• 生产者

  1
  import org.apache.dubbo.common.constants.CommonConstants;
  2
  import org.apache.dubbo.common.extension.Activate;
  3
  import org.apache.dubbo.rpc.*;
  4
  
  5
  @Activate(group = CommonConstants.PROVIDER)
  6
  public class ParamProviderFilter implements Filter {
  7
  
  8
      @Override
  9
      public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
  10
          String wormholeContext = RpcContext.getContext().getAttachment(WormholeContextHolder.WORMHOLE_REQUEST_MARK);
  11
          if (StringUtils.isNotBlank(wormholeContext)) {
  12
              RpcContext.getContext().setAttachment(WormholeContextHolder.WORMHOLE_REQUEST_MARK, wormholeContext);
  13
              WormholeContextHolder.setContext(wormholeContext);
  14
          }
  15
          return invoker.invoke(invocation);
  16
      }
  17
  }

  展开收起

• 过滤器的SPI配置

  在资源文件夹下创建一个3层的文件夹 — dubbo — internal ,注意不要一块写中间加.分隔的形式,这样创建的是META-INF.dubbo.internal,变成了一个文件夹.

  文件夹下创建一个名为org.apache.dubbo.rpc.Filter的文件.配置这一块可以参考dubbo源码的过滤器部分

  文件内容如下,名称=类名

  1
  paramConsumerFilter=com.tianch.wormhole.component.dubbo.ParamConsumerFilter
  2
  paramProviderFilter=com.tianch.wormhole.component.dubbo.ParamProviderFilter