[incubator-celeborn] branch main updated: [CELEBORN-477][FOLLOWUP] Report failed partition to flink framework. (#1393)

[et...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

ethanfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-celeborn.git


The following commit(s) were added to refs/heads/main by this push:
     new cbe65a7e2 [CELEBORN-477][FOLLOWUP] Report failed partition to flink framework. (#1393)
cbe65a7e2 is described below

commit cbe65a7e2e25b81601c996fa12ac897c3b4ae9fc
Author: Ethan Feng <et...@apache.org>
AuthorDate: Tue Mar 28 18:56:39 2023 +0800

    [CELEBORN-477][FOLLOWUP] Report failed partition to flink framework. (#1393)
---
 .../org/apache/celeborn/plugin/flink/RemoteBufferStreamReader.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/client-flink/flink-1.14/src/main/java/org/apache/celeborn/plugin/flink/RemoteBufferStreamReader.java b/client-flink/flink-1.14/src/main/java/org/apache/celeborn/plugin/flink/RemoteBufferStreamReader.java
index 51ae02941..0f9c2c2a1 100644
--- a/client-flink/flink-1.14/src/main/java/org/apache/celeborn/plugin/flink/RemoteBufferStreamReader.java
+++ b/client-flink/flink-1.14/src/main/java/org/apache/celeborn/plugin/flink/RemoteBufferStreamReader.java
@@ -80,14 +80,14 @@ public class RemoteBufferStreamReader extends CreditListener {
         };
   }
 
-  public void open(int initialCredit) throws IOException {
+  public void open(int initialCredit) {
     try {
       this.bufferStream =
           client.readBufferedPartition(
               applicationId, shuffleId, partitionId, subPartitionIndexStart, subPartitionIndexEnd);
       bufferStream.open(
           RemoteBufferStreamReader.this::requestBuffer, initialCredit, client, messageConsumer);
-    } catch (InterruptedException e) {
+    } catch (Exception e) {
       logger.warn("Failed to open stream and report to flink framework. ", e);
       messageConsumer.accept(new TransportableError(0L, e));
     }