| 159 | } |
| 160 | |
| 161 | void AThreadPool::setWorkersCount(std::size_t workersCount) { |
| 162 | AUI_ASSERTX(workersCount >= 2 && workersCount <= 1000, "invalid worker count"); |
| 163 | std::unique_lock lck(mQueueLock); |
| 164 | if (mWorkers.size() >= workersCount) { |
| 165 | while (mWorkers.size() > workersCount) { |
| 166 | mCV.notify_all(); |
| 167 | mWorkers.last()->aboutToDelete(); |
| 168 | lck.unlock(); |
| 169 | mWorkers.pop_back(); |
| 170 | lck.lock(); |
| 171 | } |
| 172 | } else { |
| 173 | // have to add new workers |
| 174 | mWorkers.reserve(workersCount); |
| 175 | while (mWorkers.size() < workersCount) { |
| 176 | mWorkers.push_back(std::make_unique<Worker>(*this, mWorkers.size())); |
| 177 | } |
| 178 | } |
| 179 | } |
| 180 | |
| 181 | size_t AThreadPool::getPendingTaskCount() { return mQueueHighest.size() + mQueueLowest.size() + mQueueMedium.size(); } |
nothing calls this directly
no test coverage detected