fix

apache · Oct 15, 2024 · f4690e3 · f4690e3
1 parent eef4ee3
commit f4690e3
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/...core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/DistributePlanner.java b/...core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/DistributePlanner.java
@@ -44,15 +44,14 @@
 import com.google.common.collect.LinkedHashMultimap;
 import com.google.common.collect.ListMultimap;
 import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
 import com.google.common.collect.SetMultimap;
 
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Objects;
-import java.util.function.Function;
-import java.util.stream.Collectors;
 
 /** DistributePlanner */
 public class DistributePlanner {
@@ -220,12 +219,10 @@ private List<AssignedJob> getFirstInstancePerShareScan(PipelineDistributedPlan p
     }
 
     private List<AssignedJob> getFirstInstancePerWorker(List<AssignedJob> instances) {
-        Map<DistributedPlanWorker, AssignedJob> firstInstancePerWorker = instances.stream()
-                .collect(Collectors.toMap(
-                        AssignedJob::getAssignedWorker,
-                        Function.identity(),
-                        (firstInstance, otherInstance) -> firstInstance)
-                );
+        Map<DistributedPlanWorker, AssignedJob> firstInstancePerWorker = Maps.newLinkedHashMap();
+        for (AssignedJob instance : instances) {
+            firstInstancePerWorker.putIfAbsent(instance.getAssignedWorker(), instance);
+        }
         return Utils.fastToImmutableList(firstInstancePerWorker.values());
     }
 }