dl_list_mt_cache.h

浏览该文件的文档.

#pragma once
 
#include <list>
 
#include "base/dl_type_def.h"
 
namespace dl
{
template<typename T>
class ListMultiThreadCache
{
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)
    {
        if (_mutex.try_lock())
        {
            _list.splice(_list.begin(), _listTemp);
            _list.push_back(buf);
            _mutex.unlock();
            return;
        }
        _listTemp.push_back(buf);
    }

    void Add(T&& buf)
    {
        if (_mutex.try_lock())
        {
            _list.splice(_list.begin(), _listTemp);
            _list.push_back(buf);
            _mutex.unlock();
            return;
        }
        _listTemp.push_back(buf);
    }

    void Move()
    {
        if (_listTemp.empty())
            return;
        _mutex.lock();
        _list.splice(_list.begin(), _listTemp);
        _mutex.unlock();
    }

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

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

    void Clear()
    {
        _mutex.lock();
        _list.clear();
        _mutex.unlock();
        _listTemp.clear();
    }
 
    void SetQueueMode(QueueMode mode)
    {
        _mode = mode;
    }
 
    ListMultiThreadCache(QueueMode mode = QueueMode::FIFO) :
        _mode{ mode }
    {}
private:
    std::mutex _mutex;// 队列锁
    std::list<T> _list;// 先入先出队列
    std::list<T> _listTemp;// 主线程缓冲
    QueueMode _mode;
};
}