YT-23817: implement try acquire in TCombinedThroughputThrottler

commit_hash:7e61a1191c39017d06254e045d3f58bbbbdadc01

1 files changed, 17 insertions, 2 deletions

diff --git a/yt/yt/core/concurrency/throughput_throttler.cpp b/yt/yt/core/concurrency/throughput_throttler.cpp
index c88a7a2ddd2..daafe952a33 100644
--- a/yt/yt/core/concurrency/throughput_throttler.cpp
+++ b/yt/yt/core/concurrency/throughput_throttler.cpp
@@ -725,11 +725,26 @@ public:
         }));
     }
 
-    bool TryAcquire(i64 /*amount*/) override
+    bool TryAcquire(i64 amount) override
     {
-        YT_ABORT();
+        size_t i = 0;
+        for (; i < Throttlers_.size(); ++i) {
+            if (!Throttlers_[i]->TryAcquire(amount)) {
+                break;
+            }
+        }
+
+        if (i != Throttlers_.size()) {
+            for (size_t j = 0; j < i; ++j) {
+                Throttlers_[j]->Release(amount);
+            }
+            return false;
+        }
+
+        return true;
     }
 
+    // TODO: implement TryAcquireAvailable the same way as TryAcquire.
     i64 TryAcquireAvailable(i64 /*amount*/) override
     {
         YT_ABORT();