[inner-2190] collections adjusted to be thread-safe

(cherry picked from commit 4eac4e1)
actiontech · Nov 10, 2023 · b9eba11 · b9eba11
1 parent 0263eba
commit b9eba11
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/com/actiontech/dble/server/status/LoadDataBatch.java b/src/main/java/com/actiontech/dble/server/status/LoadDataBatch.java
@@ -9,14 +9,18 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import java.util.*;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
 
 
 public final class LoadDataBatch {
     private static final Logger LOGGER = LoggerFactory.getLogger(LoadDataBatch.class);
     private volatile int size;
     private volatile boolean enableBatchLoadData;
-    private Set<String> successFileNames = new HashSet<>();
+    private Set<String> successFileNames = ConcurrentHashMap.newKeySet();
     private int currentNodeSize = 0;
     private Map<String, List<String>> warnings = new HashMap<>();