Set tcmalloc soft limit when profiling memory allocations (#12379)

Author: abyss7

ydb/core/memory_controller/memory_controller.cpp

@@ -35,7 +35,7 @@ ::NFormatPrivate::THumanReadableSize HumanReadableBytes(ui64 bytes) {
 }
 
 TString HumanReadableBytes(std::optional<ui64> bytes) {
-    return bytes.has_value() ? TString(TStringBuilder() << HumanReadableBytes(bytes.value())) : "none"; 
+    return bytes.has_value() ? TString(TStringBuilder() << HumanReadableBytes(bytes.value())) : "none";
 }
 
 }
@@ -126,6 +126,14 @@ class TMemoryController : public TActorBootstrapped<TMemoryController> {
             new NConsole::TEvConfigsDispatcher::TEvSetConfigSubscriptionRequest({
                     NKikimrConsole::TConfigItem::MemoryControllerConfigItem}));
 
+        // When profiling memory it's convenient to set initial tcmalloc soft limit
+#ifdef PROFILE_MEMORY_ALLOCATIONS
+        tcmalloc::MallocExtension::MemoryLimit limit;
+        limit.hard = false;
+        limit.limit = GetSoftLimitBytes(Config, hardLimitBytes);
+        tcmalloc::MallocExtension::SetMemoryLimit(limit);
+#endif
+
         HandleWakeup(ctx);
 
         LOG_INFO_S(ctx, NKikimrServices::MEMORY_CONTROLLER, "Bootstrapped with config " << Config.ShortDebugString());

ydb/core/memory_controller/ya.make

@@ -7,13 +7,14 @@ SRCS(
 
 PEERDIR(
     ydb/core/base
-    ydb/library/actors/core
-    ydb/library/services
-    yql/essentials/minikql
     ydb/core/cms/console
     ydb/core/mon_alloc
     ydb/core/node_whiteboard
     ydb/core/tablet
+    ydb/library/actors/core
+    ydb/library/services
+    yql/essentials/minikql
+    yql/essentials/utils/memory_profiling
 )
 
 YQL_LAST_ABI_VERSION()
@@ -22,4 +23,4 @@ END()
 
 RECURSE_FOR_TESTS(
     ut
-)
+)