mirror of
https://github.com/netchx/netch.git
synced 2026-03-14 17:43:18 +08:00
20 lines
319 B
C++
20 lines
319 B
C++
#include "WaitGroup.h"
|
|
|
|
void WaitGroup::Add(int size)
|
|
{
|
|
this->counter += size;
|
|
}
|
|
|
|
void WaitGroup::Done()
|
|
{
|
|
if (--this->counter <= 0)
|
|
this->condition.notify_all();
|
|
}
|
|
|
|
void WaitGroup::Wait()
|
|
{
|
|
std::unique_lock<std::mutex> lock(this->mutex);
|
|
|
|
condition.wait(lock, [&] { return counter <= 0; });
|
|
}
|