add shuffleBucket

englefly · Jun 25, 2024 · b87ce36 · b87ce36
1 parent b9bd791
commit b87ce36
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/algebra/Join.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/plans/algebra/Join.java
@@ -42,6 +42,7 @@ enum ShuffleType {
         shuffle,
         broadcast,
         bucketShuffle,
+        shuffleBucket,
         colocated,
         unknown
     }

diff --git a/...ore/src/main/java/org/apache/doris/nereids/trees/plans/physical/AbstractPhysicalJoin.java b/...ore/src/main/java/org/apache/doris/nereids/trees/plans/physical/AbstractPhysicalJoin.java
@@ -307,7 +307,11 @@ public boolean isBroadCastJoin() {
 
     protected Join.ShuffleType shuffleType() {
         if (left() instanceof PhysicalDistribute) {
-            return ShuffleType.shuffle;
+            if (right() instanceof PhysicalDistribute) {
+                return ShuffleType.shuffle;
+            } else {
+                return ShuffleType.shuffleBucket;
+            }
         }
         if (right() instanceof PhysicalDistribute) {
             PhysicalDistribute buildDist = (PhysicalDistribute) right();