C++同步队列

  • 本文实现了一个带锁同步队列,可以让多个线程向队列里读写数据而不会产生异常,很方便的应用于生产者消费者模式的程序中

  • 稍微复杂的地方在于怎么样让生产者消费者停止运行,或者说,怎么才能让同步队列退出;话不多说,直接上代码:

#ifndef SYNC_QUEUE_H
#define SYNC_QUEUE_H

/*
 * 同步队列,实现了生产者消费者
 */
#include <deque>
#include <pthread.h>


template<typename T>
class SyncQueue{
public:
    SyncQueue(){
        running_ = true;
        pthread_mutex_init(&mutex_,NULL);
        pthread_cond_init(&cond_,NULL);
    }

    ~SyncQueue(){
        stop();
        pthread_cond_destroy(&cond_);
        pthread_mutex_destroy(&mutex_);
    }

    // 入队
    void push_back(T info){
        SafeLock guard(mutex_);
        if (!running_) {
            return;
        }

        bool empty = data_list_.empty();
        data_list_.push_back(info);
        if (empty) {
            Signal();
        }
    }

    // 返回第一个元素,不等待
    bool head(T& val) {
        SafeLock guard(mutex_);
        if ( data_list_.empty()) {
            return false;
        }

        val = data_list_[0];
        return true;
    }

    // 返回最后一个元素,不等待
    bool tail(T& val) {
        SafeLock guard(mutex_);
        if (data_list_.empty()) {
            return false;
        }

        val = data_list_[data_list_.size() -1];

        return true;
    }

    // 获取某个元素,不等待
    bool get(size_t index, T& val) {
        SafeLock guard(mutex_);
        if (data_list_.empty()) {
            return false;
        }

        if (index >= data_list_.size()) {
            return false;
        }

        val = data_list_[index];

        return true;
    }

    // 出队,等待
    bool pop_front(T& val){
        SafeLock guard(mutex_);
        if (!running_ && data_list_.empty()) {
            return false;
        }

        bool ret = false;

        // 已经不运行了
        if (!running_) {
            if (data_list_.empty() == false) {
                val = data_list_[0];
                data_list_.pop_front();
                ret = true;
            }
            else {
                ret = false;
            }

            return ret;
        }

        // 还在运行
        Wait();

        // 到了这一步将有两种情况:
        // 1、队列中有数据
        // 2、队列退出!
        if (running_) {
            if (!data_list_.empty()) {
                val = data_list_[0];
                data_list_.pop_front();
                ret = true;
            }
            else {
                // 一般来说这一步是不存在的
                ret = false;
            }

        }
        else {
            ret = false;
        }

        return ret;
    }

    size_t size(){
        SafeLock guard(mutex_);
        size_t size = data_list_.size();
        return size;
    }

    bool empty(){
        SafeLock guard(mutex_);
        bool empty = data_list_.empty();
        return empty;
    }

    bool is_running() {
        SafeLock guard(mutex_);
        return running_;
    }

    /*void start() {
        SafeLock guard(mutex_);
        if(running_){
            return;
        }

        //data_list_.clear();
        running_ = true;
    }*/

    // 由数据发送方调用
    void stop() {
        SafeLock guard(mutex_);
        running_ = false;
        Broadcast();
    }

private:
    void Wait(){
        while(data_list_.empty() && running_){
            pthread_cond_wait(&cond_,&mutex_);
        }
    }

    void Signal(){
        pthread_cond_signal(&cond_);
    }

    void Broadcast() {
        pthread_cond_broadcast(&cond_);
    }

    class SafeLock {
    public:
        SafeLock(pthread_mutex_t& l):mutex(l) {
            pthread_mutex_lock(&mutex);
        }
        ~SafeLock() {
            pthread_mutex_unlock(&mutex);
        }
    private:
        pthread_mutex_t& mutex;
    };
private:
    SyncQueue & operator=(const SyncQueue&) {}

    // 队列锁和队列
    pthread_mutex_t mutex_;
    pthread_cond_t cond_;
    std::deque<T> data_list_;

    bool running_;
};


#endif // SYNC_QUEUE_H

  • 测试代码
#include <iostream>
#include <thread>
#include <chrono>
#include "sync_queue.h"

using namespace std;

SyncQueue<int> Q;

// 生产者
void produce(){
    for(int i = 0; i < 1024; ++i){
        Q.push_back(i);
    }
    // 通知各个消费者,同步队列已经停止
    Q.stop();
}

// 消费者
void consume(){
    int val;
    while (Q.pop_front(val)) {
        printf("val = %d\n",val);
        std::this_thread::sleep_for(chrono::milliseconds(100));
    }
}

int main()
{
	// 创建生产者者线程
    thread pth(produce);
    pth.detach();

	// 创建消费者线程
    thread cth1(consume);
    thread cth2(consume);
    thread cth3(consume);
    thread cth4(consume);
    thread cth5(consume);
    thread cth6(consume);
    thread cth7(consume);
    thread cth8(consume);

    cth1.join();
    cth2.join();
    cth3.join();
    cth4.join();
    cth5.join();
    cth6.join();
    cth7.join();
    cth8.join();

    return 0;
}

完整的代码实现:
https://github.com/mobinsheng/sync_queue

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值