houxiao
2017-08-17 0250edfd9c4d453f25a2f9827698ccf87ff5afff
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#ifndef _PREALLOCBUFFERQUEUE_H_
#define _PREALLOCBUFFERQUEUE_H_
 
#include <cstdint>
#include <vector>
#include <queue>
 
class PreAllocBufferQueue
{
public:
    struct Config
    {
        bool multithreadSafe;
        bool fullQueueDropFront;
        bool fullQueueSync;
        size_t count;
        size_t maxBuffSize;
 
        Config()
            : multithreadSafe(false), fullQueueDropFront(false), fullQueueSync(false), count(0), maxBuffSize(0)
        { }
    };
 
    struct Buffer
    {
        uint8_t* buffer;
        size_t buffSize;
 
        Buffer() : buffer(nullptr), buffSize(0) { }
    };
 
    PreAllocBufferQueue(const Config& _cfg);
 
    ~PreAllocBufferQueue();
 
    Buffer* Dequeue();
    void Release(Buffer* buffer);
 
    Buffer* Enqueue();
    
    bool Empty() const;
    bool Full() const;
 
    const Config& GetConfig() const { return cfg; }
 
private:
    const Config cfg;
    void* mtsMux;
 
    typedef std::vector<Buffer*> buffers_vec_t;
    buffers_vec_t allBuffers;
    buffers_vec_t freeBuffers;
 
    typedef std::queue<Buffer*> buffers_que_t;
    buffers_que_t usedBuffers;
};
 
#endif