lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ProtocolCodec优化

Browse Source
This commit is contained in:
redkale
2023-01-30 10:49:55 +08:00
parent 4b70e19211
commit d428c58e15
1 changed files with 42 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/main/java/org/redkale/net/ProtocolCodec.java
View File

@@ -24,6 +24,8 @@ class ProtocolCodec implements CompletionHandler<Integer, ByteBuffer> {
private final Consumer<Response> responseConsumer;
private final ReadCompletionHandler readHandler = new ReadCompletionHandler();
private AsyncConnection channel;
private Response resp;
@@ -186,30 +188,47 @@ class ProtocolCodec implements CompletionHandler<Integer, ByteBuffer> {
}
} else {
channel.setReadBuffer(buffer);
channel.read(new CompletionHandler<Integer, ByteBuffer>() {
@Override
public void completed(Integer count, ByteBuffer attachment) {
if (count < 1) {
channel.offerReadBuffer(attachment);
channel.dispose();
return;
}
attachment.flip();
decode(attachment, response, pipelineIndex, lastReq);
}
@Override
public void failed(Throwable exc, ByteBuffer attachment) {
context.prepare.incrIllegalRequestCounter();
channel.offerReadBuffer(attachment);
response.error(exc);
if (exc != null) {
request.context.logger.log(Level.FINER, "Servlet read channel erroneous, force to close channel ", exc);
}
}
});
channel.read(readHandler.prepare(request, response, pipelineIndex, lastReq));
}
}
private class ReadCompletionHandler implements CompletionHandler<Integer, ByteBuffer> {
private Request request;
private Response response;
private int pipelineIndex;
private Request lastReq;
public ReadCompletionHandler prepare(Request request, Response response, int pipelineIndex, Request lastReq) {
this.request = request;
this.response = response;
this.pipelineIndex = pipelineIndex;
this.lastReq = lastReq;
return this;
}
@Override
public void completed(Integer count, ByteBuffer attachment) {
if (count < 1) {
channel.offerReadBuffer(attachment);
channel.dispose();
return;
}
attachment.flip();
decode(attachment, response, pipelineIndex, lastReq);
}
@Override
public void failed(Throwable exc, ByteBuffer attachment) {
context.prepare.incrIllegalRequestCounter();
channel.offerReadBuffer(attachment);
response.error(exc);
if (exc != null) {
request.context.logger.log(Level.FINER, "Servlet read channel erroneous, force to close channel ", exc);
}
}
}
}