Update peak for global memory tracker even if it is blocked

azat · azat · commit 24e16b04cc2e · 2025-09-26T12:37:06.000+02:00
diff --git a/src/Common/MemoryTracker.cpp b/src/Common/MemoryTracker.cpp
@@ -255,8 +255,9 @@ AllocationTrace MemoryTracker::allocImpl(Int64 size, bool throw_if_memory_exceed
         if (level == VariableContext::Global)
         {
             /// For global memory tracker always update memory usage.
-            amount.fetch_add(size, std::memory_order_relaxed);
+            Int64 will_be = amount.fetch_add(size, std::memory_order_relaxed);
             rss.fetch_add(size, std::memory_order_relaxed);
+            updatePeak(will_be, /*log_memory_usage=*/ false);
 
             auto metric_loaded = metric.load(std::memory_order_relaxed);
             if (metric_loaded != CurrentMetrics::end())
