[fix](planner)join node should output required slot from parent node a…

…pache#27526
HappenLee · Nov 24, 2023 · 59efb1b · 59efb1b
1 parent eb250bc
commit 59efb1b
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/planner/JoinNodeBase.java b/fe/fe-core/src/main/java/org/apache/doris/planner/JoinNodeBase.java
@@ -275,6 +275,9 @@ public void initOutputSlotIds(Set<SlotId> requiredSlotIdSet, Analyzer analyzer)
         SlotId firstMaterializedSlotId = null;
         for (TupleDescriptor tupleDescriptor : outputTupleDescList) {
             for (SlotDescriptor slotDescriptor : tupleDescriptor.getSlots()) {
+                if ((requiredSlotIdSet != null && requiredSlotIdSet.contains(slotDescriptor.getId()))) {
+                    slotDescriptor.setIsMaterialized(true);
+                }
                 if (slotDescriptor.isMaterialized()) {
                     if ((requiredSlotIdSet == null || requiredSlotIdSet.contains(slotDescriptor.getId()))) {
                         outputSlotIds.add(slotDescriptor.getId());

diff --git a/regression-test/suites/correctness_p0/test_join_with_projection.groovy b/regression-test/suites/correctness_p0/test_join_with_projection.groovy
@@ -79,4 +79,35 @@ suite("test_join_with_projection") {
     sql """
         drop table if exists test_join_with_projection_outerjoin_B;
     """
+
+    sql """set enable_nereids_planner=false;"""
+    sql """drop table if exists c5870_t;"""
+    sql """create table  c5870_t ( 
+            order_item_seq_id varchar(80) NOT NULL,
+            last_updated_stamp datetime,
+            quantity DECIMAL(18, 2)
+            )
+            ENGINE=OLAP
+            DISTRIBUTED BY HASH(order_item_seq_id) BUCKETS 1
+            PROPERTIES (
+            "replication_allocation" = "tag.location.default: 1",
+            "in_memory" = "false",
+            "storage_format" = "V2"
+            );
+            """
+    sql """with c5870_t_pk_set AS 
+                (SELECT oi.order_item_seq_id,
+                    oi.quantity
+                FROM `c5870_t` oi
+                WHERE order_item_seq_id >= 
+                    (SELECT last_updated_stamp
+                    FROM `c5870_t` )
+                    ORDER BY  oi.last_updated_stamp limit 1 )
+                SELECT order_item_seq_id
+            FROM 
+                (SELECT oia.order_item_seq_id
+                FROM `c5870_t` oia
+                INNER JOIN c5870_t_pk_set lips
+                    ON true ) oia ;"""
+    sql """drop table if exists c5870_t;"""
 }