优化fastHandler

src/main/java/org/redkale/net/AsyncConnection.java

@@ -224,7 +224,9 @@ public abstract class AsyncConnection implements Channel, AutoCloseable {
 
     public abstract void setWriteTimeoutSeconds(int writeTimeoutSeconds);
 
-    public abstract <A> void fastWrite(byte[] data, A attachment, CompletionHandler<Integer, ? super A> handler);
+    public abstract <A> AsyncConnection fastHandler(CompletionHandler<Integer, ? super A> handler);
+
+    public abstract <A> void fastWrite(byte[] data);
 
     protected abstract void readRegisterImpl(CompletionHandler<Integer, ByteBuffer> handler);
 
@@ -240,10 +242,6 @@ public abstract class AsyncConnection implements Channel, AutoCloseable {
         read(handler);
     }
 
-    public final <A> void fastWrite(byte[] data, CompletionHandler<Integer, ? super A> handler) {
-        fastWrite(data, null, handler);
-    }
-
     public final void startReadInIOThread(CompletionHandler<Integer, ByteBuffer> handler) {
         if (inCurrReadThread()) {
             startRead(handler);

src/main/java/org/redkale/net/AsyncNioConnection.java

@@ -85,6 +85,8 @@ abstract class AsyncNioConnection extends AsyncConnection {
 
     protected SelectionKey writeKey;
 
+    protected CompletionHandler<Integer, Object> writeFastHandler;
+
     public AsyncNioConnection(boolean clientMode, AsyncIOGroup ioGroup, AsyncIOThread ioReadThread,
         AsyncIOThread ioWriteThread, final int bufferCapacity, SSLBuilder sslBuilder, SSLContext sslContext) {
         super(clientMode, ioGroup, ioReadThread, ioWriteThread, bufferCapacity, sslBuilder, sslContext);
@@ -115,6 +117,13 @@ abstract class AsyncNioConnection extends AsyncConnection {
         return this.writeTimeoutSeconds;
     }
 
+    @Override
+    public <A> AsyncConnection fastHandler(CompletionHandler<Integer, ? super A> handler) {
+        Objects.requireNonNull(handler);
+        this.writeFastHandler = (CompletionHandler) handler;
+        return this;
+    }
+
     @Override
     protected void startHandshake(final Consumer<Throwable> callback) {
         ioReadThread.register(t -> super.startHandshake(callback));
@@ -291,18 +300,19 @@ abstract class AsyncNioConnection extends AsyncConnection {
     }
 
     @Override
-    public <A> void fastWrite(byte[] data, A attachment, CompletionHandler<Integer, ? super A> handler) {
+    public <A> void fastWrite(byte[] data) {
+        CompletionHandler<Integer, ? super A> handler = this.writeFastHandler;
+        Objects.requireNonNull(data);
+        Objects.requireNonNull(handler, "fastHandler is null");
         if (!this.isConnected()) {
             handler.failed(new NotYetConnectedException(), null);
             return;
         }
-        Objects.requireNonNull(data);
-        Objects.requireNonNull(handler);
         this.writePending = true;
         this.fastWriteQueue.offer(data);
         this.fastWriteCount.incrementAndGet();
         this.writeCompletionHandler = (CompletionHandler) handler;
-        this.writeAttachment = attachment;
+        this.writeAttachment = null;
         try {
             if (writeKey == null) {
                 ioWriteThread.register(selector -> {

src/main/java/org/redkale/net/client/ClientConnection.java

@@ -102,7 +102,7 @@ public abstract class ClientConnection<R extends ClientRequest, P> implements Co
         this.index = index;
         this.connEntry = index >= 0 ? null : client.connAddrEntrys.get(channel.getRemoteAddress());
         this.respWaitingCounter = index >= 0 ? client.connRespWaitings[index] : this.connEntry.connRespWaiting;
-        this.channel = channel.beforeCloseListener(this);
+        this.channel = channel.beforeCloseListener(this).fastHandler(writeHandler);
         this.writeBuffer = channel.pollWriteBuffer();
     }
 
@@ -180,7 +180,7 @@ public abstract class ClientConnection<R extends ClientRequest, P> implements Co
                 pauseWriting.set(true);
                 currHalfWriteFuture = respFuture;
             }
-            channel.fastWrite(array.getBytes(), writeHandler);
+            channel.fastWrite(array.getBytes());
         } else { //旧方式
             //发送请求数据包
             writeArray.clear();
@@ -221,7 +221,7 @@ public abstract class ClientConnection<R extends ClientRequest, P> implements Co
                 }
             }
         }
-        channel.fastWrite(array.getBytes(), writeHandler);
+        channel.fastWrite(array.getBytes());
     }
 
     //发送半包和积压的请求数据包