1 files changed, 48 insertions, 0 deletions
diff --git a/library/cpp/yt/system/thread_name.cpp b/library/cpp/yt/system/thread_name.cpp
new file mode 100644
index 00000000000..11e20abcf5c
--- /dev/null
+++ b/library/cpp/yt/system/thread_name.cpp
@@ -0,0 +1,48 @@
+#include "thread_name.h"
+
+#include <library/cpp/yt/misc/tls.h>
+
+#include <util/generic/string.h>
+#include <util/system/thread.h>
+
+#include <algorithm>
+#include <cstring>
+
+namespace NYT {
+
+////////////////////////////////////////////////////////////////////////////////
+
+TStringBuf TThreadName::ToStringBuf() const
+{
+    // Buffer is zero terminated.
+    return TStringBuf(Buffer.data(), Length);
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+TThreadName::TThreadName(TStringBuf name)
+{
+    Length = std::min<int>(TThreadName::BufferCapacity - 1, name.length());
+    ::memcpy(Buffer.data(), name.data(), Length);
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+// This function uses cached TThread::CurrentThreadName() result
+YT_PREVENT_TLS_CACHING TThreadName GetCurrentThreadName()
+{
+    static thread_local TThreadName ThreadName;
+
+    if (ThreadName.Length == 0) {
+        if (auto name = TThread::CurrentThreadName()) {
+            auto length = std::min<int>(TThreadName::BufferCapacity - 1, name.length());
+            ThreadName.Length = length;
+            ::memcpy(ThreadName.Buffer.data(), name.data(), length);
+        }
+    }
+    return ThreadName;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT