YT-17341: Use thread name instead of thread id in error origins.

6 files changed, 65 insertions, 0 deletions

diff --git a/library/cpp/yt/misc/CMakeLists.darwin-x86_64.txt b/library/cpp/yt/misc/CMakeLists.darwin-x86_64.txt
index 92907b63dc..bb8ae6f368 100644
--- a/library/cpp/yt/misc/CMakeLists.darwin-x86_64.txt
+++ b/library/cpp/yt/misc/CMakeLists.darwin-x86_64.txt
@@ -19,5 +19,6 @@ target_link_libraries(cpp-yt-misc PUBLIC
 target_sources(cpp-yt-misc PRIVATE
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/guid.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/source_location.cpp
+  ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/thread_name.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/variant.cpp
 )
diff --git a/library/cpp/yt/misc/CMakeLists.linux-aarch64.txt b/library/cpp/yt/misc/CMakeLists.linux-aarch64.txt
index 9ed005067b..7715f3febe 100644
--- a/library/cpp/yt/misc/CMakeLists.linux-aarch64.txt
+++ b/library/cpp/yt/misc/CMakeLists.linux-aarch64.txt
@@ -20,5 +20,6 @@ target_link_libraries(cpp-yt-misc PUBLIC
 target_sources(cpp-yt-misc PRIVATE
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/guid.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/source_location.cpp
+  ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/thread_name.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/variant.cpp
 )
diff --git a/library/cpp/yt/misc/CMakeLists.linux-x86_64.txt b/library/cpp/yt/misc/CMakeLists.linux-x86_64.txt
index 9ed005067b..7715f3febe 100644
--- a/library/cpp/yt/misc/CMakeLists.linux-x86_64.txt
+++ b/library/cpp/yt/misc/CMakeLists.linux-x86_64.txt
@@ -20,5 +20,6 @@ target_link_libraries(cpp-yt-misc PUBLIC
 target_sources(cpp-yt-misc PRIVATE
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/guid.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/source_location.cpp
+  ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/thread_name.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/variant.cpp
 )
diff --git a/library/cpp/yt/misc/CMakeLists.windows-x86_64.txt b/library/cpp/yt/misc/CMakeLists.windows-x86_64.txt
index d05719a445..0d302cdb6d 100644
--- a/library/cpp/yt/misc/CMakeLists.windows-x86_64.txt
+++ b/library/cpp/yt/misc/CMakeLists.windows-x86_64.txt
@@ -16,5 +16,6 @@ target_link_libraries(cpp-yt-misc PUBLIC
 target_sources(cpp-yt-misc PRIVATE
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/guid.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/source_location.cpp
+  ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/thread_name.cpp
   ${CMAKE_SOURCE_DIR}/library/cpp/yt/misc/variant.cpp
 )
diff --git a/library/cpp/yt/misc/thread_name.cpp b/library/cpp/yt/misc/thread_name.cpp
new file mode 100644
index 0000000000..372e14462e
--- /dev/null
+++ b/library/cpp/yt/misc/thread_name.cpp
@@ -0,0 +1,37 @@
+#include "thread_name.h"
+
+#include <util/generic/string.h>
+#include <util/system/thread.h>
+
+#include <algorithm>
+#include <cstring>
+
+namespace NYT {
+
+////////////////////////////////////////////////////////////////////////////////
+
+TString TThreadName::ToString() const
+{
+    // Buffer is zero terminated.
+    return Buffer.data();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+// This function uses cached TThread::CurrentThreadName() result
+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
diff --git a/library/cpp/yt/misc/thread_name.h b/library/cpp/yt/misc/thread_name.h
new file mode 100644
index 0000000000..5f87cb1fd8
--- /dev/null
+++ b/library/cpp/yt/misc/thread_name.h
@@ -0,0 +1,24 @@
+#pragma once
+
+#include <util/generic/string.h>
+
+#include <array>
+
+namespace NYT {
+
+////////////////////////////////////////////////////////////////////////////////
+
+struct TThreadName
+{
+    static constexpr int BufferCapacity = 16; // including zero terminator
+    std::array<char, BufferCapacity> Buffer{}; // zero-terminated
+    int Length; // not including zero terminator
+
+    TString ToString() const;
+};
+
+TThreadName GetCurrentThreadName();
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT