ConcurrentTaskQueue.h

Go to the documentation of this file.

 
#pragma once
 
#include <trantor/utils/TaskQueue.h>
#include <trantor/exports.h>
#include <list>
#include <memory>
#include <vector>
#include <queue>
#include <string>
 
namespace trantor
{
class TRANTOR_EXPORT ConcurrentTaskQueue : public TaskQueue
{
  public:
    ConcurrentTaskQueue(size_t threadNum, const std::string &name);

    virtual void runTaskInQueue(const std::function<void()> &task);
    virtual void runTaskInQueue(std::function<void()> &&task);

    virtual std::string getName() const
    {
        return queueName_;
    };

    size_t getTaskCount();

    void stop();
 
    ~ConcurrentTaskQueue();
 
  private:
    size_t queueCount_;
    std::string queueName_;
 
    std::queue<std::function<void()>> taskQueue_;
    std::vector<std::thread> threads_;
 
    std::mutex taskMutex_;
    std::condition_variable taskCond_;
    std::atomic_bool stop_;
    void queueFunc(int queueNum);
};
 
}  // namespace trantor