Rust 并发同步之屏障（Barrier）：让多线程步调一致

文章目录

Rust 并发同步之屏障（Barrier）：让多线程步调一致

Rust 并发同步之屏障（Barrier）：让多线程步调一致

在 Rust 多线程并发编程中，同步是保证数据安全、逻辑正确的核心环节。当我们需要多个线程完成各自的前置任务后，再同时进入下一阶段执行时，标准库中提供的屏障（Barrier）就是最简洁高效的解决方案。本文将从基础概念、代码实践、特性解析到注意事项，带你全面掌握 Barrier 的使用，轻松解决多线程同步中的“集合等待”问题。

什么是 Barrier？

Barrier（屏障）是一种轻量级同步原语，核心作用是“集合等待”。它会设定一个“等待阈值”（即需要参与同步的线程数量），每个线程执行到屏障点时会主动阻塞，直到所有设定数量的线程都到达该屏障点，所有线程才会被同时唤醒，继续执行后续代码。比如“先各自准备数据，再统一处理结果”的场景。

快速了解 API

Rust 标准库中 Barrier 的 API 非常简洁，核心只有两个：

Barrier::new(n: usize)：创建一个屏障，参数 n 是需要等待的线程数量，也是屏障的“触发阈值”。
barrier.wait(&self) -> BarrierWaitResult：线程调用该方法后会阻塞，直到第 n 个线程也调用 wait()；返回值 BarrierWaitResult 提供 is_leader() 方法，用于判断当前线程是否为“领导线程”。

实战示例

下面我们通过一个多线程数据处理的模拟场景，展示 Barrier 的完整用法。场景需求：三个线程各自完成数据加载（前置任务），等待所有线程加载完成后，由一个领导线程汇总数据，再所有线程同时开始数据处理（后续任务）。

use std::sync::{Arc, Barrier};
use std::thread;
use std::time::Duration;

fn main() {
    // 创建屏障：设定需要等待3个工作线程
    let barrier = Arc::new(Barrier::new(3));
    let mut thread_handles = Vec::new();

    // 启动3个工作线程，通 过Arc 共享 Barrier
    for thread_id in 0..3 {
        let barrier_clone = Arc::clone(&barrier);
        let handle = thread::spawn(move || {
            // 第一阶段：各自执行前置任务（模拟数据加载）
            println!("线程 {}: 正在加载数据（前置任务）...", thread_id);
            thread::sleep(Duration::from_secs(1)); // 模拟耗时操作
            println!("线程 {}: 数据加载完成，等待其他线程", thread_id);

            // 到达屏障点，阻塞等待其他线程
            let wait_result = barrier_clone.wait();

            // 所有线程到达后，唤醒并执行后续任务
            // is_leader()：随机选出一个领导线程，执行额外工作
            if wait_result.is_leader() {
                println!("\n【领导线程】: 所有线程已就绪，开始汇总数据...\n");
                // 模拟汇总操作
                thread::sleep(Duration::from_secs(1));
            }

            // 所有线程同时执行后续任务（数据处理）
            println!("线程 {}: 开始处理数据（后续任务）...", thread_id);
            thread::sleep(Duration::from_secs(1));
            println!("线程 {}: 数据处理完成！", thread_id);
        });
        thread_handles.push(handle);
    }

    // 等待所有工作线程执行完毕，避免主线程提前退出
    for handle in thread_handles {
        handle.join().unwrap();
    }

    println!("\n所有线程任务全部完成！");
}

Barrier 特性解析

可重复使用性

Barrier 并非一次性同步工具，而是可以重复使用的。当所有线程通过一次屏障后，屏障会自动重置计数器，下次调用 wait() 时，会再次等待设定数量的线程到达。

这种特性非常适合多轮迭代并行任务，比如机器学习中的批量训练，每一轮都需要所有线程完成当前批次处理，再进入下一批次。只需在循环中多次调用 wait()，即可实现多轮同步。

领导线程机制

wait() 返回的 BarrierWaitResult 的 is_leader() 方法，会随机从所有到达屏障的线程中选出一个领导线程，该线程会先于其他线程执行后续代码或执行额外任务。

线程安全与所有权管理

Barrier 本身实现了 Sync 和 Send 特征，所以可以安全地在多线程间共享。在多线程场景中，通常需要配合 Arc（原子引用计数）来共享 Barrier 的所有权，因为每个线程需要拥有 Barrier 的引用才能调用 wait()，而 Arc 可以实现多线程安全的共享。

注意事项

线程数量必须与屏障阈值严格匹配

Barrier::new(n)中的 n（阈值）必须与实际调用 wait() 的线程数量完全一致，否则会导致严重问题：

线程数量 < 阈值：所有调用 wait() 的线程会永远阻塞（死锁），因为永远达不到触发屏障的条件；
线程数量 > 阈值：第 n 个线程调用 wait() 后，前 n 个线程会被唤醒，剩余线程会阻塞在 wait() 处，直到再次有 n 个线程调用 wait()，若没有后续调用，同样会死锁。