Sync the future on wait (#43)

GeyserMC · Jan 9, 2022 · d908ccc · d908ccc
1 parent cc069e6
commit d908ccc
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/main/java/com/github/steveice10/packetlib/tcp/TcpClientSession.java b/src/main/java/com/github/steveice10/packetlib/tcp/TcpClientSession.java
@@ -129,9 +129,14 @@ public void initChannel(Channel channel) {
             bootstrap.remoteAddress(remoteAddress);
             bootstrap.localAddress(bindAddress, bindPort);
 
-            bootstrap.connect().addListener((future) -> {
-                if (!future.isSuccess()) {
-                    exceptionCaught(null, future.cause());
+            ChannelFuture future = bootstrap.connect();
+            if (wait) {
+                future.sync();
+            }
+
+            future.addListener((futureListener) -> {
+                if (!futureListener.isSuccess()) {
+                    exceptionCaught(null, futureListener.cause());
                 }
             });
         } catch(Throwable t) {