add traces to http backend (#5874)

Author: adameat

ydb/core/viewer/json_pipe_req.h

@@ -2,6 +2,8 @@
 
 #include <ydb/library/actors/core/actor.h>
 #include <ydb/library/actors/core/actor_bootstrapped.h>
+#include <ydb/library/wilson_ids/wilson.h>
+#include <ydb/library/actors/wilson/wilson_span.h>
 #include <ydb/core/base/tablet_pipe.h>
 #include <ydb/core/cms/console/console.h>
 #include <ydb/core/base/hive.h>
@@ -31,6 +33,7 @@ class TViewerPipeClient : public TActorBootstrapped<TDerived> {
     bool WithRetry = true;
     ui32 Requests = 0;
     static constexpr ui32 MaxRequestsInFlight = 50;
+    NWilson::TSpan Span;
 
     struct TPipeInfo {
         TActorId PipeClient;
@@ -53,6 +56,20 @@ class TViewerPipeClient : public TActorBootstrapped<TDerived> {
         return clientConfig;
     }
 
+    TViewerPipeClient() = default;
+
+    TViewerPipeClient(NMon::TEvHttpInfo::TPtr& ev) {
+        InitConfig(ev->Get()->Request.GetParams());
+        TStringBuf traceparent = ev->Get()->Request.GetHeader("traceparent");
+        if (traceparent) {
+            NWilson::TTraceId traceId = NWilson::TTraceId::FromTraceparentHeader(traceparent, TComponentTracingLevels::ProductionVerbose);
+            if (traceId) {
+                Span = {TComponentTracingLevels::THttp::TopLevel, std::move(traceId), "http", NWilson::EFlags::AUTO_END};
+                Span.Attribute("request_type", TString(ev->Get()->Request.GetPathInfo()));
+            }
+        }
+    }
+
     TActorId ConnectTabletPipe(NNodeWhiteboard::TTabletId tabletId) {
         TPipeInfo& pipeInfo = PipeInfo[tabletId];
         if (!pipeInfo.PipeClient) {

ydb/core/viewer/json_query.h

@@ -106,7 +106,8 @@ class TJsonQuery : public TViewerPipeClient<TJsonQuery> {
     }
 
     TJsonQuery(IViewer* viewer, NMon::TEvHttpInfo::TPtr& ev)
-        : Viewer(viewer)
+        : TBase(ev)
+        , Viewer(viewer)
         , Event(ev)
     {
     }
@@ -193,6 +194,9 @@ class TJsonQuery : public TViewerPipeClient<TJsonQuery> {
         auto event = std::make_unique<NKqp::TEvKqp::TEvCreateSessionRequest>();
         if (Database) {
             event->Record.MutableRequest()->SetDatabase(Database);
+            if (Span) {
+                Span.Attribute("database", Database);
+            }
         }
         BLOG_TRACE("Creating session");
         Send(NKqp::MakeKqpProxyID(SelfId().NodeId()), event.release());
@@ -440,6 +444,9 @@ class TJsonQuery : public TViewerPipeClient<TJsonQuery> {
 
     void ReplyAndPassAway(TString data) {
         Send(Event->Sender, new NMon::TEvHttpInfoRes(data, 0, NMon::IEvHttpInfoRes::EContentType::Custom));
+        if (Span) {
+            Span.End();
+        }
         PassAway();
     }
 
@@ -454,14 +461,19 @@ class TJsonQuery : public TViewerPipeClient<TJsonQuery> {
         while (protoIssues->size() > 0 && (*protoIssues)[0].issuesSize() > 0) {
             protoIssues = (*protoIssues)[0].mutable_issues();
         }
+        TString message;
         if (protoIssues->size() > 0) {
             const Ydb::Issue::IssueMessage& issue = (*protoIssues)[0];
             NProtobufJson::Proto2Json(issue, jsonResponse["error"]);
+            message = issue.message();
         }
         for (const auto& queryIssue : *protoIssues) {
             NJson::TJsonValue& issue = jsonIssues.AppendValue({});
             NProtobufJson::Proto2Json(queryIssue, issue);
         }
+        if (Span) {
+            Span.EndError(message);
+        }
     }
 
     void MakeOkReply(NJson::TJsonValue& jsonResponse, NKikimrKqp::TEvQueryResponse& record) {
@@ -591,6 +603,9 @@ class TJsonQuery : public TViewerPipeClient<TJsonQuery> {
         if (response.HasQueryStats()) {
             NProtobufJson::Proto2Json(response.GetQueryStats(), jsonResponse["stats"]);
         }
+        if (Span) {
+            Span.EndOk();
+        }
     }
 };
 
@@ -619,6 +634,11 @@ YAML::Node TJsonRequestSwagger<TJsonQuery>::GetSwagger() {
                * `explain-scan` - explain scan query (ScanQuery)
                * `explain-script` - explain script query (ScriptingService)
                * `cancel-query` - cancel query (using query_id)
+          - name: database
+            in: query
+            description: database name
+            type: string
+            required: false
           - name: query
             in: query
             description: SQL query text
@@ -637,11 +657,6 @@ YAML::Node TJsonRequestSwagger<TJsonQuery>::GetSwagger() {
             type: string
             enum: [yql_v1, pg]
             required: false
-          - name: database
-            in: query
-            description: database name
-            type: string
-            required: false
           - name: schema
             in: query
             description: >

ydb/library/wilson_ids/wilson.h

@@ -20,6 +20,7 @@ namespace NKikimr {
         };
 
 
+        DEFINE_TRACING_LEVELS(THttp, 0, 5, 10, 14, 15)
         DEFINE_TRACING_LEVELS(TGrpcProxy, 0, 5, 10, 14, 15)
         DEFINE_TRACING_LEVELS(TQueryProcessor, 1, 6, 10, 14, 15)
         DEFINE_TRACING_LEVELS(TDistributedTransactions, 2, 7, 11, 14, 15)
@@ -77,7 +78,7 @@ namespace NKikimr {
 
                 LookupActor = TComponentTracingLevels::TQueryProcessor::Basic,
                     LookupActorShardsResolve = TComponentTracingLevels::TQueryProcessor::Detailed,
-                
+
             BulkUpsertActor = TComponentTracingLevels::TQueryProcessor::TopLevel,
         };
     };