| 34 | } |
| 35 | |
| 36 | typename ThreadPool::Task ThreadPool::getNextTask() { |
| 37 | std::unique_lock<std::mutex> l(mu_); |
| 38 | for (;;) { |
| 39 | if (!tasks_.empty()) { |
| 40 | auto task = tasks_.front(); |
| 41 | tasks_.pop_front(); |
| 42 | return task; |
| 43 | } |
| 44 | |
| 45 | if (waiting_to_finish_) |
| 46 | return Task(); |
| 47 | else |
| 48 | cv_.wait(l); |
| 49 | } |
| 50 | return Task(); |
| 51 | } |
| 52 | |
| 53 | void ThreadPool::add(const Task& task) { |
| 54 | std::lock_guard<std::mutex> l(mu_); |