C++ 原子类型（atomic）

1、原子类型介绍

原子类型（Atomic Types）是用于多线程编程中的一种特殊数据类型，它们提供了对共享数据的安全访问和修改，避免了数据竞争（data race）问题。原子操作是不可分割的操作，这意味着它们要么完全执行，要么完全不执行，不会在执行过程中被其他线程中断。

2、原子类型特点

1. 原子性: 原子类型的操作是不可分割的，确保在并发环境下的安全性。
2. 无锁: 原子操作通常不需要使用互斥锁（mutex），从而减少了上下文切换的开销，提高了性能。
3. 可移植性: C++标准库中的原子类型是跨平台的，确保在不同平台上的一致性。

3、C++ 中的原子类型

C++11引入了std::atomic，提供了一组原子类型。以下是一些常用的原子类型：

• std::atomic<int>: 原子整数
• std::atomic<bool>: 原子布尔值
• std::atomic<float>: 原子浮点数
• std::atomic<double>: 原子双精度浮点数
• std::atomic<T*>: 原子指针

4、原子类型的使用

1、基本使用

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

std::atomic<int> counter(0); // 原子整数

void increment() {
    for (int i = 0; i < 1000; ++i) {
        counter++; // 原子自增操作
    }
}

int main() {
    std::thread t1(increment);
    std::thread t2(increment);

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

    std::cout << "Counter: " << counter.load() << std::endl; // 输出最终计数
    return 0;
}

2、原子操作

• load(): 读取原子变量的值。
• store(): 设置原子变量的值。
• exchange(): 设置原子变量的值并返回旧值。
• compare_exchange_weak(): 比较并交换，通常用于实现锁-free数据结构。
• fetch_add(): 原子加法，返回旧值。
• fetch_sub(): 原子减法，返回旧值。

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

std::atomic<int> value(0);

void threadFunc() {
    int oldValue = value.fetch_add(1); // 原子自增，并返回旧值
    std::cout << "Old Value: " << oldValue << ", New Value: " << value.load() << std::endl;
}

int main() {
    std::thread threads[5];

    for (int i = 0; i < 5; ++i) {
        threads[i] = std::thread(threadFunc);
    }

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

    return 0;
}

5、注意事项

1. 初始化: 在使用原子类型之前，确保它们已被正确初始化。
2. 不适合复杂数据结构: 原子类型适用于简单的数据类型，对于复杂的数据结构，建议使用互斥锁。
3. 内存序: 原子操作可以指定内存序（memory order），控制操作的可见性。常见的内存序有：
   • memory_order_relaxed
   • memory_order_acquire
   • memory_order_release
   • memory_order_acq_rel
   • memory_order_seq_cst

内存序的使用示例：

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

std::atomic<int> data(0);
std::atomic<bool> ready(false);

void producer() {
    data.store(42, std::memory_order_relaxed); // 放置数据
    ready.store(true, std::memory_order_release); // 设置标志
}

void consumer() {
    while (!ready.load(std::memory_order_acquire)); // 等待数据准备好
    std::cout << "Data: " << data.load(std::memory_order_relaxed) << std::endl; // 读取数据
}

int main() {
    std::thread t1(producer);
    std::thread t2(consumer);

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

    return 0;
}

6、总结

原子类型是多线程编程中重要的工具，能够有效地避免数据竞争问题，提高程序的安全性和性能。通过合理使用原子类型和理解内存序，可以编写出高效的并发代码。