【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 退出
适用场景:
多线程环境下需要一次性初始化(如全局变量、单例模式、资源加载等)。
避免重复初始化导致的资源浪费或竞争条件。