library/cpp/threading/future/subscription/wait_ut_common.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

#include "wait_ut_common.h"

#include <util/random/shuffle.h>
#include <util/system/event.h>
#include <util/thread/pool.h>

namespace NThreading::NTest::NPrivate {

void ExecuteAndWait(TVector<std::function<void()>> jobs, TFuture<void> waiter, size_t threads) {
    Y_ENSURE(threads > 0);
    Shuffle(jobs.begin(), jobs.end());
    auto pool = CreateThreadPool(threads);
    TManualEvent start;
    for (auto& j : jobs) {
        pool->SafeAddFunc(
                        [&start, job = std::move(j)]() {
                            start.WaitI();
                            job();
                        });
    }
    start.Signal();
    waiter.Wait();
    pool->Stop();
}

}