From dcf0fecdb077bfeafb8c28371743140ec145692e Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Fri, 26 Jun 2020 00:38:24 +0800
Subject: [PATCH]

---
 src/org/redkale/net/TcpNioAsyncConnection.java | 2 +-
 src/org/redkale/net/nio/NioThread.java         | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/org/redkale/net/TcpNioAsyncConnection.java b/src/org/redkale/net/TcpNioAsyncConnection.java
index c2eb11f0d..00d2223f9 100644
--- a/src/org/redkale/net/TcpNioAsyncConnection.java
+++ b/src/org/redkale/net/TcpNioAsyncConnection.java
@@ -390,6 +390,7 @@ public class TcpNioAsyncConnection extends AsyncConnection {
                 totalCount += readCount;
             }
             if (totalCount != 0 || !hasRemain) {
+                if (readKey != null) readKey.interestOps(readKey.interestOps() & ~SelectionKey.OP_READ);
                 CompletionHandler<Integer, ByteBuffer> handler = this.readCompletionHandler;
                 ByteBuffer attach = this.readByteBuffer;
                 clearRead();
@@ -401,7 +402,6 @@ public class TcpNioAsyncConnection extends AsyncConnection {
                         this.workExecutor.execute(() -> handler.completed(totalCount0, attach));
                     }
                 }
-                if (readKey != null) readKey.interestOps(readKey.interestOps() & ~SelectionKey.OP_READ);
             } else if (readKey == null) {
                 ioThread.register(selector -> {
                     try {
diff --git a/src/org/redkale/net/nio/NioThread.java b/src/org/redkale/net/nio/NioThread.java
index 10308bfde..ee6d5e0c6 100644
--- a/src/org/redkale/net/nio/NioThread.java
+++ b/src/org/redkale/net/nio/NioThread.java
@@ -68,7 +68,6 @@ public class NioThread extends Thread {
                     it.remove();
                     TcpNioAsyncConnection conn = (TcpNioAsyncConnection) key.attachment();
                     if (key.isWritable()) {
-                        key.interestOps(key.interestOps() & ~SelectionKey.OP_WRITE);
                         conn.doWrite();
                     } else if (key.isReadable()) {
                         conn.doRead();