Extract memory tag API to library

1 files changed, 17 insertions, 1 deletions

diff --git a/library/cpp/ytalloc/impl/bridge.cpp b/library/cpp/ytalloc/impl/bridge.cpp
index c8287d80de5..5444a8d48b7 100644
--- a/library/cpp/ytalloc/impl/bridge.cpp
+++ b/library/cpp/ytalloc/impl/bridge.cpp
@@ -4,6 +4,8 @@
 
 #include <library/cpp/malloc/api/malloc.h>
 
+#include <library/cpp/yt/memory/memory_tag.h>
+
 namespace NYT::NYTAlloc {
 
 ////////////////////////////////////////////////////////////////////////////////
@@ -49,11 +51,19 @@ size_t GetAllocationSize(size_t size)
 #endif
 
 ////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT::NYTAlloc
+
+namespace NYT {
+
+using namespace NYTAlloc;
+
+////////////////////////////////////////////////////////////////////////////////
 // Memory tags API bridge
 
 TMemoryTag GetCurrentMemoryTag()
 {
-    return TThreadManager::GetCurrentMemoryTag();
+    return NYTAlloc::TThreadManager::GetCurrentMemoryTag();
 }
 
 void SetCurrentMemoryTag(TMemoryTag tag)
@@ -75,6 +85,12 @@ size_t GetMemoryUsageForTag(TMemoryTag tag)
 }
 
 ////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT
+
+namespace NYT::NYTAlloc {
+
+////////////////////////////////////////////////////////////////////////////////
 // Memory zone API bridge
 
 void SetCurrentMemoryZone(EMemoryZone zone)