如何将blocking-waitUntil()方法写入到使用std::atomics的池中

Question

我有一个验证类，它使用线程池来处理它的所有作业。

现在，当用户询问时，我启动一个线程，通过从磁盘读取来为我的验证类提供作业。我确信，在某一点上，阅读会比处理更快。所以我想写一个方法，如果处理的作业超过1000个，这个线程可以等待。

我已经介绍了一个原子，它在添加作业时增加，在完成作业时减少。

我尝试添加一个方法的效果并不理想。我知道一定有可能使用更好的东西。

void Validator::waitUntilAvailable() {
while (m_blocksInFlight > 1000) { // thats my atomic
usleep(50000); // seems to be unavailable on Windows.
}
}

这里有人能帮助使用非轮询方法来解决我的问题吗？

谢谢。

Answer 1

您希望等待一个条件，但没有等待机制。

该机制是std::condition_variable和std::mutex。例如：

class Validator
{
std::mutex m_mutex;
std::condition_variable m_condition;
std::atomic<int> m_blocksInFlight{0};
bool test() const {
return m_blocksInFlight.load(std::memory_order_relaxed) > 1000;
}
void addJob() {
++m_blocksInFlight;
// Only lock the mutex when the test succeeds.
if(this->test()) {
std::unique_lock<decltype(m_mutex)> lock(m_mutex);
m_condition.notify_one();
}
}
void waitUntilAvailable() {
std::unique_lock<decltype(m_mutex)> lock(m_mutex);
while(!this->test())
m_condition.wait(lock);
}
};