[CELEBORN-871] Init PartitionReader OpenStream failed won't show loca…

…tion info
apache · Aug 2, 2023 · ec5687b · ec5687b
1 parent 6cd1355
commit ec5687b
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/client/src/main/java/org/apache/celeborn/client/read/WorkerPartitionReader.java b/client/src/main/java/org/apache/celeborn/client/read/WorkerPartitionReader.java
@@ -101,14 +101,14 @@ public void onFailure(int chunkIndex, Throwable e) {
     TransportClient client = null;
     try {
       client = clientFactory.createClient(location.getHost(), location.getFetchPort());
-    } catch (InterruptedException ie) {
-      logger.error("PartitionReader thread interrupted while creating client.");
-      throw ie;
+      OpenStream openBlocks =
+          new OpenStream(shuffleKey, location.getFileName(), startMapIndex, endMapIndex);
+      ByteBuffer response = client.sendRpcSync(openBlocks.toByteBuffer(), fetchTimeoutMs);
+      streamHandle = (StreamHandle) Message.decode(response);
+    } catch (IOException | InterruptedException e) {
+      logger.error("PartitionReader thread interrupted while creating client.", e);
+      throw e;
     }
-    OpenStream openBlocks =
-        new OpenStream(shuffleKey, location.getFileName(), startMapIndex, endMapIndex);
-    ByteBuffer response = client.sendRpcSync(openBlocks.toByteBuffer(), fetchTimeoutMs);
-    streamHandle = (StreamHandle) Message.decode(response);
 
     this.location = location;
     this.clientFactory = clientFactory;