Add mem check in fetch partition data (#2385) (#2387)

StarRocks · Dec 23, 2021 · dfdd624 · dfdd624
1 parent 42b97d5
commit dfdd624
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/be/src/exec/vectorized/analytic_node.cpp b/be/src/exec/vectorized/analytic_node.cpp
@@ -296,6 +296,7 @@ Status AnalyticNode::_get_next_for_unbounded_preceding_rows_frame(RuntimeState*
 Status AnalyticNode::_try_fetch_next_partition_data(RuntimeState* state, int64_t* partition_end) {
     *partition_end = _analytor->find_partition_end();
     while (!_analytor->is_partition_finished(*partition_end)) {
+        RETURN_IF_ERROR(state->check_mem_limit("analytic node fetch next partition data"));
         RETURN_IF_ERROR(_fetch_next_chunk(state));
         *partition_end = _analytor->find_partition_end();
     }