C++11 实现简单的阻塞队列

r囧r小猫 2022-12-04 14:59 110阅读 0赞

一个blocking queue是一个支持 take 和 put 的 queue。

take操作从队列中取出一个元素，如果队列为空，那么take会阻塞到直到队列中有可用元素。

put操作往队列里放入一个元素，如果队列是满，那么put会阻塞到队列有可用空间。

一个简单的blocking queue的实现，首先我们要有一个非常底层的容器，至于是vector还是deque甚至是list都无所谓，我们这里选择**deque作为容器**。

前面定义的blocking queue是有界的，也就是说元素有上限，所以才会出现满元素阻塞的情况。

我们先从简单的无界阻塞队列开始，默认这个队列可以一直存储元素。于是我们就不用去判断队列满元素的情况了。这个时候我们只需要一个条件变量来判断队列为空并且阻塞就行。大概框架如下。

template<typename T>
    class BlockQueue{
        std::deque<T> que;
        std::condition_variable empty;
        std::mutex mtx; // declared for unique_lock;
    public:
        T take();
        void put(T elem);
        int size();
    }

**那么核心的take和put要怎么实现呢？**

**首先修改队列的操作要加锁，并发编程的时候只要涉及到修改可变变量都必须要加锁，不巧put和take都是在修改队列**。

这里使用**unique\_lock**来加锁

std::unique_lock<std::mutex> lock{mtx};

两个操作中只有take操作会出现队列为空的情况，**所以take必须判断队列为空，这里使用条件变量**

empty.wait(lock, [this]{return !que.empty();});

于是我们的put和take操作就出来了

template<typename T>
    T BlockQueue::take(){
         std::unique_lock<std::mutex> lock{mtx};
         empty.wait(lock, [this]{return !que.empty();});
         auto res = que.front():
         que.pop_front():
         return res;
    }
    
    template<typename T>
    void BlockQueue::put(T elem){
         std::unique_lock<std::mutex> lock{mtx};
         que.push_back(T elem);
    }

作者：不动点P  
链接：https://www.jianshu.com/p/9dd1b7863667  
来源：简书  
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。