Merge pull request #3691

db55263 threadpool: allow constructing an object, and misc tweaks (moneromooo-monero)
ce173cb core: remove threadpool dependency from header (moneromooo-monero)
3147468 unit_tests: add threadpool unit test (moneromooo-monero)
This commit is contained in:
luigi1111 2018-06-17 22:04:39 -05:00
commit cfd66dae4b
No known key found for this signature in database
GPG key ID: F4ACA0183641E010
6 changed files with 118 additions and 15 deletions

View file

@ -39,11 +39,11 @@ static __thread int depth = 0;
namespace tools
{
threadpool::threadpool() : running(true), active(0) {
threadpool::threadpool(unsigned int max_threads) : running(true), active(0) {
boost::thread::attributes attrs;
attrs.set_stack_size(THREAD_STACK_SIZE);
max = tools::get_max_concurrency();
size_t i = max;
max = max_threads ? max_threads : tools::get_max_concurrency();
unsigned int i = max;
while(i--) {
threads.push_back(boost::thread(attrs, boost::bind(&threadpool::run, this)));
}
@ -78,7 +78,7 @@ void threadpool::submit(waiter *obj, std::function<void()> f) {
}
}
int threadpool::get_max_concurrency() {
unsigned int threadpool::get_max_concurrency() const {
return max;
}

View file

@ -46,6 +46,9 @@ public:
static threadpool instance;
return instance;
}
static threadpool *getNewForUnitTests(unsigned max_threads = 0) {
return new threadpool(max_threads);
}
// The waiter lets the caller know when all of its
// tasks are completed.
@ -66,11 +69,12 @@ public:
// task to finish.
void submit(waiter *waiter, std::function<void()> f);
int get_max_concurrency();
unsigned int get_max_concurrency() const;
~threadpool();
private:
threadpool();
~threadpool();
threadpool(unsigned int max_threads = 0);
typedef struct entry {
waiter *wo;
std::function<void()> f;
@ -79,8 +83,8 @@ public:
boost::condition_variable has_work;
boost::mutex mutex;
std::vector<boost::thread> threads;
int active;
int max;
unsigned int active;
unsigned int max;
bool running;
void run();
};