dl_list_mt.h

浏览该文件的文档.

#pragma once
 
#include <list>
 
#include "base/dl_type_def.h"
 
namespace dl
{
template<typename T>
class ListMultiThread
{
public:
    bool PopFront(T& a)
    {
        _mutex.lock();
        if (_list.empty())
        {
            _mutex.unlock();
            return false;
        }
        a = _list.front();
        _list.pop_front();
        _mutex.unlock();
        return true;
    }

    void Add(const T& buf)
    {
        _mutex.lock();
        _list.push_back(buf);
        _mutex.unlock();
    }

    void Add(T&& buf)
    {
        _mutex.lock();
        _list.push_back(buf);
        _mutex.unlock();
    }

    size_t GetSize()
    {
        std::lock_guard guard{ _mutex };
        return _list.size();
    }

    bool IsEmpty()
    {
        std::lock_guard guard{ _mutex };
        return _list.empty();
    }

    void Clear()
    {
        _mutex.lock();
        _list.clear();
        _mutex.unlock();
    }
 
    void SetQueueMode(QueueMode mode)
    {
        _mode = mode;
    }
 
    ListMultiThread(QueueMode mode = QueueMode::FIFO) :
        _mode{ mode }
    {}
private:
    std::mutex _mutex;// 队列锁
    std::list<T> _list;// 队列
    QueueMode _mode;
};
}