From abe5de913f3ae92c94cc5ed747b1aecd6074f131 Mon Sep 17 00:00:00 2001 From: 924060929 Date: Thu, 2 Jan 2025 16:38:16 +0800 Subject: [PATCH] fix --- .../worker/job/LocalShuffleBucketJoinAssignedJob.java | 4 ++++ .../trees/plans/distribute/worker/job/StaticAssignedJob.java | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/LocalShuffleBucketJoinAssignedJob.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/LocalShuffleBucketJoinAssignedJob.java index 5c7e6dec37ed30c..270d5ed14ff4352 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/LocalShuffleBucketJoinAssignedJob.java +++ b/fe/fe-core/src/main/java/org/apache/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 getAssignedJoinBucketIndexes() { diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/StaticAssignedJob.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/StaticAssignedJob.java index 77494d621ee4f0c..cccde582d3b211a 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/distribute/worker/job/StaticAssignedJob.java +++ b/fe/fe-core/src/main/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");