c++怎么使用std::latch和std::barrier进行线程同步_C++20多线程同步原语使用教程

std::latch用于一次性同步,主线程等待多个工作线程完成任务,计数归零后释放所有线程;std::barrier支持重复使用,适合多轮协作场景,每轮所有线程到达后执行可选完成函数并继续下一轮。

在C++20中std::latchstd::barrier 是两个新的线程同步原语,用于简化多线程程序中的等待逻辑。它们比传统的条件变量更简洁、高效,适用于不同场景下的线程协调。

std::latch:一次性倒计时同步

std::latch 是一个一次性使用的同步机制,它允许一个或多个线程等待,直到一个计数器减到零。一旦计数归零,所有等待的线程被释放,且 latch 不可重用。

常用方法:

  • count_down(n):将内部计数器减少 n,不会阻塞。
  • wait():阻塞当前线程,直到计数器变为 0。
  • arrive_and_wait(n):等价于 count_down(n) 后调用 wait()。

适用场景:主线程启动多个工作线程,等待它们全部完成某阶段任务。

示例代码:

#include 
#include 
#include 
#include 

int main() { std::latch latch(3); // 需要3个线程到达

std::vector threads;
for (int i = 0; i < 3; ++i) {
    threads.emplace_back([&](int id) {
        std::cout << "线程 " << id << " 完成第一阶段\n";
        latch.count_down(1); // 完成后倒计时减1
    }, i);
}

latch.wait(); // 主线程等待所有线程完成
std::cout << "所有线程已就绪,继续执行...\n";

for (auto& t : threads) {
    t.join();
}

}

std::barrier:可重复使用的屏障

std::barrier 与 latch 类似,但支持重复使用。当指定数量的线程调用 arrive()arrive_and_wait() 后,所有线程被同时释放,并可以进入下一轮同步。

关键特性:

  • 可多次使用,适合循环协作场景。
  • 可选地指定“完成函数”(completion function),在每次所有线程到达后执行一次。

示例:三个线程进行两轮并行计算,每轮结束后同步。

#include 
#include 
#include 
#include 

int main() { std::barrier sync_point(3, []{ std::cout << "【所有线程完成本轮】\n"; });

auto worker = [&](int id) {
    for (int round = 0; round < 2; ++round) {
        std::cout << "线程 " << id << " 执行第 " << round + 1 << " 轮\n";
        std::this_thread::sleep_for(std::chrono::milliseconds(100 * (id + 1)));
        sync_point.arrive_and_wait(); // 等待其他线程完成本轮
    }
};

std::vector threads;
for (int i = 0; i < 3; ++i) {
    threads.emplace_back(worker, i);
}

for (auto& t : threads) {
    t.join();
}

}

latch 与 barrier 的选择建议

根据使用需求决定:

  • 如果只需要一次等待(如初始化完成、任务结束通知),用 std::latch 更轻量。
  • 如果多个阶段需要反复同步(如并行迭代算法),则选择 std::barrier
  • 两者都不支持动态调整参与线程数,设计时需提前确定数量。

基本上就这些。这两个新工具让C++20的多线程同步更直观、安全,减少了手动管理条件变量的复杂性。不复杂但容易忽略的是:确保每个线程对 latch/barrier 的调用次数正确,避免死锁或未定义行为。