[CELEBORN-309] Fix some potential concurrent issues in InFlightRequestTracker (#1243)

This commit is contained in:
Keyong Zhou 2023-02-17 10:01:19 +08:00 committed by GitHub
parent 57f775a7e9
commit 89b4eab3b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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) {