library/cpp/messagebus/lfqueue_batch.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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; 
        } 
    } 
};