scheduler.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#pragma once

#include <functional>
#include <queue>
#include <vector>
#include <thread>
#include <mutex>

using namespace std;


class Scheduler {
	struct Job {
		function<void()> callback;

		Job(const function<void()> callback)
				: callback(callback) {}
	};

	queue<Job*> jobs;
	bool finishFlag = false;
	mutex commMut;

	bool hasJoined = false;

	vector<thread> workers;

	void workerEntry();

public:
	const int nthreads;

	Scheduler(int nthreads);
	~Scheduler();

	// func is run in child thread
	void submit(const function<void()> &func);

	void finish();
};