本章将NettyClient中发送请求后的阻塞改为异步非阻塞,利用CompletableFutue.get(),将阻塞环节放置到了代理类中,能更大地提高服务器地吞吐量。本章地commit为8dcf034。
使用future获取相应结果具体流程如下:
利用Map缓存未处理地请求
将每一个请求以及其对应的CompletableFuture放入缓存中,之后通过异步地方式去获取请求地结果,首先需要建一个UnprocessedRequest用来缓存request对应地处理结果,为了使得缓存全局唯一,使用单例工厂模式创建UnprocessedRequest:
public class UnprocessedRequest {
private static final Logger logger = LoggerFactory.getLogger(UnprocessedRequest.class);
private static Map<String, CompletableFuture<RpcResponse>> unprocessedMap = new ConcurrentHashMap<>();
public void put(String rpcRequestId,CompletableFuture<RpcResponse> future){
unprocessedMap.put(rpcRequestId,future);
}
public void remove(String rpcRequestId){
unprocessedMap.remove(rpcRequestId);
}
public void complete(RpcResponse rpcResponse){
CompletableFuture<RpcResponse> future = unprocessedMap.remove(rpcResponse.getRequestId());
if(future != null){
future.complete(rpcResponse);
}else{
logger.error("异步获取结果出现错误");
throw new RpcException(RpcError.COMPLETABLE_ERROR);
}
}
}
修改客户端数据获取方式
之前在客户端获取入站处理器处理后的数据是通过AttributeKey全局获取的,但是为了确保拿到的不为null值,因此我们将写入过程阻塞了。我们引入一个CompletableFuture用来存储入站处理器处理后的数据,之后在代理类中异步调用处理结果,这样能够很大程度提高整个服务器的吞吐量:
public class NettyClientHandler extends SimpleChannelInboundHandler<RpcResponse> {
private static final Logger logger = LoggerFactory.getLogger(NettyClientHandler.class);
private final UnprocessedRequest unprocessedRequest = SingletonFactory.getInstance(UnprocessedRequest.class);
@Override
protected void channelRead0(ChannelHandlerContext ctx, RpcResponse msg) throws Exception {
try {
logger.info("客户端获取到服务端返回的信息:{}",msg);
unprocessedRequest.complete(msg);
ctx.channel().close();
} finally {
ReferenceCountUtil.release(msg);
}
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
logger.error("在信息从服务端返回客户端发生错误");
cause.printStackTrace();
ctx.close();
}
}
在代理类中拿到Rpcresponse:
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
RpcRequest rpcRequest = RpcRequest.builder()
.requestId(UUID.randomUUID().toString())
.interfaceName(method.getDeclaringClass().getName())
.methodName(method.getName())
.parameterTypes(method.getParameterTypes())
.parameters(args)
.build();
CompletableFuture<RpcResponse> result = null;
if(client instanceof SocketClient){
return ((RpcResponse) client.sendRequest(rpcRequest,host,port)).getData();
}
if(client instanceof NettyClient){
result = (CompletableFuture<RpcResponse>) client.sendRequest(rpcRequest,host,port);
}
return result.get().getData();
}
客户端的sendRequest即可以设置为非阻塞的:
@Override
public CompletableFuture<RpcResponse> sendRequest(RpcRequest rpcRequest, String host, int port) {
Channel channel = ChannelProvider.get(new InetSocketAddress(host,port),serializer);
CompletableFuture<RpcResponse> future = new CompletableFuture<>();
unprocessedRequest.put(rpcRequest.getRequestId(),future);
channel.writeAndFlush(rpcRequest).addListener((ChannelFutureListener) future1 -> {
if(future1.isSuccess()){
logger.info("成功向服务器发送请求:{}",rpcRequest);
}else{
future1.channel().close();
logger.error("向服务器发送消息失败:{}",future1.cause());
future.completeExceptionally(future1.cause());
}
});
return future;
}
服务端不需要做任何修改
测试
测试代码无任何改动,结果如下:
服务端:
[main] INFO cn.fzzfrjf.core.DefaultServerPublisher - 向接口:[interface cn.fzzfrjf.entity.HelloService]注册服务:cn.fzzfrjf.service.HelloServiceImpl
[main] INFO cn.fzzfrjf.core.DefaultServerPublisher - 向接口:[interface cn.fzzfrjf.entity.ByeService]注册服务:cn.fzzfrjf.service.ByeServiceImpl
[main] WARN io.netty.bootstrap.ServerBootstrap - Unknown channel option 'SO_KEEPALIVE' for channel '[id: 0x21ad7250]'
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250] REGISTERED
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250] BIND: 0.0.0.0/0.0.0.0:10000
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250, L:/0:0:0:0:0:0:0:0:10000] ACTIVE
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250, L:/0:0:0:0:0:0:0:0:10000] READ: [id: 0x4cf92cbc, L:/127.0.0.1:10000 - R:/127.0.0.1:60873]
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250, L:/0:0:0:0:0:0:0:0:10000] READ COMPLETE
[nioEventLoopGroup-3-1] INFO cn.fzzfrjf.core.NettyServerHandler - 服务器接收到请求:cn.fzzfrjf.entity.RpcRequest@73f95071
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250, L:/0:0:0:0:0:0:0:0:10000] READ: [id: 0x4a9347a0, L:/127.0.0.1:10000 - R:/127.0.0.1:60874]
[nioEventLoopGroup-2-1] INFO io.netty.handler.logging.LoggingHandler - [id: 0x21ad7250, L:/0:0:0:0:0:0:0:0:10000] READ COMPLETE
[nioEventLoopGroup-3-2] INFO cn.fzzfrjf.core.NettyServerHandler - 服务器接收到请求:cn.fzzfrjf.entity.RpcRequest@715c8fc
Process finished with exit code 130
服务端:
[nioEventLoopGroup-2-1] INFO cn.fzzfrjf.core.ChannelProvider - 获取channel连接成功,连接到服务器activate.navicat.com:10000
[nioEventLoopGroup-2-1] INFO cn.fzzfrjf.core.NettyClient - 成功向服务器发送请求:cn.fzzfrjf.entity.RpcRequest@10cc8b10
[nioEventLoopGroup-2-1] INFO cn.fzzfrjf.core.NettyClientHandler - 客户端获取到服务端返回的信息:RpcResponse(code=200, requestId=14a50ff6-8e3c-4b44-9e3a-4adfa9f94caa, data=这是id为:2发送的:This is NettyClient!)
[nioEventLoopGroup-2-2] INFO cn.fzzfrjf.core.ChannelProvider - 获取channel连接成功,连接到服务器activate.navicat.com:10000
[nioEventLoopGroup-2-2] INFO cn.fzzfrjf.core.NettyClient - 成功向服务器发送请求:cn.fzzfrjf.entity.RpcRequest@286ffbfa
[nioEventLoopGroup-2-2] INFO cn.fzzfrjf.core.NettyClientHandler - 客户端获取到服务端返回的信息:RpcResponse(code=200, requestId=faa2278e-18a0-4ae4-8b32-6e577881f00a, data=(This is NettyClient!),bye!)
这是id为:2发送的:This is NettyClient!
(This is NettyClient!),bye!
在客户端输出中,我们可以明显看到,发送请求与进行反射调用很明显不是顺序调用了,到达了我们需要的效果。
未完。。。。。。