src/org/redkale/net/nio/NioThreadGroup.java

@@ -28,11 +28,13 @@ class NioThreadGroup {
         return timeoutExecutor.schedule(callable, delay, unit);
     }
 
-    public void interestOps(NioThread ioThread, SelectionKey key, int opt) {
+    public void interestOpsOr(NioThread ioThread, SelectionKey key, int opt) {
+        if (key == null) return;
         if ((key.interestOps() & opt) != 0) return;
         key.interestOps(key.interestOps() | opt);
         if (ioThread.inSameThread()) return;
         //非IO线程中
         key.selector().wakeup();
     }
+
 }

src/org/redkale/net/nio/TcpNioAsyncConnection.java

@@ -325,9 +325,7 @@ class TcpNioAsyncConnection extends AsyncConnection {
                         this.workExecutor.execute(() -> handler.completed(totalCount0, attach));
                     }
                 }
-                if (readKey != null) {
+                if (readKey != null) readKey.interestOps(readKey.interestOps() & ~SelectionKey.OP_READ);
-                    readKey.interestOps(readKey.interestOps() & ~SelectionKey.OP_READ);
-                }
             } else if (readKey == null) {
                 ioThread.register(selector -> {
                     try {
@@ -347,7 +345,7 @@ class TcpNioAsyncConnection extends AsyncConnection {
                     }
                 });
             } else {
-                ioGroup.interestOps(ioThread, readKey, SelectionKey.OP_READ);
+                ioGroup.interestOpsOr(ioThread, readKey, SelectionKey.OP_READ);
             }
         } catch (Exception e) {
             CompletionHandler<Integer, ByteBuffer> handler = this.readCompletionHandler;
@@ -428,7 +426,7 @@ class TcpNioAsyncConnection extends AsyncConnection {
                     }
                 });
             } else {
-                ioGroup.interestOps(ioThread, writeKey, SelectionKey.OP_WRITE);
+                ioGroup.interestOpsOr(ioThread, writeKey, SelectionKey.OP_WRITE);
             }
         } catch (IOException e) {
             CompletionHandler<Integer, Object> handler = this.writeCompletionHandler;

src/org/redkale/net/nio/TcpNioProtocolServer.java

@@ -25,6 +25,8 @@ public class TcpNioProtocolServer extends ProtocolServer {
 
     private ServerSocketChannel serverChannel;
 
+    private NioThreadGroup ioGroup;
+
     public TcpNioProtocolServer(Context context) {
         super(context);
     }