atomic：std::atomic_flag_wait和std::atomic_flag_wait_explicit

引入

随着多线程编程的不断普及，确保线程之间的高效同步变得尤为重要。C++标准库中的<atomic>头文件提供了多种原子操作，其中std::atomic_flag_wait和std::atomic_flag_wait_explicit是两项重要的新特性。这两个函数允许线程等待直到特定的原子标志处于预期状态，进而简化线程间的信号传递机制。使用这些原子等待函数可以有效减少CPU资源的浪费并实现更高效的多线程程序设计。本文将深入探讨这两个函数的特性、功能语法、完整示例代码及其应用场景。

特性/函数/功能语法介绍

std::atomic_flag_wait

std::atomic_flag_wait的主要特性包括：

• 原子等待：允许线程阻塞在原子标志上，直至标志被修改。
• 高效性：减少了忙等待，降低了CPU使用率。

语法

#include <atomic>

void std::atomic_flag_wait(std::atomic_flag* flag, bool expected);

std::atomic_flag_wait_explicit

std::atomic_flag_wait_explicit则提供了更多的灵活性，允许开发者指定内存序，主要特性包括：

• 内存序控制：允许在等待时自定义内存序，以满足不同的同步需求。
• 适应复杂场景：适合性能优化和更复杂的多线程场合。

语法

#include <atomic>

void std::atomic_flag_wait_explicit(std::atomic_flag* flag, bool expected, std::memory_order order);

参数 order 可以是：

• memory_order_relaxed
• memory_order_acquire
• memory_order_release
• memory_order_acq_rel
• memory_order_seq_cst

完整示例代码

以下示例展示了如何使用std::atomic_flag_wait和std::atomic_flag_wait_explicit来进行线程同步：

#include <iostream>
#include <atomic>
#include <thread>
#include <chrono>

std::atomic_flag ready_flag = ATOMIC_FLAG_INIT; // 初始化原子标志

void producer() {
    std::this_thread::sleep_for(std::chrono::seconds(1)); // 模拟生产时间
    std::cout << "Producer is ready." << std::endl;
    ready_flag.test_and_set(std::memory_order_release); // 设置标志并通知消费者
}

void consumer() {
    std::cout << "Consumer is waiting for the producer..." << std::endl;
    // 等待生产者通知
    std::atomic_flag_wait(&ready_flag, false);

    // 一旦获取到标志，继续执行
    std::cout << "Consumer has been notified." << std::endl;
}

int main() {
    std::thread prodThread(producer);   // 生产者线程
    std::thread consThread(consumer);    // 消费者线程

    prodThread.join(); // 等待生产者完成
    consThread.join(); // 等待消费者完成

    return 0;
}

代码解析

1. 初始化原子标志：

   • std::atomic_flag ready_flag = ATOMIC_FLAG_INIT; 用于表示生产者的状态，初始值设置为未准备状态（清除）。

2. 生产者函数：

   • 在 producer 函数中，模拟了生产过程（通过std::this_thread::sleep_for）。当生产完成后，通过 ready_flag.test_and_set(std::memory_order_release); 来设置标志，表示生产者已经准备好。

3. 消费者函数：

   • 在 consumer 函数中，调用 std::atomic_flag_wait(&ready_flag, false); 来等待标志被设置。该函数将使消费者线程在标志状态为false时阻塞，直到生产者线程设置了标志。

4. 主函数：

   • 在 main 函数中，分别创建并启动生产者和消费者线程，并等待它们完成。

适用场景分析

std::atomic_flag_wait和std::atomic_flag_wait_explicit的应用场景包括：

1. 生产者-消费者模式：常用于实现生产者-消费者模型中，消费者线程等待生产者完成数据生产后被通知继续处理。

2. 多线程任务协作：在复杂的多线程任务中，多个线程可能需要等待某些条件满足后才能继续执行。

3. 状态同步：在状态机或事件驱动的编程范式中，原子等待可以方便地实现状态的动态切换与调整。

总结

std::atomic_flag_wait 和 std::atomic_flag_wait_explicit 提供了高效的原子等待机制，能够协调线程间的同步与合作。通过精简的接口和摆脱忙等待的缺陷，这些函数能够大幅度提升多线程程序的性能与可读性。本文通过示例阐明了如何在实际应用中使用这些函数，帮助开发者更好地管理并发程序的复杂性。掌握这些原子操作将为构建有效的现代并发应用打下坚实的基础，让多线程编程变得更为高效持续。