龙少华 發表於 2025-12-18 11:12:43

C语言原子操作的实现示例

<div id="navCategory"><h5 class="catalogue">目录</h5><ul class="first_class_ul"><li>什么是原子操作?</li><ul class="second_class_ul"><li>比喻理解</li></ul><li>非原子操作的问题</li><ul class="second_class_ul"></ul><li>C语言中的原子操作</li><ul class="second_class_ul"><li>C11标准引入的原子类型</li></ul><li>常见的原子操作函数</li><ul class="second_class_ul"><li>基本操作</li><li>算术运算</li><li>位运算</li></ul><li>原子操作的实际例子</li><ul class="second_class_ul"><li>1. 无锁计数器</li><li>2. 自旋锁实现</li><li>3. 无锁栈(Lock-Free Stack)</li></ul><li>内存顺序(Memory Order)</li><ul class="second_class_ul"></ul><li>不同平台的原生原子操作</li><ul class="second_class_ul"><li>x86架构</li><li>GCC内置原子操作</li></ul><li>原子操作的优缺点</li><ul class="second_class_ul"><li>优点:</li><li>缺点:</li></ul><li>原子操作 vs 互斥锁</li><ul class="second_class_ul"></ul><li>总结</li><ul class="second_class_ul"></ul></ul></div><p><strong>原子操作</strong>是并发编程中的一个核心概念。让我详细解释:</p>
<p class="maodian"></p><h2>什么是原子操作?</h2>
<p><strong>原子操作</strong>是指在执行过程中<strong>不会被中断</strong>的操作,要么完全执行,要么完全不执行,不会出现部分执行的状态。</p>
<p class="maodian"></p><h3>比喻理解</h3>
<p>就像原子是不可分割的基本单位一样,原子操作是不可分割的操作单位。</p>
<p class="maodian"></p><h2>非原子操作的问题</h2>
<p>先看一个非原子操作的例子:</p>
<div class="jb51code"><pre class="brush:cpp;">int counter = 0;

// 这不是原子操作!
void increment() {
    counter = counter + 1;
}</pre></div>
<p>在汇编层面,这通常分为三步:</p>
<div class="jb51code"><pre class="brush:cpp;">mov eax, ; 1. 读取counter到寄存器
inc eax             ; 2. 寄存器值加1
mov , eax; 3. 写回内存</pre></div>
<p><strong>竞态条件问题</strong>:</p>
<blockquote><p>线程A:读取counter=0<br />线程B:读取counter=0 &nbsp;<br />线程A:计算0+1=1<br />线程B:计算0+1=1<br />线程A:写入counter=1<br />线程B:写入counter=1 &nbsp;&larr; 结果应该是2,但实际是1!</p></blockquote>
<p class="maodian"></p><h2>C语言中的原子操作</h2>
<p class="maodian"></p><h3>C11标准引入的原子类型</h3>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;

// 声明原子变量
atomic_int atomic_counter = ATOMIC_VAR_INIT(0);

// 原子操作
void increment_atomic() {
    atomic_fetch_add(&amp;atomic_counter, 1);// 原子加法
}

int main() {
    atomic_counter = 5;                  // 原子存储
    int value = atomic_counter;            // 原子加载
    printf("Counter: %d\n", value);
   
    return 0;
}</pre></div>
<p class="maodian"></p><h2>常见的原子操作函数</h2>
<p class="maodian"></p><h3>基本操作</h3>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;

atomic_int counter = ATOMIC_VAR_INIT(0);

// 加载和存储
int load_value = atomic_load(&amp;counter);      // 原子读取
atomic_store(&amp;counter, 42);                  // 原子写入

// 交换操作
int old_value = atomic_exchange(&amp;counter, 100); // 原子交换

