4rkal/monero
1
0
Fork 0
mirror of https://github.com/monero-project/monero.git synced 2024-12-14 04:16:32 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

threadpool: allow constructing an object, and misc tweaks

Browse Source 
use unsigned int to avoid having to range check negative numbers,
use const where possible, don't needlessly create empty objects,
use std::move where possible
This commit is contained in:
moneromooo-monero 2018-04-23 18:53:38 +01:00
parent 6b9d9f56a1
commit db55263b4c
No known key found for this signature in database
GPG Key ID: 686F07454D6CEFC3
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/common/threadpool.cpp
View File

@ -39,11 +39,11 @@ static __thread int depth = 0;
namespace tools namespace tools
{ {
threadpool::threadpool() : running(true), active(0) { threadpool::threadpool(unsigned int max_threads) : running(true), active(0) {
boost::thread::attributes attrs; boost::thread::attributes attrs;
attrs.set_stack_size(THREAD_STACK_SIZE); attrs.set_stack_size(THREAD_STACK_SIZE);
max = tools::get_max_concurrency(); max = max_threads ? max_threads : tools::get_max_concurrency();
size_t i = max; unsigned int i = max;
while(i--) { while(i--) {
threads.push_back(boost::thread(attrs, boost::bind(&threadpool::run, this))); 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; return max;
} }

14
src/common/threadpool.h
View File

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