intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 36 insertions, 0 deletions

diff --git a/library/cpp/messagebus/lfqueue_batch.h b/library/cpp/messagebus/lfqueue_batch.h
new file mode 100644
index 0000000000..8128d3154d
--- /dev/null
+++ b/library/cpp/messagebus/lfqueue_batch.h
@@ -0,0 +1,36 @@
+#pragma once
+
+#include <library/cpp/messagebus/actor/temp_tls_vector.h>
+
+#include <util/generic/vector.h>
+#include <util/thread/lfstack.h>
+
+template <typename T, template <typename, class> class TVectorType = TVector>
+class TLockFreeQueueBatch {
+private:
+    TLockFreeStack<TVectorType<T, std::allocator<T>>*> Stack;
+
+public:
+    bool IsEmpty() {
+        return Stack.IsEmpty();
+    }
+
+    void EnqueueAll(TAutoPtr<TVectorType<T, std::allocator<T>>> vec) {
+        Stack.Enqueue(vec.Release());
+    }
+
+    void DequeueAllSingleConsumer(TVectorType<T, std::allocator<T>>* r) {
+        TTempTlsVector<TVectorType<T, std::allocator<T>>*> vs;
+        Stack.DequeueAllSingleConsumer(vs.GetVector());
+
+        for (typename TVector<TVectorType<T, std::allocator<T>>*>::reverse_iterator i = vs.GetVector()->rbegin();
+             i != vs.GetVector()->rend(); ++i) {
+            if (i == vs.GetVector()->rend()) {
+                r->swap(**i);
+            } else {
+                r->insert(r->end(), (*i)->begin(), (*i)->end());
+            }
+            delete *i;
+        }
+    }
+};