Use TraceId from TEvProposeTransaction (#29003)

Author: Alek5andr-Kotov

ydb/core/persqueue/pqtablet/common/tracing_support.cpp

@@ -5,13 +5,15 @@
 
 namespace NKikimr::NPQ {
 
-NWilson::TSpan GenerateSpan(const TStringBuf name, NJaegerTracing::TSamplingThrottlingControl& tracingControl)
+NWilson::TSpan GenerateSpan(const TStringBuf name, NJaegerTracing::TSamplingThrottlingControl& tracingControl, NWilson::TTraceId&& traceId)
 {
     NJaegerTracing::TRequestDiscriminator discriminator;
     discriminator.RequestType = NJaegerTracing::ERequestType::TOPIC_PROPOSE_TRANSACTION;
     discriminator.Database = {};
 
-    NWilson::TTraceId traceId = tracingControl.HandleTracing(discriminator, {});
+    if (!traceId) {
+        traceId = tracingControl.HandleTracing(discriminator, {});
+    }
 
     if (traceId) {
         return {TWilsonTopic::TopicTopLevel, std::move(traceId), TString(name), NWilson::EFlags::AUTO_END};

ydb/core/persqueue/pqtablet/common/tracing_support.h

@@ -6,7 +6,7 @@
 
 namespace NKikimr::NPQ {
 
-NWilson::TSpan GenerateSpan(const TStringBuf name, NJaegerTracing::TSamplingThrottlingControl& tracingControl);
+NWilson::TSpan GenerateSpan(const TStringBuf name, NJaegerTracing::TSamplingThrottlingControl& tracingControl, NWilson::TTraceId&& traceId);
 NWilson::TSpan GenerateSpan(const TStringBuf name, ui8 verbosity);
 
 }

ydb/core/persqueue/pqtablet/pq_impl.cpp

@@ -3122,7 +3122,7 @@ void TPersQueue::Handle(TEvPersQueue::TEvProposeTransaction::TPtr& ev, const TAc
     const NKikimrPQ::TEvProposeTransaction& event = ev->Get()->GetRecord();
     PQ_LOG_TX_D("Handle TEvPersQueue::TEvProposeTransaction " << event.ShortDebugString());
 
-    auto span = GenerateSpan("Topic.Transaction", *SamplingControl);
+    auto span = GenerateSpan("Topic.Transaction", *SamplingControl, std::move(ev->TraceId));
     span.Attribute("TxId", static_cast<i64>(event.GetTxId()));
     span.Attribute("TabletId", static_cast<i64>(TabletID()));