diff --git a/src/org/redkale/source/PoolTcpSource.java b/src/org/redkale/source/PoolTcpSource.java index 647d4b4e4..58d381986 100644 --- a/src/org/redkale/source/PoolTcpSource.java +++ b/src/org/redkale/source/PoolTcpSource.java @@ -51,7 +51,7 @@ public abstract class PoolTcpSource extends PoolSource { return pollAsync().join(); } - protected abstract ByteBuffer reqConnectBuffer(); + protected abstract ByteBuffer reqConnectBuffer(AsyncConnection conn); protected abstract void respConnectBuffer(final ByteBuffer buffer, CompletableFuture future, AsyncConnection conn); @@ -59,7 +59,7 @@ public abstract class PoolTcpSource extends PoolSource { public CompletableFuture pollAsync() { return AsyncConnection.createTCP(group, this.servaddr, this.readTimeoutSeconds, this.writeTimeoutSeconds).thenCompose(conn -> { CompletableFuture future = new CompletableFuture(); - final ByteBuffer buffer = reqConnectBuffer(); + final ByteBuffer buffer = reqConnectBuffer(conn); conn.write(buffer, null, new CompletionHandler() { @Override public void completed(Integer result, Void attachment1) {