From 80853b7d446d97310e4d84d74c5e34d680772e4f Mon Sep 17 00:00:00 2001
From: Josh Rosen <joshrosen@databricks.com>
Date: Tue, 22 Aug 2023 17:36:03 -0700
Subject: [PATCH] Change awaitUninterruptibly() -> await()

---
 .../org/apache/spark/network/client/TransportClientFactory.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/common/network-common/src/main/java/org/apache/spark/network/client/TransportClientFactory.java b/common/network-common/src/main/java/org/apache/spark/network/client/TransportClientFactory.java
index 3df72e65c2ab0..4c1efd69206c5 100644
--- a/common/network-common/src/main/java/org/apache/spark/network/client/TransportClientFactory.java
+++ b/common/network-common/src/main/java/org/apache/spark/network/client/TransportClientFactory.java
@@ -279,7 +279,7 @@ public void initChannel(SocketChannel ch) {
     ChannelFuture cf = bootstrap.connect(address);
 
     if (connCreateTimeout <= 0) {
-      cf.awaitUninterruptibly();
+      cf.await();
       assert cf.isDone();
       if (cf.isCancelled()) {
         throw new IOException(String.format("Connecting to %s cancelled", address));