feat token-bucket: add statistics

Tests: добавлена метрика
commit_hash:eb84472a15f607b97087ebd49a542d70078ac9be

Author: vansokol

.mapping.json

@@ -2079,7 +2079,6 @@
   "core/src/utils/statistics/impl/histogram_view_utils.hpp":"taxi/uservices/userver/core/src/utils/statistics/impl/histogram_view_utils.hpp",
   "core/src/utils/statistics/impl/rate_serialization.hpp":"taxi/uservices/userver/core/src/utils/statistics/impl/rate_serialization.hpp",
   "core/src/utils/statistics/json.cpp":"taxi/uservices/userver/core/src/utils/statistics/json.cpp",
-  "core/src/utils/statistics/labels.cpp":"taxi/uservices/userver/core/src/utils/statistics/labels.cpp",
   "core/src/utils/statistics/metadata.cpp":"taxi/uservices/userver/core/src/utils/statistics/metadata.cpp",
   "core/src/utils/statistics/metric_value.cpp":"taxi/uservices/userver/core/src/utils/statistics/metric_value.cpp",
   "core/src/utils/statistics/metrics_storage.cpp":"taxi/uservices/userver/core/src/utils/statistics/metrics_storage.cpp",
@@ -5649,6 +5648,7 @@
   "universal/src/utils/small_string_benchmark.cpp":"taxi/uservices/userver/universal/src/utils/small_string_benchmark.cpp",
   "universal/src/utils/small_string_test.cpp":"taxi/uservices/userver/universal/src/utils/small_string_test.cpp",
   "universal/src/utils/span_test.cpp":"taxi/uservices/userver/universal/src/utils/span_test.cpp",
+  "universal/src/utils/statistics/labels.cpp":"taxi/uservices/userver/universal/src/utils/statistics/labels.cpp",
   "universal/src/utils/statistics/rate.cpp":"taxi/uservices/userver/universal/src/utils/statistics/rate.cpp",
   "universal/src/utils/statistics/rate_counter.cpp":"taxi/uservices/userver/universal/src/utils/statistics/rate_counter.cpp",
   "universal/src/utils/statistics/request.cpp":"taxi/uservices/userver/universal/src/utils/statistics/request.cpp",

universal/include/userver/utils/token_bucket.hpp

@@ -6,6 +6,8 @@
 #include <atomic>
 #include <chrono>
 
+#include <userver/utils/statistics/rate_counter.hpp>
+
 USERVER_NAMESPACE_BEGIN
 
 namespace utils {
@@ -88,15 +90,21 @@ class TokenBucket final {
     static double GetRatePs(Duration interval);
 
 private:
+    friend void DumpMetric(statistics::Writer& writer, const TokenBucket& bucket);
+
     void Update();
 
     std::atomic<size_t> max_size_;
     std::atomic<size_t> token_refill_amount_;
     std::atomic<Duration> token_refill_interval_;
     std::atomic<size_t> tokens_;
     std::atomic<TimePoint> last_update_;
+
+    statistics::RateCounter obtain_failed_;
 };
 
+void DumpMetric(statistics::Writer& writer, const TokenBucket& bucket);
+
 }  // namespace utils
 
 USERVER_NAMESPACE_END

core/src/utils/statistics/labels.cpp universal/src/utils/statistics/labels.cppcore/src/utils/statistics/labels.cpp renamed to universal/src/utils/statistics/labels.cpp

@@ -3,6 +3,7 @@
 #include <stdexcept>
 
 #include <userver/utils/mock_now.hpp>
+#include <userver/utils/statistics/writer.hpp>
 
 USERVER_NAMESPACE_BEGIN
 
@@ -112,6 +113,7 @@ bool TokenBucket::Obtain() { return ObtainAll(1); }
 bool TokenBucket::ObtainAll(size_t count) {
     if (max_size_ < count) {
         // not satisfiable
+        obtain_failed_.Add(statistics::Rate{count});
         return false;
     }
 
@@ -125,6 +127,7 @@ bool TokenBucket::ObtainAll(size_t count) {
     auto expected = tokens_.load();
     do {
         if (expected < count) {
+            obtain_failed_.Add(statistics::Rate{count});
             return false;
         }
     } while (!tokens_.compare_exchange_strong(expected, expected - count));
@@ -188,6 +191,12 @@ void TokenBucket::Update() {
     }
 }
 
+void DumpMetric(statistics::Writer& writer, const TokenBucket& bucket) {
+    const static std::string kObtainFailed = "obtain_failed";
+
+    writer[kObtainFailed] = bucket.obtain_failed_.Load();
+}
+
 }  // namespace utils
 
 USERVER_NAMESPACE_END