Allow to change (reduce) max queue size for common elastic queue

Add same method as in TFastElasticQueue, and test.
commit_hash:0a2b618325e57c32fd269254a7dbe912849c3f10

1 files changed, 2 insertions, 1 deletions

diff --git a/library/cpp/threading/equeue/equeue.cpp b/library/cpp/threading/equeue/equeue.cpp
index 54b088635d5..bc0b6fc8144 100644
--- a/library/cpp/threading/equeue/equeue.cpp
+++ b/library/cpp/threading/equeue/equeue.cpp
@@ -10,7 +10,7 @@ size_t TElasticQueue::ObjectCount() const {
 }
 
 bool TElasticQueue::TryIncCounter() {
-    if (++GuardCount_ > MaxQueueSize_) {
+    if (++GuardCount_ > CurrentMaxQueueSize_) {
         --GuardCount_;
         return false;
     }
@@ -68,6 +68,7 @@ bool TElasticQueue::Add(IObjectInQueue* obj) {
 
 void TElasticQueue::Start(size_t threadCount, size_t maxQueueSize) {
     MaxQueueSize_ = maxQueueSize;
+    CurrentMaxQueueSize_ = maxQueueSize;
     SlaveQueue_->Start(threadCount, maxQueueSize);
 }