2023-09-20 00:25:38 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
#include <queue>
|
|
|
|
#include <vector>
|
|
|
|
#include <thread>
|
|
|
|
#include <mutex>
|
|
|
|
#include <condition_variable>
|
2023-09-20 19:45:22 +00:00
|
|
|
#include <functional>
|
2023-09-20 00:25:38 +00:00
|
|
|
|
|
|
|
class TaskManager {
|
|
|
|
public:
|
2023-09-20 19:45:22 +00:00
|
|
|
using Job = std::function<void ()>;
|
2023-09-20 00:25:38 +00:00
|
|
|
TaskManager();
|
|
|
|
~TaskManager();
|
|
|
|
|
|
|
|
void start();
|
|
|
|
void stop();
|
2023-09-20 19:45:22 +00:00
|
|
|
void wait() const;
|
|
|
|
void queue(const Job& job);
|
|
|
|
bool busy() const;
|
2023-09-20 00:25:38 +00:00
|
|
|
|
|
|
|
private:
|
|
|
|
void loop();
|
|
|
|
|
|
|
|
private:
|
|
|
|
bool running;
|
|
|
|
bool stopping;
|
|
|
|
uint32_t maxThreads;
|
|
|
|
uint32_t activeThreads;
|
2023-09-20 19:45:22 +00:00
|
|
|
std::queue<Job> jobs;
|
|
|
|
mutable std::mutex mutex;
|
|
|
|
mutable std::condition_variable loopConditional;
|
|
|
|
mutable std::condition_variable waitConditional;
|
2023-09-20 00:25:38 +00:00
|
|
|
std::vector<std::thread> threads;
|
|
|
|
};
|