AI 摘要

文章介绍将Netty RPC客户端接收服务端返回结果的方式,从阻塞的AttributeMap改为CompletableFuture,实现异步非阻塞调用,代码更清晰,使用更简单,并建议限制未处理请求Map容量防OOM。

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

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

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

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

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

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

使用 CompletableFuture 进行优化

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

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

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

UnprocessedRequests.java 存放了未处理的请求(建议限制 map 容器大小,避免未处理请求过多 OOM

现在,在你只需要通过下面的方式就能成功接收到客户端返回的结果:

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