From 89b4eab3b6f3c2bea0f4cc40eb8b8fa09fb53817 Mon Sep 17 00:00:00 2001
From: Keyong Zhou <zhouky@apache.org>
Date: Fri, 17 Feb 2023 10:01:19 +0800
Subject: [PATCH] [CELEBORN-309] Fix some potential concurrent issues in
 InFlightRequestTracker (#1243)

---
 .../apache/celeborn/common/write/InFlightRequestTracker.java   | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/common/src/main/java/org/apache/celeborn/common/write/InFlightRequestTracker.java b/common/src/main/java/org/apache/celeborn/common/write/InFlightRequestTracker.java
index c8c4d8458..3b3e8da6b 100644
--- a/common/src/main/java/org/apache/celeborn/common/write/InFlightRequestTracker.java
+++ b/common/src/main/java/org/apache/celeborn/common/write/InFlightRequestTracker.java
@@ -60,9 +60,6 @@ public class InFlightRequestTracker {
   public void removeBatch(int batchId, String hostAndPushPort) {
     Set<Integer> batchIdSet = inflightBatchesPerAddress.get(hostAndPushPort);
     batchIdSet.remove(batchId);
-    if (batchIdSet.size() == 0) {
-      inflightBatchesPerAddress.remove(hostAndPushPort);
-    }
   }
 
   public void onSuccess(String hostAndPushPort) {