Netty4.0.24.Final 版本中 IdleStateHandler 使用时的局限性

时间:2024-08-08 20:35:32

使用Netty在客户端和服务端建立通讯通道,一般来说,一个连接可能很久没有访问,由于各种各样的网络问题导致连接已经失效,客户端再次发送请求时会产生连接异常。

基于这个原因,需要在客户端和服务端之间建立ping-pong的心跳机制,我的本意是想通过IdleStateHandler这个netty提供的工具来自动发现连接空闲状态,却出现了以下问题:

1.客户端和服务端使用 IdleStateHandler 时,原本的请求响应机制失效

2.IdleStateHandler的空闲通知功能正常,但是却不准确

最开始排查这个问题的原因是一头雾水,通过断点和分析程序已经走到哪个位置,最终确定IdleStateHandler在输出服务端的响应到channel时,出现了问题

    /**
* 发送服务端的响应
* @param messageId
* @param channel
* @param messageResult
*/
protected void sendResponse(long messageId, Channel channel, Object messageResult) {
MsgHeader msgHeader = new MsgHeader(Constants.RESPONSE_MSG);
if(messageResult != null) {
msgHeader.setClz(messageResult.getClass().getName());
}else {
msgHeader.setClz(Constants.NULL_RESULT_CLASS);
} ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setReceiveTime(System.currentTimeMillis());
responseMsg.setResponse(messageResult);
responseMsg.setMsgHeader(msgHeader);
responseMsg.getMsgHeader().setMsgId(messageId); //如果使用voidPromise,则无法和IdleStateHandler同时使用,因为它会触发voidPromise的addListener(...)操作,从而导致write失败
channel.writeAndFlush(responseMsg, channel.voidPromise());
}

在处理完请求进行响应输出时,最后一行在写响应时,使用了空的promise,最终会导致IdleStateHandler的write方法处获得这个空的Promise,进而导致异常

    @Override
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
promise.addListener(new ChannelFutureListener() {
@Override
public void operationComplete(ChannelFuture future) throws Exception {
lastWriteTime = System.nanoTime();
firstWriterIdleEvent = firstAllIdleEvent = true;
}
});
ctx.write(msg, promise);
}

空的Promise的addListener实现方式是简单粗暴的抛出异常,这就导致末行的ctx.write(msg, promise)永远也不能执行到。

final class VoidChannelPromise extends AbstractFuture<Void> implements ChannelPromise {
... @Override
public VoidChannelPromise addListener(GenericFutureListener<? extends Future<? super Void>> listener) {
fail();
return this;
} private static void fail() {
throw new IllegalStateException("void future");
}
... }

找到这个原因后,立即修改了Server端Handler的发送方法

    /**
* 发送服务端的响应
* @param messageId
* @param channel
* @param messageResult
*/
protected void sendResponse(long messageId, Channel channel, Object messageResult) {
MsgHeader msgHeader = new MsgHeader(Constants.RESPONSE_MSG);
if(messageResult != null) {
msgHeader.setClz(messageResult.getClass().getName());
}else {
msgHeader.setClz(Constants.NULL_RESULT_CLASS);
} ResponseMsg responseMsg = new ResponseMsg();
responseMsg.setReceiveTime(System.currentTimeMillis());
responseMsg.setResponse(messageResult);
responseMsg.setMsgHeader(msgHeader);
responseMsg.getMsgHeader().setMsgId(messageId); //如果使用voidPromise,则无法和IdleStateHandler同时使用,因为它会触发voidPromise的addListener(...)操作,从而导致write失败
channel.writeAndFlush(responseMsg, channel.newPromise());
}

末行new一个默认的Promise,执行addListener不会产生异常

这样server端的问题就解决了

然后client端却无法解决这个问题,原因时client端接收响应后,netty内部会new一个VoidPromise,这就导致了IdleStateHandler放置于客户端时无论如何都不能正常使用,因为一旦触发write方法就直接异常了,目前还没有找到这两者同时存在的方案。这是Netty4.0.24.Final版本的硬伤,好在的是,Netty4的最新版本已经fix了这个问题

        <dependency>
<groupId>io.netty</groupId>
<artifactId>netty-all</artifactId>
<version>4.1.12.Final</version>
</dependency>

IdleStateHandler的write方法已经对此做出了修正

    @Override
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
// Allow writing with void promise if handler is only configured for read timeout events.
if (writerIdleTimeNanos > 0 || allIdleTimeNanos > 0) {
ChannelPromise unvoid = promise.unvoid(); //此处保证返回的一定是非void的
unvoid.addListener(writeListener);
ctx.write(msg, unvoid);
} else {
ctx.write(msg, promise);
}
}

总结,一个复杂的框架无论如何会有一些毛病,就像业务系统中的一个bug,两个东西不是同一个人开发的,就会出现使用时的异常情况,找到这个原因进行分析,解决的思路也就跃然于纸上。

最终,我把netty4的版本升级到了最新的4.1.12.Final,防止更多这样的问题产生。