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生成的代理类，调用栈如下：

解决办法

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