C11中函数作用域静态变量的线程安全初始化方案探讨

你好！首先得夸你一下，你已经敏锐地发现了第一个版本中失败线程不会等待初始化完成的问题，不过第二个版本还有优化空间。下面我来拆解更优的实现方式，以及关于yield的权衡点：

一、更简洁且正确的无锁实现 #

你的第二个版本引入了额外的initialize_begin原子变量，其实完全可以只用一个原子变量来同时标记「初始化中」和「已完成」状态，同时通过正确的内存序保证初始化操作的内存可见性。以下是修正后的代码：

#include <stdatomic.h>
#include <stdbool.h>

typedef /* 你的结构体定义 */ some_t;

const some_t* func(void) {
    static some_t thing;
    static atomic_bool initialized = ATOMIC_VAR_INIT(false);

    // 快速路径：已初始化时仅做一次relaxed加载，开销极低
    if (!atomic_load_explicit(&initialized, memory_order_relaxed)) {
        bool expected = false;
        // 尝试抢占初始化权限
        if (atomic_compare_exchange_strong_explicit(
                &initialized, &expected, true,
                memory_order_acquire, memory_order_relaxed)) {
            // 获胜线程执行复杂初始化（调用其他函数等操作）
            // 示例：initialize_some_t(&thing);
            
            // 初始化完成后，用release语义确保所有修改对其他线程可见
            atomic_store_explicit(&initialized, true, memory_order_release);
        } else {
            // 失败线程等待初始化完成，acquire语义保证能看到完整的初始化结果
            while (!atomic_load_explicit(&initialized, memory_order_acquire)) {
                // 这里是否yield？下面单独讨论
            }
        }
    }
    return &thing;
}

关键改进细节： #

• 移除多余原子变量：只用initialized一个变量，通过CAS操作区分「未初始化」「正在初始化」「已完成」三种状态（CAS成功意味着从「未初始化」进入「正在初始化」，最终用store标记为「已完成」）。
• 内存序精准控制：
  • 快速路径用memory_order_relaxed：已初始化后，所有线程看到的都是最终状态，不需要额外同步，开销降到最低。
  • CAS成功时用memory_order_acquire：确保后续初始化操作的内存修改不会被重排到CAS之前；失败时用memory_order_relaxed：只是没抢到权限，不需要同步。
  • 初始化完成的store用memory_order_release：把所有初始化的内存操作"发布"给其他线程，保证可见性。
  • 等待循环的load用memory_order_acquire：确保等待线程能看到获胜线程完成的所有初始化修改。

二、失败线程是否需要yield？ #

这个问题没有绝对答案，得根据初始化操作的耗时来权衡：

• 如果初始化很快（比如几微秒级别）：不要yield！空循环（或者用平台专用的pause指令，比如x86的__builtin_ia32_pause()）的开销比yield小得多——yield会触发上下文切换，反而浪费更多CPU资源。pause指令还能告诉CPU当前在自旋等待，避免乱序执行优化，减少功耗。
• 如果初始化耗时较长（比如毫秒级别）：建议调用sched_yield()让出CPU，让其他线程能利用这段时间执行任务，避免空循环占着CPU不放。

三、额外推荐：用C11的call_once（如果允许） #

如果你项目可以使用C11标准库，call_once是官方提供的线程安全初始化方案，它的底层实现已经优化了无锁快速路径，还能处理初始化异常等边缘情况，可靠性拉满：

#include <threads.h>

typedef /* 你的结构体定义 */ some_t;

static once_flag init_flag = ONCE_FLAG_INIT;
static some_t thing;

static void init_thing(void) {
    // 这里执行你的复杂初始化逻辑
}

const some_t* func(void) {
    call_once(&init_flag, init_thing);
    return &thing;
}

这个方案的快速路径开销同样极低，而且不需要你手动处理原子操作和内存序，省心又靠谱。

内容的提问来源于stack exchange，提问作者Bill Webster