YT-27835: Client tracing

* Changelog entry
  Type: fix
  Component: cpp-sdk

Create trace\_id on client side
commit_hash:270019c138bed296ed934649a0b157f514fca0cb

1 files changed, 10 insertions, 0 deletions

diff --git a/yt/cpp/mapreduce/http_client/raw_client.cpp b/yt/cpp/mapreduce/http_client/raw_client.cpp
index 557e4afdbd4..bf63a7ae3bb 100644
--- a/yt/cpp/mapreduce/http_client/raw_client.cpp
+++ b/yt/cpp/mapreduce/http_client/raw_client.cpp
@@ -308,6 +308,11 @@ TTransactionId THttpRawClient::StartTransaction(
 
 void THttpRawClient::PingTransaction(const TTransactionId& transactionId)
 {
+    auto traceContext = Context_.Config->EnableClientTracing
+        ? NTracing::CreateTraceContextFromCurrent("ping_tx")
+        : nullptr;
+    NTracing::TCurrentTraceContextGuard traceContextGuard(traceContext);
+
     std::call_once(PingClientInitOnceFlag_, [this] () {
         InitPingClient();
     });
@@ -332,6 +337,11 @@ void THttpRawClient::PingTransaction(const TTransactionId& transactionId)
     headers->Add("Content-Encoding", "identity");
     headers->Add("Accept-Encoding", "identity");
 
+    if (traceContext) {
+        auto traceparent = FormatTraceParentHeader(traceContext->GetTraceId(), traceContext->GetSpanId());
+        headers->Add("traceparent", traceparent);
+    }
+
     TNode node;
     node["transaction_id"] = GetGuidAsString(transactionId);
     auto strParams = NodeToYsonString(node);