ThreadPool.h

浏览该文件的文档.

#pragma once
 
#include "thread/Task.h"
 
#include <condition_variable>
#include <cstddef>
#include <functional>
#include <memory>
#include <mutex>
#include <queue>
#include <string>
#include <thread>
#include <vector>
 
namespace eve {
namespace thread {
 
class Channel;
 
class ThreadPool {
public:
    explicit ThreadPool(int workerCount);
    ~ThreadPool();
 
    ThreadPool(const ThreadPool &) = delete;
    ThreadPool &operator=(const ThreadPool &) = delete;
 
    int getWorkerCount() const;
    int getPendingCount() const;
    bool isRunning() const;
 
    Task *submit(std::function<void()> fn);
 
    Task *submitSleep(int ms);
 
    Task *submitPush(Channel *channel, std::string message, int delayMs = 0);
 
    Task *submitPost(std::string name, std::string data = "", int delayMs = 0);
 
    void waitAll();
 
    void stop();
 
private:
    struct State;
 
    static void workerMain(std::shared_ptr<State> state);
    void stopImpl(bool allowWorkerCaller);
 
    int workerCount_ = 0;
    std::vector<std::thread> workers_;
    std::shared_ptr<State> state_;
    std::mutex lifecycleMu_;
};
 
}  // namespace thread
}  // namespace eve