Moved yql/utils YQL-19206

Также была выделена жирная зависимость из yql/utils в yql/utils/network, в результате library/cpp/getopt была добавлена явно в те проекты, которые ее ранее наследовали, а не указывали явно
commit_hash:36aa4c41f807b4cbbf70a3ed7ac0a1a5079bb75d

1 files changed, 35 insertions, 0 deletions

diff --git a/yql/essentials/utils/limiting_allocator.cpp b/yql/essentials/utils/limiting_allocator.cpp
new file mode 100644
index 00000000000..0ff84f90378
--- /dev/null
+++ b/yql/essentials/utils/limiting_allocator.cpp
@@ -0,0 +1,35 @@
+#include "limiting_allocator.h"
+
+#include <util/memory/pool.h>
+#include <util/generic/yexception.h>
+
+namespace {
+class TLimitingAllocator : public IAllocator {
+public:
+    TLimitingAllocator(size_t limit, IAllocator* allocator) : Alloc_(allocator), Limit_(limit) {};
+    TBlock Allocate(size_t len) override final {
+        if (Allocated_ + len > Limit_) {
+            throw std::runtime_error("Out of memory");
+        }
+        Allocated_ += len;
+        return Alloc_->Allocate(len);
+    }
+
+    void Release(const TBlock& block) override final {
+        Y_ENSURE(Allocated_ >= block.Len);
+        Allocated_ -= block.Len;
+        Alloc_->Release(block);
+    }
+
+private:
+    IAllocator* Alloc_;
+    size_t Allocated_ = 0;
+    const size_t Limit_;
+};
+}
+
+namespace NYql {
+std::unique_ptr<IAllocator> MakeLimitingAllocator(size_t limit, IAllocator* underlying) {
+    return std::make_unique<TLimitingAllocator>(limit, underlying);
+}
+}