fix bufferstream

apache · Aug 27, 2023 · 0ce5938 · 0ce5938
1 parent 7740c3b
commit 0ce5938
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/...ommon/src/main/java/org/apache/celeborn/plugin/flink/readclient/CelebornBufferStream.java b/...ommon/src/main/java/org/apache/celeborn/plugin/flink/readclient/CelebornBufferStream.java
@@ -149,16 +149,8 @@ public void close() {
   public void moveToNextPartitionIfPossible(long endedStreamId) {
     if (currentLocationIndex < locations.length) {
       if (currentLocationIndex > 0) {
-        if (endedStreamId == streamId) {
-          logger.debug("Get end streamId {}", endedStreamId);
-          cleanStream(endedStreamId);
-        } else {
-          logger.warn(
-              "Received unexpected stream end, current stream id {} received ended stream id {}",
-              this.streamId,
-              endedStreamId);
-          return;
-        }
+        logger.debug("Get end streamId {}", endedStreamId);
+        cleanStream(endedStreamId);
       }
       try {
         openStreamInternal();