【TDengine源码阅读】举例说明pthread_once_t和PTHREAD_ONCE_INIT

2025年5月25日，周日上午

以下是一个简单的代码示例，展示如何在Linux中使用pthread_once_t和PTHREAD_ONCE_INIT来确保某个初始化函数只执行一次：

#include <stdio.h>
#include <pthread.h>

// 定义once_control变量并初始化为PTHREAD_ONCE_INIT
pthread_once_t once_control = PTHREAD_ONCE_INIT;

// 初始化函数，只会被执行一次
void init_function(void) {
    printf("初始化函数被执行，线程ID: %lu\n", pthread_self());
}

// 线程函数
void* thread_func(void* arg) {
    printf("线程 %lu 进入\n", pthread_self());
    // 调用pthread_once确保init_function只执行一次
    pthread_once(&once_control, init_function);
    printf("线程 %lu 退出\n", pthread_self());
    return NULL;
}

int main() {
    pthread_t thread1, thread2;

    // 创建两个线程
    pthread_create(&thread1, NULL, thread_func, NULL);
    pthread_create(&thread2, NULL, thread_func, NULL);

    // 等待线程结束
    pthread_join(thread1, NULL);
    pthread_join(thread2, NULL);

    return 0;
}

代码说明：

pthread_once_t once_control = PTHREAD_ONCE_INIT;  

定义并初始化一个pthread_once_t类型的变量once_control，必须使用PTHREAD_ONCE_INIT宏初始化。

pthread_once(&once_control, init_function);  

调用pthread_once函数，确保init_function只会被执行一次，即使多个线程同时调用。
运行结果

两个线程都会调用pthread_once，但init_function只会被执行一次（由其中一个线程执行），其他线程会等待初始化完成。

输出示例：

线程 128707865867968 进入
线程 128707857475264 进入
初始化函数被执行，线程ID: 128707857475264
线程 128707857475264 退出
线程 128707865867968 退出

适用场景：

多线程环境下需要一次性初始化（如全局变量、单例模式、资源加载等）。

避免重复初始化导致的资源浪费或竞争条件。