#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);
};
}