fix

apache · Dec 17, 2024 · 485d42d · 485d42d
1 parent 6d7c57a
commit 485d42d
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/...che/doris/nereids/trees/plans/distribute/worker/job/UnassignedScanBucketOlapTableJob.java b/...che/doris/nereids/trees/plans/distribute/worker/job/UnassignedScanBucketOlapTableJob.java
@@ -38,6 +38,7 @@
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.ListMultimap;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
@@ -170,24 +171,25 @@ protected void assignLocalShuffleJobs(ScanSource scanSource, int instanceNum, Li
             ConnectContext context, DistributedPlanWorker worker) {
         // only generate one instance to scan all data, in this step
         List<ScanSource> assignJoinBuckets = scanSource.parallelize(
-                scanNodes, instanceNum
+                scanNodes, 1
         );
 
         // one scan range generate multiple instances,
         // different instances reference the same scan source
         int shareScanId = shareScanIdGenerator.getAndIncrement();
 
+        Set<Integer> firstInstanceAssignedJoinBuckets
+                = ((BucketScanSource) assignJoinBuckets.get(0)).bucketIndexToScanNodeToTablets.keySet();
+
         BucketScanSource shareScanSource = (BucketScanSource) scanSource;
         ScanSource emptyShareScanSource = shareScanSource.newEmpty();
 
-        for (int i = 0; i < assignJoinBuckets.size(); i++) {
-            Set<Integer> assignedJoinBuckets
-                    = ((BucketScanSource) assignJoinBuckets.get(i)).bucketIndexToScanNodeToTablets.keySet();
+        for (int i = 0; i < instanceNum; i++) {
             LocalShuffleBucketJoinAssignedJob instance = new LocalShuffleBucketJoinAssignedJob(
                     instances.size(), shareScanId, i > 0,
                     context.nextInstanceId(), this, worker,
                     i == 0 ? shareScanSource : emptyShareScanSource,
-                    Utils.fastToImmutableSet(assignedJoinBuckets)
+                    i == 0 ? Utils.fastToImmutableSet(firstInstanceAssignedJoinBuckets) : ImmutableSet.of()
             );
             instances.add(instance);
         }