// 比较交换(CAS - Compare And Swap)
int expected = 100;
if (atomic_compare_exchange_strong(&amp;counter, &amp;expected, 200)) {
    printf("CAS成功: 旧值=%d, 新值=200\n", expected);
}</pre></div>
<p class="maodian"></p><h3>算术运算</h3>
<div class="jb51code"><pre class="brush:cpp;">// 原子加法
int old = atomic_fetch_add(&amp;counter, 5);    // counter += 5,返回旧值

// 原子减法
atomic_fetch_sub(&amp;counter, 3);            // counter -= 3

// 原子自增/自减
atomic_fetch_add(&amp;counter, 1);            // counter++
atomic_fetch_sub(&amp;counter, 1);            // counter--</pre></div>
<p class="maodian"></p><h3>位运算</h3>
<div class="jb51code"><pre class="brush:cpp;">atomic_int flags = ATOMIC_VAR_INIT(0);

// 原子位操作
atomic_fetch_or(&amp;flags, 0x01);   // flags |= 0x01
atomic_fetch_and(&amp;flags, ~0x01);   // flags &amp;= ~0x01
atomic_fetch_xor(&amp;flags, 0x03);    // flags ^= 0x03</pre></div>
<p class="maodian"></p><h2>原子操作的实际例子</h2>
<p class="maodian"></p><h3>1. 无锁计数器</h3>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;
#include &lt;pthread.h&gt;
#include &lt;stdio.h&gt;

atomic_int counter = ATOMIC_VAR_INIT(0);

void* worker(void* arg) {
    for (int i = 0; i &lt; 100000; i++) {
      atomic_fetch_add(&amp;counter, 1);// 原子自增
    }
    return NULL;
}

int main() {
    pthread_t t1, t2;
   
    pthread_create(&amp;t1, NULL, worker, NULL);
    pthread_create(&amp;t2, NULL, worker, NULL);
   
    pthread_join(t1, NULL);
    pthread_join(t2, NULL);
   
    printf("最终计数: %d (应该是200000)\n", atomic_load(&amp;counter));
    return 0;
}</pre></div>
<p class="maodian"></p><h3>2. 自旋锁实现</h3>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;

typedef atomic_flag spinlock_t;

void spinlock_init(spinlock_t* lock) {
    atomic_flag_clear(lock);
}

void spinlock_lock(spinlock_t* lock) {
    // 忙等待,直到获得锁
    while (atomic_flag_test_and_set(lock)) {
      // 可选的:减少CPU占用
      // __builtin_ia32_pause();// x86的PAUSE指令
    }
}

void spinlock_unlock(spinlock_t* lock) {
    atomic_flag_clear(lock);
}</pre></div>
<p class="maodian"></p><h3>3. 无锁栈(Lock-Free Stack)</h3>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;
#include &lt;stdlib.h&gt;

typedef struct Node {
    int data;
    struct Node* next;
} Node;

typedef struct {
    _Atomic(Node*) top;
} LockFreeStack;

void stack_init(LockFreeStack* stack) {
    atomic_store(&amp;stack-&gt;top, NULL);
}

void stack_push(LockFreeStack* stack, int value) {
    Node* new_node = malloc(sizeof(Node));
    new_node-&gt;data = value;
   
    Node* old_top;
    do {
      old_top = atomic_load(&amp;stack-&gt;top);
      new_node-&gt;next = old_top;
    } while (!atomic_compare_exchange_weak(&amp;stack-&gt;top, &amp;old_top, new_node));
}

int stack_pop(LockFreeStack* stack) {
    Node* old_top;
    Node* new_top;
   
    do {
      old_top = atomic_load(&amp;stack-&gt;top);
      if (old_top == NULL) return -1;// 栈空
      new_top = old_top-&gt;next;
    } while (!atomic_compare_exchange_weak(&amp;stack-&gt;top, &amp;old_top, new_top));
   
    int value = old_top-&gt;data;
    free(old_top);
    return value;
}</pre></div>
<p class="maodian"></p><h2>内存顺序(Memory Order)</h2>
<p>原子操作还涉及内存可见性问题:</p>
<div class="jb51code"><pre class="brush:cpp;">#include &lt;stdatomic.h&gt;

