intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 42 insertions, 0 deletions

diff --git a/library/cpp/messagebus/remote_server_session_semaphore.h b/library/cpp/messagebus/remote_server_session_semaphore.h
new file mode 100644
index 0000000000..de714fd342
--- /dev/null
+++ b/library/cpp/messagebus/remote_server_session_semaphore.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include "cc_semaphore.h"
+
+#include <util/generic/noncopyable.h>
+
+namespace NBus {
+    namespace NPrivate {
+        class TRemoteServerSessionSemaphore: public TComplexConditionSemaphore<TRemoteServerSessionSemaphore> {
+        private:
+            const char* const Name;
+
+            TAtomicBase const LimitCount;
+            TAtomicBase const LimitSize;
+            TAtomic CurrentCount;
+            TAtomic CurrentSize;
+            TAtomic PausedByUser;
+            TAtomic StopSignal;
+
+        public:
+            TRemoteServerSessionSemaphore(TAtomicBase limitCount, TAtomicBase limitSize, const char* name = "unnamed");
+            ~TRemoteServerSessionSemaphore();
+
+            TAtomicBase GetCurrentCount() const {
+                return AtomicGet(CurrentCount);
+            }
+            TAtomicBase GetCurrentSize() const {
+                return AtomicGet(CurrentSize);
+            }
+
+            void IncrementMultiple(TAtomicBase count, TAtomicBase size);
+            bool TryWait();
+            void ReleaseMultiple(TAtomicBase count, TAtomicBase size);
+            void Stop();
+            void PauseByUsed(bool pause);
+
+        private:
+            void CheckNeedToUnlock();
+        };
+
+    }
+}