Dubbo异步方法调用里有个坑(更新)

Dubbo是一个分布式服务框架，是阿里巴巴SOA服务化治理方案的核心框架。

使用Dubbo进行服务化后，现有如下场景时序图：

场景描述：客户端远程异步调用ServiceA，ServiceA在处理客户端请求的过程中需要远程同步调用ServiceB，ServiceA从ServiceB的响应中取数据时，得到的是null，对就是这个坑。

使用DEBUG模式，分析Dubbo源码得到问题的起因。

分析过程如下：

客户端和服务端通信，配置使用netty进行网络传输，通过

NettyHandler进行具体的消息收发操作，所以从此入手进行源码分析。

client到ServiceA的远程方法异步调用，会在RpcContext(RpcContext是一个临时状态记录器，当接收到RPC请求，或发起RPC请求时，RpcContext的状态都会变化。比如：A调B，B再调C，则B机器上，在B调C之前，RpcContext记录的是A调B的信息，在B调C之后，RpcContext记录的是B调C的信息)的attachments(Map结构)属性中添加async=true的键值对，同时也会在RpcInvocation的attachments(Map结构)中添加async=true的键值对。

经过一系列的Filter，程序运行到AbstractInvoker的invoke方法，注意该方法中的如下代码段，

Map<String, String> context = RpcContext.getContext().getAttachments();

if (context != null) {

invocation.addAttachmentsIfAbsent(context);

}

这里会把当前RpcContext中的attachments添加到调用ServiceB的RpcInvocation中，这时候async=true已经添加了，接着执行如下代码段，

if (getUrl().getMethodParameter(invocation.getMethodName(), Constants.ASYNC_KEY, false)){

invocation.setAttachment(Constants.ASYNC_KEY, Boolean.TRUE.toString());

}

上面代码判断调用ServiceB的URL中是否含有async=true，如果有将设置async=true到RpcInvocation的attachments中，这时候是不包含的。

继续跟踪代码，运行到DubboInvoker中，调用doInvoke方法，该方法中有如下的代码段，boolean isAsync = RpcUtils.isAsync(getUrl(), invocation)，这个isAsync方法具体声明如下，

public static boolean isAsync(URL url, Invocation inv) {

boolean isAsync ;

//如果Java代码中设置优先.

if (Boolean.TRUE.toString().equals(inv.getAttachment(Constants.ASYNC_KEY))) {

isAsync = true;

} else {

isAsync = url.getMethodParameter(getMethodName(inv), Constants.ASYNC_KEY, false);

}

return isAsync;

}

上面方法首先判断RpcInvocation的attachments中async=true是否成立，如果成立则这是一次异步调用，否则判断请求URL中async=true是否成立，如果成立则是一次异步调用，否则是一次同步调用，根据上面传递的参数，此时isAsync方法返回的是true，ServiceA同步调用ServiceB变成了异步调用，继续看下面的异步调用，代码段如下，

else if (isAsync) {

ResponseFuture future = currentClient.request(inv, timeout) ;

RpcContext.getContext().setFuture(new FutureAdapter<Object>(future));

return new RpcResult();

}

这里直接返回了一个RpcResult对象，没有数据内容，所以到这里，这个案子也就破了，ServiceA想从响应中取目标数据得到的当然是null。再延伸一下，如果ServiceB再同步调用ServiceC，这是可以正常同步调用的，因为ServiceA调用完ServiceB后，ConsumerContextFilter的invoke方法会清除attachements，所以ServiceB可以正常同步调用ServiceC了。

对于上面的问题，解决办法有三个：

1.方法调用两次

ServiceA调用ServiceB的地方写两次一样的调用，这个方法原理就像ServiceB调用ServiceC一样，即清除attachements，这个方法最简单，但是可能对不了解的人来说，这块业务代码写重复了，会不小心删除掉，而且从写代码的角度来说，这个很鸡肋，所以不推荐。

2.修改Dubbo源码

修改AbstractInvoker第137行，改成每次都对async进行实际赋值，

boolean isAsync = getUrl().getMethodParameter(invocation.getMethodName(), Constants.ASYNC_KEY, false);

invocation.setAttachment(Constants.ASYNC_KEY, String.valueOf(isAsync));

3.自定义Filter

实现com.alibaba.dubbo.rpc.Filter，在RpcContext中清除这个async，

@Activate(group = {Constants.PROVIDER})

public class AsyncFilter implements Filter {

@Override

public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {

RpcContext.getContext().getAttachments().remove(Constants.ASYNC_KEY);

return invoker.invoke(invocation);

}

}

同时在src/main/resources/META-INF/dubbo/下添加com.alibaba.dubbo.rpc.Filter文件，内容文件如下：

asyncFilter=com.abc.filter.AsyncFilter