fix

924060929 · Jan 2, 2025 · abe5de9 · abe5de9
1 parent a24223f
commit abe5de9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/...he/doris/nereids/trees/plans/distribute/worker/job/LocalShuffleBucketJoinAssignedJob.java b/...he/doris/nereids/trees/plans/distribute/worker/job/LocalShuffleBucketJoinAssignedJob.java
@@ -22,6 +22,7 @@
 import org.apache.doris.thrift.TUniqueId;
 
 import com.amazonaws.services.dynamodbv2.xspec.B;
+import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableSet;
 import org.apache.commons.lang3.StringUtils;
 
@@ -35,6 +36,9 @@ public LocalShuffleBucketJoinAssignedJob(
             TUniqueId instanceId, UnassignedJob unassignedJob,
             DistributedPlanWorker worker, ScanSource scanSource) {
         super(indexInUnassignedJob, receiveDataFromLocal, instanceId, unassignedJob, worker, scanSource);
+        Preconditions.checkArgument(scanSource instanceof BucketScanSource,
+                "scanSource should be of type BucketScanSource"
+        );
     }
 
     public Set<Integer> getAssignedJoinBucketIndexes() {

diff --git a/...in/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/StaticAssignedJob.java b/...in/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/StaticAssignedJob.java
@@ -36,8 +36,8 @@ public class StaticAssignedJob implements AssignedJob {
     private final ScanSource scanSource;
 
     public StaticAssignedJob(
-            int indexInUnassignedJob, TUniqueId instanceId, UnassignedJob unassignedJob, DistributedPlanWorker worker,
-            ScanSource scanSource) {
+            int indexInUnassignedJob, TUniqueId instanceId, UnassignedJob unassignedJob,
+            DistributedPlanWorker worker, ScanSource scanSource) {
         this.indexInUnassignedJob = indexInUnassignedJob;
         this.instanceId = Objects.requireNonNull(instanceId, "instanceId can not be null");
         this.unassignedJob = Objects.requireNonNull(unassignedJob, "unassignedJob can not be null");