如何在c++中创建屏障？

#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>

class Barrier {
public:
    explicit Barrier(std::size_t count) : count_(count), generation_(0) {}

    void Wait() {
        std::unique_lock<std::mutex> lock(mutex_);
        auto gen = generation_;

        if (--count_ == 0) {
            generation_++;
            count_ = static_cast<std::size_t>(gen);
            cv_.notify_all();
        } else {
            cv_.wait(lock, [this, gen] { return gen != generation_; });
        }
    }

private:
    std::mutex mutex_;
    std::condition_variable cv_;
    std::size_t count_;
    std::size_t generation_;
};

Barrier barrier(3); // 假设有3个线程需要等待

void ThreadFunction() {
    // 线程执行的代码

    barrier.Wait(); // 等待其他线程到达屏障

    // 继续执行其他代码
}

int main() {
    std::thread t1(ThreadFunction);
    std::thread t2(ThreadFunction);
    std::thread t3(ThreadFunction);

    t1.join();
    t2.join();
    t3.join();

    return 0;
}