2

apache · Sep 2, 2024 · d0293e1 · d0293e1
1 parent fe0e3f7
commit d0293e1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/be/src/pipeline/pipeline_fragment_context.cpp b/be/src/pipeline/pipeline_fragment_context.cpp
@@ -1181,6 +1181,10 @@ Status PipelineFragmentContext::_create_operator(ObjectPool* pool, const TPlanNo
  break;
  }
  case TPlanNodeType::GROUP_COMMIT_SCAN_NODE: {
+#ifndef NDEBUG
+ DCHECK(state->get_query_ctx() != nullptr);
+ state->get_query_ctx()->query_mem_tracker->is_group_commit_load = true;
+#endif
  op.reset(new GroupCommitOperatorX(pool, tnode, next_operator_id(), descs, _num_instances));
  RETURN_IF_ERROR(cur_pipe->add_operator(op));
  if (request.__isset.parallel_instances) {

diff --git a/be/src/runtime/query_context.cpp b/be/src/runtime/query_context.cpp
@@ -138,11 +138,6 @@ void QueryContext::_init_query_mem_tracker() {
  if (_query_options.__isset.is_report_success && _query_options.is_report_success) {
  query_mem_tracker->enable_print_log_usage();
  }
-#ifndef NDEBUG
- if (_query_source == QuerySource::GROUP_COMMIT_LOAD) {
- query_mem_tracker->is_group_commit_load = true;
- }
-#endif
 }
 
 QueryContext::~QueryContext() {