[incubator-celeborn] branch main updated: [CELEBORN-264] InFlight requests should not be expired if it's not pushed yet (#1196)

[zh...@apache.org]

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

zhouky 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 02100471 [CELEBORN-264] InFlight requests should not be expired if it's not pushed yet (#1196)
02100471 is described below

commit 021004714bf5c1db4527c686a7d71c4a3a6ca72b
Author: Rex(Hui) An <an...@126.com>
AuthorDate: Wed Feb 1 22:16:55 2023 +0800

    [CELEBORN-264] InFlight requests should not be expired if it's not pushed yet (#1196)
---
 .../org/apache/celeborn/client/write/InFlightRequestTracker.java     | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/client/src/main/java/org/apache/celeborn/client/write/InFlightRequestTracker.java b/client/src/main/java/org/apache/celeborn/client/write/InFlightRequestTracker.java
index f286c799..60bd6c16 100644
--- a/client/src/main/java/org/apache/celeborn/client/write/InFlightRequestTracker.java
+++ b/client/src/main/java/org/apache/celeborn/client/write/InFlightRequestTracker.java
@@ -181,7 +181,8 @@ public class InFlightRequestTracker {
                     .values()
                     .forEach(
                         info -> {
-                          if (currentTime - info.pushTime > pushTimeoutMs) {
+                          if (info.pushTime != -1
+                              && (currentTime - info.pushTime > pushTimeoutMs)) {
                             if (info.callback != null) {
                               info.channelFuture.cancel(true);
                               info.callback.onFailure(
@@ -214,7 +215,7 @@ public class InFlightRequestTracker {
 
   static class BatchInfo {
     ChannelFuture channelFuture;
-    long pushTime;
+    long pushTime = -1;
     RpcResponseCallback callback;
   }
 }