fix

apache · Oct 10, 2024 · 275528a · 275528a
1 parent 0b594f4
commit 275528a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/be/src/pipeline/exec/exchange_sink_buffer.cpp b/be/src/pipeline/exec/exchange_sink_buffer.cpp
@@ -165,9 +165,10 @@ Status ExchangeSinkBuffer::add_block(TransmitInfo&& request) {
         if (request.block) {
             RETURN_IF_ERROR(
                     BeExecVersionManager::check_be_exec_version(request.block->be_exec_version()));
+            COUNTER_UPDATE(_parent->memory_used_counter(), request.block->ByteSizeLong());
+            COUNTER_SET(_parent->peak_memory_usage_counter(),
+                        _parent->memory_used_counter()->value());
         }
-        COUNTER_UPDATE(_parent->memory_used_counter(), request.block->ByteSizeLong());
-        COUNTER_SET(_parent->peak_memory_usage_counter(), _parent->memory_used_counter()->value());
         _instance_to_package_queue[ins_id].emplace(std::move(request));
         _total_queue_size++;
         if (_queue_dependency && _total_queue_size > _queue_capacity) {