From a973834325cc000a346f4f5304918640b615e915 Mon Sep 17 00:00:00 2001
From: Redkale <redkale@qq.com>
Date: Sat, 14 Jan 2023 11:58:15 +0800
Subject: [PATCH] =?UTF-8?q?ProtocolCodec=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../org/redkale/net/AsyncNioTcpProtocolServer.java  | 13 +++++++------
 src/main/java/org/redkale/net/ProtocolCodec.java    | 13 +++++++++++--
 2 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/src/main/java/org/redkale/net/AsyncNioTcpProtocolServer.java b/src/main/java/org/redkale/net/AsyncNioTcpProtocolServer.java
index 778287f15..5bb9122de 100644
--- a/src/main/java/org/redkale/net/AsyncNioTcpProtocolServer.java
+++ b/src/main/java/org/redkale/net/AsyncNioTcpProtocolServer.java
@@ -9,7 +9,7 @@ import java.io.IOException;
 import java.net.*;
 import java.nio.ByteBuffer;
 import java.nio.channels.*;
-import java.util.*;
+import java.util.Set;
 import java.util.concurrent.atomic.LongAdder;
 import java.util.function.*;
 import java.util.logging.Level;
@@ -133,17 +133,17 @@ class AsyncNioTcpProtocolServer extends ProtocolServer {
                 final AsyncIOThread[] ioWriteThreads = ioGroup.ioWriteThreads;
                 int threads = ioReadThreads.length;
                 int threadIndex = -1;
+                Set<SelectionKey> keys = null;
                 while (!closed) {
                     try {
                         int count = selector.select();
                         if (count == 0) {
                             continue;
                         }
-                        Set<SelectionKey> keys = selector.selectedKeys();
-                        Iterator<SelectionKey> it = keys.iterator();
-                        while (it.hasNext()) {
-                            SelectionKey key = it.next();
-                            it.remove();
+                        if (keys == null) {
+                            keys = selector.selectedKeys();
+                        }
+                        for (SelectionKey key : keys) {
                             if (key.isAcceptable()) {
                                 if (++threadIndex >= threads) {
                                     threadIndex = 0;
@@ -151,6 +151,7 @@ class AsyncNioTcpProtocolServer extends ProtocolServer {
                                 accept(key, ioReadThreads[threadIndex], ioWriteThreads[threadIndex]);
                             }
                         }
+                        keys.clear();
                     } catch (Throwable t) {
                         server.logger.log(Level.SEVERE, "server accept error", t);
                     }
diff --git a/src/main/java/org/redkale/net/ProtocolCodec.java b/src/main/java/org/redkale/net/ProtocolCodec.java
index 6d1494fb7..fecbee54a 100644
--- a/src/main/java/org/redkale/net/ProtocolCodec.java
+++ b/src/main/java/org/redkale/net/ProtocolCodec.java
@@ -18,14 +18,14 @@ import java.util.logging.Level;
  */
 class ProtocolCodec implements CompletionHandler<Integer, ByteBuffer> {
 
-    private final AsyncConnection channel;
-
     private final Context context;
 
     private final Supplier<Response> responseSupplier;
 
     private final Consumer<Response> responseConsumer;
 
+    private AsyncConnection channel;
+
     private Response resp;
 
     public ProtocolCodec(Context context, Supplier<Response> responseSupplier,
@@ -36,6 +36,15 @@ class ProtocolCodec implements CompletionHandler<Integer, ByteBuffer> {
         this.responseConsumer = responseConsumer;
     }
 
+    public void prepare() {
+    }
+
+    protected boolean recycle() {
+        this.channel = null;
+        this.resp = null;
+        return true;
+    }
+
     public ProtocolCodec response(Response resp) {
         this.resp = resp;
         return this;