atomic_int data = ATOMIC_VAR_INIT(0);
atomic_int flag = ATOMIC_VAR_INIT(0);

// 生产者线程
void producer() {
    atomic_store_explicit(&amp;data, 42, memory_order_relaxed);
    atomic_store_explicit(&amp;flag, 1, memory_order_release);// 释放语义
}

// 消费者线程
void consumer() {
    while (atomic_load_explicit(&amp;flag, memory_order_acquire) == 0) {
      // 等待
    }
    int value = atomic_load_explicit(&amp;data, memory_order_relaxed);
    printf("Data: %d\n", value);// 保证看到42
}</pre></div>
<p class="maodian"></p><h2>不同平台的原生原子操作</h2>
<p class="maodian"></p><h3>x86架构</h3>
<div class="jb51code"><pre class="brush:cpp;">// 内联汇编实现原子操作
int atomic_increment(int* value) {
    __asm__ __volatile__(
      "lock incl %0"          // lock前缀确保原子性
      : "+m" (*value)
      :
      : "cc"
    );
    return *value;
}</pre></div>
<p class="maodian"></p><h3>GCC内置原子操作</h3>
<div class="jb51code"><pre class="brush:cpp;">int counter = 0;

// GCC内置的原子操作
void increment_gcc() {
    __sync_fetch_and_add(&amp;counter, 1);
}

int compare_and_swap_gcc(int* ptr, int oldval, int newval) {
    return __sync_val_compare_and_swap(ptr, oldval, newval);
}</pre></div>
<p class="maodian"></p><h2>原子操作的优缺点</h2>
<p class="maodian"></p><h3>优点:</h3>
<ul><li><strong>高性能</strong>:避免锁的开销</li><li><strong>无死锁</strong>:不会出现锁顺序问题</li><li><strong>可扩展性</strong>:在多核系统中表现良好</li></ul>
<p class="maodian"></p><h3>缺点:</h3>
<ul><li><strong>复杂性</strong>:正确实现很困难</li><li><strong>ABA问题</strong>:在CAS操作中可能出现</li><li><strong>平台依赖性</strong>:不同硬件支持程度不同</li></ul>
<p class="maodian"></p><h2>原子操作 vs 互斥锁</h2>
<table><thead><tr><th>特性</th><th>原子操作</th><th>互斥锁</th></tr></thead><tbody><tr><td>性能</td><td>高(硬件支持)</td><td>中等(系统调用)</td></tr><tr><td>复杂度</td><td>高</td><td>低</td></tr><tr><td>适用场景</td><td>简单操作(计数器、标志位)</td><td>复杂临界区</td></tr><tr><td>死锁风险</td><td>无</td><td>有</td></tr><tr><td>可扩展性</td><td>好</td><td>一般</td></tr></tbody></table>
<p class="maodian"></p><h2>总结</h2>
<p>原子操作是:</p>
<ul><li><strong>不可分割</strong>的操作单位</li><li><strong>线程安全</strong>的,无需额外同步</li><li><strong>硬件支持</strong>的,通常通过CPU指令实现</li><li><strong>高性能</strong>的并发编程基础</li></ul>
<p>在现代多核处理器系统中,原子操作是实现高效并发程序的重要工具,特别是在实现无锁数据结构、计数器、标志位等场景中。</p>
<p>到此这篇关于C语言原子操作的实现示例的文章就介绍到这了,更多相关C语言 原子操作内容请搜索琼殿技术社区以前的文章或继续浏览下面的相关文章希望大家以后多多支持琼殿技术社区!</p>
                           
                            <div class="art_xg">
                              <b>您可能感兴趣的文章:</b><ul><li>C语言中互斥锁与自旋锁及原子操作使用浅析</li></ul>
                            </div>

                        </div>
                        <!--endmain-->
頁: [1]
查看完整版本: C语言原子操作的实现示例