1 files changed, 112 insertions, 0 deletions
diff --git a/library/cpp/yt/error/origin_attributes.cpp b/library/cpp/yt/error/origin_attributes.cpp
new file mode 100644
index 00000000000..5ff0b039339
--- /dev/null
+++ b/library/cpp/yt/error/origin_attributes.cpp
@@ -0,0 +1,112 @@
+#include "origin_attributes.h"
+
+#include <library/cpp/yt/assert/assert.h>
+
+#include <library/cpp/yt/misc/thread_name.h>
+#include <library/cpp/yt/misc/tls.h>
+
+#include <library/cpp/yt/string/format.h>
+
+#include <util/system/thread.h>
+
+namespace NYT {
+
+////////////////////////////////////////////////////////////////////////////////
+
+YT_DEFINE_THREAD_LOCAL(bool, ErrorSanitizerEnabled, false);
+YT_DEFINE_THREAD_LOCAL(TInstant, ErrorSanitizerDatetimeOverride);
+YT_DEFINE_THREAD_LOCAL(TSharedRef, ErrorSanitizerLocalHostNameOverride);
+
+TErrorSanitizerGuard::TErrorSanitizerGuard(TInstant datetimeOverride, TSharedRef localHostNameOverride)
+    : SavedEnabled_(ErrorSanitizerEnabled())
+    , SavedDatetimeOverride_(ErrorSanitizerDatetimeOverride())
+    , SavedLocalHostNameOverride_(ErrorSanitizerLocalHostNameOverride())
+{
+    ErrorSanitizerEnabled() = true;
+    ErrorSanitizerDatetimeOverride() = datetimeOverride;
+    ErrorSanitizerLocalHostNameOverride() = std::move(localHostNameOverride);
+}
+
+TErrorSanitizerGuard::~TErrorSanitizerGuard()
+{
+    YT_ASSERT(ErrorSanitizerEnabled());
+
+    ErrorSanitizerEnabled() = SavedEnabled_;
+    ErrorSanitizerDatetimeOverride() = SavedDatetimeOverride_;
+    ErrorSanitizerLocalHostNameOverride() = std::move(SavedLocalHostNameOverride_);
+}
+
+bool IsErrorSanitizerEnabled() noexcept
+{
+    return ErrorSanitizerEnabled();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+bool TOriginAttributes::operator==(const TOriginAttributes& other) const noexcept
+{
+    return
+        Host == other.Host &&
+        Datetime == other.Datetime &&
+        Pid == other.Pid &&
+        Tid == other.Tid &&
+        ExtensionData == other.ExtensionData;
+}
+
+void TOriginAttributes::Capture()
+{
+    if (ErrorSanitizerEnabled()) {
+        Datetime = ErrorSanitizerDatetimeOverride();
+        HostHolder = ErrorSanitizerLocalHostNameOverride();
+        Host = HostHolder.empty() ? TStringBuf() : TStringBuf(HostHolder.Begin(), HostHolder.End());
+        return;
+    }
+
+    Datetime = TInstant::Now();
+    Pid = GetPID();
+    Tid = TThread::CurrentThreadId();
+    ThreadName = GetCurrentThreadName();
+    ExtensionData = NDetail::GetExtensionData();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+namespace NDetail {
+
+std::optional<TOriginAttributes::TErasedExtensionData> GetExtensionData()
+{
+    using TFunctor = TOriginAttributes::TErasedExtensionData(*)();
+
+    if (auto strong = NGlobal::GetErasedVariable(GetExtensionDataTag)) {
+        return strong->AsConcrete<TFunctor>()();
+    }
+    return std::nullopt;
+}
+
+TString FormatOrigin(const TOriginAttributes& attributes)
+{
+    using TFunctor = TString(*)(const TOriginAttributes&);
+
+    if (auto strong = NGlobal::GetErasedVariable(FormatOriginTag)) {
+        return strong->AsConcrete<TFunctor>()(attributes);
+    }
+
+    return Format(
+        "%v (pid %v, thread %v)",
+        attributes.Host,
+        attributes.Pid,
+        MakeFormatterWrapper([&] (auto* builder) {
+            auto threadName = attributes.ThreadName.ToStringBuf();
+            if (threadName.empty()) {
+                FormatValue(builder, attributes.Tid, "v");
+                return;
+            }
+            FormatValue(builder, threadName, "v");
+        }));
+}
+
+} // namespace NDetail
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT