You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
768 B
40 lines
768 B
11 years ago
|
#pragma once
|
||
|
|
||
|
/* Simple thread pool, no fancy stuff.
|
||
|
It is thread safe (i.e. concurrent adds).
|
||
|
Threads terminate if there is no work (we need wait_and_pop for that).
|
||
|
Works best if all work is added, and then run.
|
||
|
Doesn't care about exceptions.
|
||
|
*/
|
||
|
|
||
|
#include <vector>
|
||
|
#include <functional>
|
||
|
#include <thread>
|
||
|
#include "lock_queue.hpp"
|
||
|
|
||
|
struct thread_pool {
|
||
|
void add(std::function<void()> f){
|
||
|
work.push(f);
|
||
|
}
|
||
|
|
||
|
void run(int number_of_threads){
|
||
|
for(int i = 0; i < number_of_threads; ++i){
|
||
|
threads.emplace_back([this]{
|
||
|
auto x = work.try_pop();
|
||
|
while(x){
|
||
|
(*x)();
|
||
|
x = work.try_pop();
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
|
||
|
~thread_pool(){
|
||
|
for(auto & t : threads) t.join();
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
queue<std::function<void()>> work;
|
||
|
std::vector<std::thread> threads;
|
||
|
};
|