andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use DirectBufferPool for NIO socket reads

Browse Source
This commit is contained in:
Trustin Lee 2010-01-28 02:50:31 +00:00
parent aa80596b33
commit bddfdba415
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/org/jboss/netty/channel/socket/nio/NioWorker.java
View File

@ -313,22 +313,38 @@ class NioWorker implements Runnable {
ChannelBuffer buffer =
bufferFactory.getBuffer(predictor.nextReceiveBufferSize());
final ByteBuffer directBuffer;
final boolean fromPool;
if (buffer.isDirect()) {
directBuffer = buffer.toByteBuffer();
fromPool = false;
} else {
directBuffer = directBufferPool.acquire(buffer.writableBytes());
fromPool = true;
}
int ret = 0;
int readBytes = 0;
boolean failure = true;
try {
while ((ret = buffer.writeBytes(ch, buffer.writableBytes())) > 0) {
while ((ret = ch.read(directBuffer)) > 0) {
readBytes += ret;
if (!buffer.writable()) {
if (!directBuffer.hasRemaining()) {
break;
}
}
directBuffer.flip();
buffer.writeBytes(directBuffer);
failure = false;
} catch (AsynchronousCloseException e) {
// Can happen, and does not need a user attention.
} catch (Throwable t) {
fireExceptionCaught(channel, t);
} finally {
if (fromPool) {
directBufferPool.release(directBuffer);
}
}
if (readBytes > 0) {