[Enhancement] push runtime in_filter through agg node (#52625)

Signed-off-by: stephen <[email protected]>
StarRocks · Nov 5, 2024 · f8d2356 · f8d2356
1 parent a5373f9
commit f8d2356
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/be/src/exec/aggregate/aggregate_base_node.cpp b/be/src/exec/aggregate/aggregate_base_node.cpp
@@ -67,6 +67,24 @@ void AggregateBaseNode::close(RuntimeState* state) {
     ExecNode::close(state);
 }
 
+void AggregateBaseNode::push_down_tuple_slot_mappings(RuntimeState* state,
+                                                      const std::vector<TupleSlotMapping>& parent_mappings) {
+    _tuple_slot_mappings = parent_mappings;
+
+    DCHECK(_tuple_ids.size() == 1);
+    for (auto& expr_ctx : _group_by_expr_ctxs) {
+        if (expr_ctx->root()->is_slotref()) {
+            auto ref = dynamic_cast<ColumnRef*>(expr_ctx->root());
+            DCHECK(ref != nullptr);
+            _tuple_slot_mappings.emplace_back(ref->tuple_id(), ref->slot_id(), _tuple_ids[0], ref->slot_id());
+        }
+    }
+
+    for (auto& child : _children) {
+        child->push_down_tuple_slot_mappings(state, _tuple_slot_mappings);
+    }
+}
+
 void AggregateBaseNode::push_down_join_runtime_filter(RuntimeState* state, RuntimeFilterProbeCollector* collector) {
     // accept runtime filters from parent if possible.
     _runtime_filter_collector.push_down(state, id(), collector, _tuple_ids, _local_rf_waiting_set);

diff --git a/be/src/exec/aggregate/aggregate_base_node.h b/be/src/exec/aggregate/aggregate_base_node.h
@@ -29,6 +29,8 @@ class AggregateBaseNode : public ExecNode {
     Status prepare(RuntimeState* state) override;
     void close(RuntimeState* state) override;
     void push_down_join_runtime_filter(RuntimeState* state, RuntimeFilterProbeCollector* collector) override;
+    void push_down_tuple_slot_mappings(RuntimeState* state,
+                                       const std::vector<TupleSlotMapping>& parent_mappings) override;
 
 protected:
     const TPlanNode& _tnode;