Ydb stable 23-1-1923.1.19

x-stable-origin-commit: c5d5a396e89d0a72e0267a55e93d8404d4fb54fe

1 files changed, 30 insertions, 0 deletions

diff --git a/library/cpp/unified_agent_client/throttling.h b/library/cpp/unified_agent_client/throttling.h
new file mode 100644
index 00000000000..1e5db1e8fa1
--- /dev/null
+++ b/library/cpp/unified_agent_client/throttling.h
@@ -0,0 +1,30 @@
+#pragma once
+
+#include <library/cpp/unified_agent_client/f_maybe.h>
+
+#include <util/datetime/base.h>
+
+namespace NUnifiedAgent {
+    class TThrottler {
+    public:
+        explicit TThrottler(double rate, TDuration updatePeriod = TDuration::MilliSeconds(100));
+
+        TThrottler(double rate, double burst);
+
+        void Consume(double& tokens, TFMaybe<TDuration>& nextCheckDelay);
+
+        bool TryConsume(double tokens);
+
+        void ConsumeAndWait(double tokens);
+
+    private:
+        ui64 UpdateTokens();
+
+    private:
+        ui64 CyclesPerMillisecond;
+        ui64 UpdatePeriod;
+        double PeriodTokens;
+        double AvailableTokens;
+        ui64 ExpirationTime;
+    };
+}