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

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

diff --git a/src/org/redkale/net/TcpNioAsyncConnection.java b/src/org/redkale/net/TcpNioAsyncConnection.java
index 00d2223f9..744c2242b 100644
--- a/src/org/redkale/net/TcpNioAsyncConnection.java
+++ b/src/org/redkale/net/TcpNioAsyncConnection.java
@@ -472,6 +472,7 @@ public class TcpNioAsyncConnection extends AsyncConnection {
             }
 
             if (totalCount > 0 || !hasRemain) {
+                if (writeKey != null) writeKey.interestOps(writeKey.interestOps() & ~SelectionKey.OP_WRITE);
                 CompletionHandler<Integer, Object> handler = this.writeCompletionHandler;
                 Object attach = this.writeAttachment;
                 clearWrite();
@@ -527,4 +528,10 @@ public class TcpNioAsyncConnection extends AsyncConnection {
         this.writeLength = 0;
         this.writePending = false; //必须放最后
     }
+
+    @Override
+    public final void close() throws IOException {
+        super.close();
+        channel.close();
+    }
 }
diff --git a/src/org/redkale/net/nio/NioThread.java b/src/org/redkale/net/nio/NioThread.java
index ee6d5e0c6..b53a68f6d 100644
--- a/src/org/redkale/net/nio/NioThread.java
+++ b/src/org/redkale/net/nio/NioThread.java
@@ -68,6 +68,7 @@ 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();