fix compile

Signed-off-by: before-Sunrise <[email protected]>
StarRocks · Nov 7, 2024 · dd678ef · dd678ef
1 parent 629ca19
commit dd678ef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/be/src/exec/pipeline/sort/local_partition_topn_context.cpp b/be/src/exec/pipeline/sort/local_partition_topn_context.cpp
@@ -185,9 +185,8 @@ Status LocalPartitionTopnContext::push_one_chunk_to_partitioner(RuntimeState* st
                 if (_enable_pre_agg) {
                     AggDataPtr agg_states = _mem_pool->allocate_aligned(_pre_agg->_agg_states_total_size,
                                                                         _pre_agg->_max_agg_state_align_size);
-                    _pre_agg->_managed_fn_states.emplace_back(
-                            std::make_unique<ManagedFunctionStates<LocalPartitionTopnContext>>(&_pre_agg->_agg_fn_ctxs,
-                                                                                               agg_states, this));
+                    _pre_agg->_managed_fn_states.emplace_back(std::make_unique<ManagedFunctionStates<PreAggState>>(
+                            &_pre_agg->_agg_fn_ctxs, agg_states, _pre_agg.get()));
                 }
             },
             [this, state](size_t partition_idx, const ChunkPtr& chunk) {

diff --git a/be/src/exec/pipeline/sort/local_partition_topn_context.h b/be/src/exec/pipeline/sort/local_partition_topn_context.h
@@ -59,7 +59,7 @@ struct PreAggState {
     std::vector<std::vector<const Column*>> _agg_input_raw_columns;
     std::vector<FunctionTypes> _agg_fn_types;
     // every partition has one Agg State
-    std::vector<ManagedFunctionStatesPtr<LocalPartitionTopnContext>> _managed_fn_states;
+    std::vector<ManagedFunctionStatesPtr<PreAggState>> _managed_fn_states;
 };
 
 // LocalPartitionTopnContext is the bridge of each pair of LocalPartitionTopn{Sink/Source}Operators