From 49c0f334611ab2c72f7f10e897ba3d1794076c1e Mon Sep 17 00:00:00 2001
From: Trustin Lee <trustin@gmail.com>
Date: Thu, 26 Feb 2009 01:29:11 +0000
Subject: [PATCH] Improved AbstractXnioChannelHandler to use
 ReceiveBufferSizePredictor

---
 .../channel/xnio/AbstractXnioChannelHandler.java    | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/jboss/netty/channel/xnio/AbstractXnioChannelHandler.java b/src/main/java/org/jboss/netty/channel/xnio/AbstractXnioChannelHandler.java
index be3fdce618..325f93ac33 100644
--- a/src/main/java/org/jboss/netty/channel/xnio/AbstractXnioChannelHandler.java
+++ b/src/main/java/org/jboss/netty/channel/xnio/AbstractXnioChannelHandler.java
@@ -11,8 +11,10 @@ import java.nio.channels.ScatteringByteChannel;
 import java.util.Queue;
 
 import org.jboss.netty.buffer.ChannelBuffer;
+import org.jboss.netty.buffer.ChannelBufferFactory;
 import org.jboss.netty.buffer.ChannelBuffers;
 import org.jboss.netty.channel.MessageEvent;
+import org.jboss.netty.channel.ReceiveBufferSizePredictor;
 import org.jboss.xnio.IoHandler;
 import org.jboss.xnio.channels.MultipointReadResult;
 import org.jboss.xnio.channels.MultipointReadableMessageChannel;
@@ -39,8 +41,10 @@ public abstract class AbstractXnioChannelHandler implements IoHandler<java.nio.c
 
         boolean closed = false;
 
-        // TODO: Use ReceiveBufferSizePredictor
-        ChannelBuffer buf = c.getConfig().getBufferFactory().getBuffer(2048);
+        ReceiveBufferSizePredictor predictor = c.getConfig().getReceiveBufferSizePredictor();
+        ChannelBufferFactory bufferFactory = c.getConfig().getBufferFactory();
+        ChannelBuffer buf = bufferFactory.getBuffer(predictor.nextReceiveBufferSize());
+
         SocketAddress remoteAddress = null;
         Throwable exception = null;
         if (channel instanceof ScatteringByteChannel) {
@@ -86,8 +90,13 @@ public abstract class AbstractXnioChannelHandler implements IoHandler<java.nio.c
         }
 
         if (buf.readable()) {
+            // Update the predictor.
+            predictor.previousReceiveBufferSize(buf.readableBytes());
+
+            // Fire the event.
             fireMessageReceived(c, buf, remoteAddress);
         }
+
         if (exception != null) {
             fireExceptionCaught(c, exception);
         }