dubbo源码:dubbo RpcContext获取不到参数值问题分析

文章目录

      • 背景
      • 原因分析
      • 解决办法

背景

在一次服务调用中(服务采用的dubbo框架),调用方法之前会有一次Spring AOP参数的检验(调用方是否设置"userId"),若没有直接返回,代码如下:

@Component
@Aspect
public class CmcLoginAspect {
    @Pointcut("execution(com.yt.asd.common.domain.RpcResult com.yt.cmc.biz.ao.shop..*.*(..))")
    public void aspectLogin() {
    }

    @Around(value = "aspectLogin()")
    public Object appLoginHandling(ProceedingJoinPoint jp) throws Throwable {
        // 重点
        String userId = RpcContext.getContext().getAttachment("userId");
        // userId不存在, 重新登录
        if (StringUtils.isBlank(userId)) {
            log.error("rpc上下文中无userId!");
            return ResultUtil.newInstance().getResult(BaseCode.NO_LOGIN_INFO, false);
        }
        Object object = jp.proceed();
        return (RpcResult) object;
    }
}

若存在"userId"则进入方法调用,而在调用方法的入口重新从RpcContext获取"userId",如下:

// Case 1:能获取到userId
public void test1(){
  // 能获取到userId
  String userId = RpcContext.getContext().getAttachment("userId");
  i//....
} 

// Case 2:不能获取到userId
public void test2(){
  // 一次远程调用
  Result result = ShopAdapter.getById(3L);
  // 不能获取到userId
  String userId = RpcContext.getContext().getAttachment("userId");
  //....
} 

问题是若调用方正常设置了"userId",test1方法能获取到userId,test2方法不能获取到userId。

原因分析

在每一次Protocol层消费者调用过程中,会依次调用ConsumerContextFilter的invoke方法、FutureFilter的invoke方法、MonitorFilter中的invoke⽅法,最后会调到DubboInvoker中invoke⽅法。这里只看ConsumerContextFilter的invoke方法,如下:

@Activate(group = Constants.CONSUMER, order = -10000)
public class ConsumerContextFilter implements Filter {

    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
        RpcContext.getContext()
                .setInvoker(invoker)
                .setInvocation(invocation)
                .setLocalAddress(NetUtils.getLocalHost(), 0)
                .setRemoteAddress(invoker.getUrl().getHost(), 
                                  invoker.getUrl().getPort());
        if (invocation instanceof RpcInvocation) {
            ((RpcInvocation)invocation).setInvoker(invoker);
        }
        try {
            return invoker.invoke(invocation);
        } finally {
            RpcContext.getContext().clearAttachments();
        }
    }
}

上述源码知每一次dubbo接口调用结束后都会清除RpcContext的attachments,在test1中由于只是一次调用,所以会获取到userId,而在test2方法中,获取userId之前已经存在一次远程调用,此时已经清空了RpcContext的attachments,所以在获取userId时获取不到。

  1. 生成代理过程。本例使用了AOP+dubbo,即服务最外层入口有AOP,接口是使用dubbo来代理的,这样的首先spring aop会对类生成代理(比如JDK动态代理),然后dubbo把该代理类封装在invoker中。

  2. 调用过程中先经过dubbo filter进入到AbstractProxyInvoker的invoke方法,然后再执行Wrapper字节码(JavassistProxyFactory中getInvoker方法生成的匿名实例的doInvoke方法)invokeMethod方法,字节码中会调用aop生成的代理类,调用栈如下:

dubbo源码:dubbo RpcContext获取不到参数值问题分析_第1张图片
dubbo源码:dubbo RpcContext获取不到参数值问题分析_第2张图片

解决办法

把参数值可放到 ThreadLocal 中便可解决问题,不再详述!

你可能感兴趣的:(研发管理,dubbo源码分析)