atomic：std::atomic_fetch_or和std::atomic_fetch_or_explicit

引入

在多线程编程中，安全地对共享数据进行操作是确保程序稳定性和正确性的关键。C++标准库的<atomic>头文件提供了多种原子操作，包含std::atomic_fetch_or和std::atomic_fetch_or_explicit，这两个函数用于执行原子按位或（OR）操作。它们允许线程安全地对原子变量进行按位或操作，同时返回操作之前的旧值，确保数据的一致性和可预测性。通过使用这些原子操作，开发者可以巧妙地管理多线程之间的共享状态。本文将详细探讨这两个函数的特性、用法、示例代码及适用场景分析。

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

std::atomic_fetch_or

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

• 原子性：确保在执行按位或操作时，不会被其他线程干扰。
• 简单易用：提供直观的接口，实现对原子变量的按位或操作。

语法

#include <atomic>

T std::atomic_fetch_or(std::atomic<T>& obj, T arg);

std::atomic_fetch_or_explicit

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

• 内存序控制：允许开发者在执行按位或操作时指定内存序，从而满足特定的同步需求。
• 增强灵活性：适合高性能应用，能够对内存操作进行精细控制。

语法

#include <atomic>

T std::atomic_fetch_or_explicit(std::atomic<T>& obj, T arg, std::memory_order order);

参数order可以是：

• memory_order_relaxed：无顺序保证。
• memory_order_acquire：在此调用之前的所有读取必须完成。
• memory_order_release：在此调用之后的所有写入必须完成。
• memory_order_acq_rel：同时保证读取和写入的顺序。
• memory_order_seq_cst：确保全局顺序一致。

完整示例代码

以下示例展示了如何使用std::atomic_fetch_or和std::atomic_fetch_or_explicit对共享变量进行原子按位或操作：

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

std::atomic<int> sharedFlags{0}; // 初始值为0

void set_flags(int flags) {
    for (int i = 0; i < 3; ++i) {
        int oldValue = std::atomic_fetch_or(sharedFlags, flags); // 使用atomic_fetch_or
        std::cout << "Set flags: " << flags 
                  << ", old value: " << oldValue 
                  << ", new value: " << (oldValue | flags) << std::endl;
        std::this_thread::sleep_for(std::chrono::milliseconds(500));
    }
}

void set_flags_explicit(int flags) {
    for (int i = 0; i < 3; ++i) {
        int oldValue = std::atomic_fetch_or_explicit(sharedFlags, flags, std::memory_order_seq_cst); // 使用atomic_fetch_or_explicit
        std::cout << "Set flags(explicit): " << flags 
                  << ", old value: " << oldValue 
                  << ", new value: " << (oldValue | flags) << std::endl;
        std::this_thread::sleep_for(std::chrono::milliseconds(500));
    }
}

int main() {
    std::thread t1(set_flags, 0b0011);  // 应用掩码0b0011 (3)
    std::thread t2(set_flags_explicit, 0b1100); // 应用掩码0b1100 (12)

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

    std::cout << "Final flags value: " << sharedFlags.load() << std::endl; //输出最终值
    return 0;
}

代码解析

在上述示例中，我们通过std::atomic_fetch_or和std::atomic_fetch_or_explicit对共享标志变量进行原子按位或操作，确保在多线程环境中安全操作。

1. 创建原子变量：

   • std::atomic<int> sharedFlags{0}; 创建了一个初始值为0的原子整型变量，用来存储状态标志。

2. 按位或的线程：

   • 在 set_flags 函数中，使用 std::atomic_fetch_or(sharedFlags, flags) 对 sharedFlags 执行按位或操作，并返回之前的值，确保操作的原子性。

3. 按位或的线程（显式内存序）：

   • 在 set_flags_explicit 函数中，利用 std::atomic_fetch_or_explicit(sharedFlags, flags, std::memory_order_seq_cst) 指定内存序为 memory_order_seq_cst，确保操作的全局顺序一致性。

4. 主函数：

   • 在main函数中，启动两个线程分别调用不同的按位或函数，这两个线程将对sharedFlags施加不同的位掩码操作，并等待它们结束。

5. 最终值显示：

   • 最后输出sharedFlags的值，以展示对该变量的安全按位或操作的结果。

适用场景分析

std::atomic_fetch_or和std::atomic_fetch_or_explicit的应用场景包括：

1. 状态标志管理：在多线程环境中使用原子按位或操作来设置不同的状态标志，确保标志在所有线程之间共享。

2. 构建无锁数据结构：在设计高效的无锁数据结构时，原子按位或操作可以用作状态的合并部分，减少锁竞争。

3. 配置管理：在并发配置管理中，按位或操作经常用于组合多个设置或选项，以控制软件的行为。

总结

std::atomic_fetch_or和std::atomic_fetch_or_explicit为C++提供了安全、有效的原子按位或操作，确保在多线程环境中数据的一致性和可预测性。通过本文的深入分析与示例，读者能够理解如何应对复杂的并发操作，并合理利用这些函数来管理共享状态。掌握这些原子操作将帮助开发者在构建复杂多线程环境下的程序时，提升其稳定性和性能。