src/org/redkale/net/NioTcpAsyncConnection.java

@@ -75,7 +75,7 @@ public class NioTcpAsyncConnection extends AsyncConnection {
 
     public NioTcpAsyncConnection(NioThreadGroup ioGroup, NioThread ioThread, SocketChannel ch,
         SSLContext sslContext, final SocketAddress addr0, AtomicLong livingCounter, AtomicLong closedCounter) {
-        super(ioThread.getBufferPool(), sslContext, livingCounter, closedCounter);
+        super(ioThread.getBufferSupplier(), ioThread.getBufferConsumer(), sslContext, livingCounter, closedCounter);
         this.ioGroup = ioGroup;
         this.ioThread = ioThread;
         this.channel = ch;

src/org/redkale/net/NioTcpPrepareRunner.java

@@ -42,7 +42,7 @@ public class NioTcpPrepareRunner implements Runnable {
             channel.read(new CompletionHandler<Integer, ByteBuffer>() {
                 @Override
                 public void completed(Integer count, ByteBuffer buffer) {
-                    if (response == null) response = ((NioThread) Thread.currentThread()).getResponsePool().get();
+                    if (response == null) response = ((NioThread) Thread.currentThread()).getResponseSupplier().get();
                     if (count < 1) {
                         buffer.clear();
                         channel.setReadBuffer(buffer);

src/org/redkale/net/NioThread.java

@@ -9,7 +9,7 @@ import java.nio.ByteBuffer;
 import java.nio.channels.*;
 import java.util.*;
 import java.util.concurrent.*;
-import java.util.function.Consumer;
+import java.util.function.*;
 import org.redkale.util.*;
 
 /**
@@ -26,21 +26,40 @@ public class NioThread extends WorkThread {
 
     final Selector selector;
 
-    private final ObjectPool<ByteBuffer> bufferPool;
+    private final Supplier<ByteBuffer> bufferSupplier;
 
-    private final ObjectPool<Response> responsePool;
+    private final Consumer<ByteBuffer> bufferConsumer;
+
+    private final Supplier<Response> responseSupplier;
+
+    private final Consumer<Response> responseConsumer;
 
     private final ConcurrentLinkedQueue<Consumer<Selector>> registers = new ConcurrentLinkedQueue<>();
 
     private boolean closed;
 
     public NioThread(String name, ExecutorService workExecutor, Selector selector,
-        ObjectPool<ByteBuffer> bufferPool, ObjectPool<Response> responsePool) {
+        ObjectPool<ByteBuffer> unsafeBufferPool, ObjectPool<ByteBuffer> safeBufferPool,
+        ObjectPool<Response> unsafeResponsePool, ObjectPool<Response> safeResponsePool) {
         super(name, workExecutor, null);
         this.selector = selector;
-        this.bufferPool = bufferPool;
-        this.responsePool = responsePool;
         this.setDaemon(true);
+        this.bufferSupplier = () -> inCurrThread() ? unsafeBufferPool.get() : safeBufferPool.get();
+        this.bufferConsumer = (v) -> {
+            if (inCurrThread()) {
+                unsafeBufferPool.accept(v);
+            } else {
+                safeBufferPool.accept(v);
+            }
+        };
+        this.responseSupplier = () -> inCurrThread() ? unsafeResponsePool.get() : safeResponsePool.get();
+        this.responseConsumer = (v) -> {
+            if (inCurrThread()) {
+                unsafeResponsePool.accept(v);
+            } else {
+                safeResponsePool.accept(v);
+            }
+        };
     }
 
     public void register(Consumer<Selector> consumer) {
@@ -48,12 +67,20 @@ public class NioThread extends WorkThread {
         selector.wakeup();
     }
 
-    public ObjectPool<ByteBuffer> getBufferPool() {
-        return bufferPool;
+    public Supplier<ByteBuffer> getBufferSupplier() {
+        return bufferSupplier;
     }
 
-    public ObjectPool<Response> getResponsePool() {
-        return responsePool;
+    public Consumer<ByteBuffer> getBufferConsumer() {
+        return bufferConsumer;
+    }
+
+    public Supplier<Response> getResponseSupplier() {
+        return responseSupplier;
+    }
+
+    public Consumer<Response> getResponseConsumer() {
+        return responseConsumer;
     }
 
     @Override

src/org/redkale/net/NioThreadGroup.java

@@ -32,18 +32,18 @@ public class NioThreadGroup {
     private ScheduledThreadPoolExecutor timeoutExecutor;
 
     public NioThreadGroup(final String serverName, ExecutorService workExecutor, int iothreads,
-        ObjectPool<ByteBuffer> bufferPool, ObjectPool<Response> responsePool) throws IOException {
+        ObjectPool<ByteBuffer> safeBufferPool, ObjectPool<Response> safeResponsePool) throws IOException {
         this.threads = new NioThread[Math.max(iothreads, 1)];
         for (int i = 0; i < this.threads.length; i++) {
-            ObjectPool<ByteBuffer> threadBufferPool = ObjectPool.createUnsafePool(bufferPool.getCreatCounter(),
-                bufferPool.getCycleCounter(), 8,
-                bufferPool.getCreator(), bufferPool.getPrepare(), bufferPool.getRecycler());
+            ObjectPool<ByteBuffer> unsafeBufferPool = ObjectPool.createUnsafePool(safeBufferPool.getCreatCounter(),
+                safeBufferPool.getCycleCounter(), 8,
+                safeBufferPool.getCreator(), safeBufferPool.getPrepare(), safeBufferPool.getRecycler());
 
-            ObjectPool<Response> threadResponsePool = ObjectPool.createUnsafePool(responsePool.getCreatCounter(),
-                responsePool.getCycleCounter(), 8,
-                responsePool.getCreator(), responsePool.getPrepare(), responsePool.getRecycler());
+            ObjectPool<Response> unsafeResponsePool = ObjectPool.createUnsafePool(safeResponsePool.getCreatCounter(),
+                safeResponsePool.getCycleCounter(), 8,
+                safeResponsePool.getCreator(), safeResponsePool.getPrepare(), safeResponsePool.getRecycler());
             String name = "Redkale-" + serverName + "-ServletThread" + "-" + (i >= 9 ? (i + 1) : ("0" + (i + 1)));
-            this.threads[i] = new NioThread(name, workExecutor, Selector.open(), threadBufferPool, threadResponsePool);
+            this.threads[i] = new NioThread(name, workExecutor, Selector.open(), unsafeBufferPool, safeBufferPool, unsafeResponsePool, safeResponsePool);
         }
         this.timeoutExecutor = (ScheduledThreadPoolExecutor) Executors.newScheduledThreadPool(1, (Runnable r) -> {
             Thread t = new Thread(r);