Merge 347cbdd into b58320e

Author: serbel324

ydb/core/load_test/group_write.cpp

@@ -6,6 +6,7 @@
 
 #include <ydb/library/yverify_stream/yverify_stream.h>
 #include <ydb/core/util/lz4_data_generator.h>
+#include <ydb/core/jaeger_tracing/throttler.h>
 
 #include <google/protobuf/text_format.h>
 
@@ -445,6 +446,8 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
         // There is no point in having more than 1 active garbage collection request at the moment
         constexpr static ui32 MaxGarbageCollectionsInFlight = 1;
 
+        std::shared_ptr<NJaegerTracing::TThrottler> TracingThrottler;
+
     public:
         TTabletWriter(TIntrusivePtr<::NMonitoring::TDynamicCounters> counters,
                 TLogWriterLoadTestActor& self, ui64 tabletId, ui32 channel,
@@ -453,7 +456,8 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
                 NKikimrBlobStorage::EGetHandleClass getHandleClass, const TRequestDispatchingSettings& readSettings,
                 TIntervalGenerator garbageCollectIntervalGen,
                 TDuration scriptedRoundDuration, TVector<TReqInfo>&& scriptedRequests,
-                const TInitialAllocation& initialAllocation)
+                const TInitialAllocation& initialAllocation,
+                const std::shared_ptr<NJaegerTracing::TThrottler>& tracingThrottler)
             : Self(self)
             , TagCounters(counters->GetSubgroup("tag", Sprintf("%" PRIu64, Self.Tag)))
             , Counters(TagCounters->GetSubgroup("channel", Sprintf("%" PRIu32, channel)))
@@ -491,6 +495,7 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
             , ScriptedRequests(std::move(scriptedRequests))
             , InitialAllocation(initialAllocation)
             , GarbageCollectIntervalGen(garbageCollectIntervalGen)
+            , TracingThrottler(tracingThrottler)
         {
             *Counters->GetCounter("tabletId") = tabletId;
             const auto& percCounters = Counters->GetSubgroup("sensor", "microseconds");
@@ -923,7 +928,13 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
                     IssueReadIfPossible(ctx);
                 }
             };
-            SendToBSProxy(ctx, GroupId, ev.release(), Self.QueryDispatcher.ObtainCookie(std::move(writeCallback)));
+
+            NWilson::TTraceId traceId = (TracingThrottler && !TracingThrottler->Throttle())
+                    ? NWilson::TTraceId::NewTraceId(15, ::Max<ui32>())
+                    : NWilson::TTraceId{};
+
+            SendToBSProxy(ctx, GroupId, ev.release(), Self.QueryDispatcher.ObtainCookie(std::move(writeCallback)),
+                    std::move(traceId));
             const auto nowCycles = GetCycleCountFast();
             WritesInFlightTimestamps.emplace_back(writeQueryId, nowCycles);
             SentTimestamp.emplace(writeQueryId, nowCycles);
@@ -1073,7 +1084,12 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
                 IssueReadIfPossible(ctx);
             };
 
-            SendToBSProxy(ctx, GroupId, ev.release(), Self.QueryDispatcher.ObtainCookie(std::move(readCallback)));
+            NWilson::TTraceId traceId = (TracingThrottler && !TracingThrottler->Throttle())
+                    ? NWilson::TTraceId::NewTraceId(15, ::Max<ui32>())
+                    : NWilson::TTraceId{};
+
+            SendToBSProxy(ctx, GroupId, ev.release(), Self.QueryDispatcher.ObtainCookie(std::move(readCallback)),
+                    std::move(traceId));
             ReadSentTimestamp.emplace(readQueryId, GetCycleCountFast());
 
             ReadSettings.InFlightTracker.Request(size);
@@ -1211,6 +1227,14 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
 
             TIntervalGenerator garbageCollectIntervalGen(profile.GetFlushIntervals());
 
+            std::shared_ptr<NJaegerTracing::TThrottler> tracingThrottler;
+
+            ui32 throttlerRate = profile.GetTracingThrottlerRate();
+            if (throttlerRate) {
+                tracingThrottler.reset(new NJaegerTracing::TThrottler(throttlerRate, profile.GetTracingThrottlerBurst(),
+                        TAppData::TimeProvider));
+            }
+
             for (const auto& tablet : profile.GetTablets()) {
                 auto scriptedRoundDuration = TDuration::MicroSeconds(tablet.GetScriptedCycleDurationSec() * 1e6);
                 TVector<TReqInfo> scriptedRequests;
@@ -1256,7 +1280,7 @@ class TLogWriterLoadTestActor : public TActorBootstrapped<TLogWriterLoadTestActo
                     getHandleClass, readSettings,
                     garbageCollectIntervalGen,
                     scriptedRoundDuration, std::move(scriptedRequests),
-                    initialAllocation));
+                    initialAllocation, tracingThrottler));
 
                 WorkersInInitialState++;
             }

ydb/core/protos/load_test.proto

@@ -86,6 +86,9 @@ message TEvLoadTestRequest {
             optional THardRateDispatcherInfo ReadHardRateDispatcher = 16;
 
             optional TInitialBlobAllocation InitialAllocation = 17;
+
+            optional uint32 TracingThrottlerRate = 18 [default = 0];
+            optional uint32 TracingThrottlerBurst = 19 [default = 0];
         };
         optional uint64 Tag = 1;
         optional uint32 DurationSeconds = 2;