(优化)使用CompletableFuture优化接受服务提供端返回结果

使用 AttributeMap 接受服务端返回结果

最开始的时候是通过 AttributeMap 绑定到Channel上实现的,相关代码如下:

NettyClientTransport.java(用来发送 RpcRequest 请求)

11(优化)使用CompletableFuture优化接受服务提供端返回结果 - 图1

NettyClientHandler.java (自定义客户端 ChannelHandler 来处理服务端发过来的数据)

11(优化)使用CompletableFuture优化接受服务提供端返回结果 - 图2

这种是实现的缺点是不清晰,而且你每次都要调用 channel.closeFuture().sync(); 阻塞来手动等待请求返回。

使用 CompletableFuture 进行优化

我使用 CompletableFuture 包装返回结果,对代码进行了重构,重要部分的代码如下:

NettyClientTransport.java(用来发送 RpcRequest 请求)

11(优化)使用CompletableFuture优化接受服务提供端返回结果 - 图3
NettyClientHandler.java (自定义客户端 ChannelHandler 来处理服务端发过来的数据)

11(优化)使用CompletableFuture优化接受服务提供端返回结果 - 图4
UnprocessedRequests.java 存放了未处理的请求(建议限制 map 容器大小,避免未处理请求过多 OOM

11(优化)使用CompletableFuture优化接受服务提供端返回结果 - 图5现在,在你只需要通过下面的方式就能成功接收到客户端返回的结果:

  1. CompletableFuture<RpcResponse> completableFuture = (CompletableFuture<RpcResponse>) clientTransport.sendRpcRequest(rpcRequest);
  2. rpcResponse = completableFuture.get();