pthread_key_create的用法

转自：http://www.douban.com/note/162329664/

面说一下线程中特有的线程存储， Thread Specific Data 。线程存储有什么用了？他是什么意思了？大家都知道，在多线程程序中，所有线程共享程序中的变量。现在有一全局变量，所有线程都可以使用它，改变它的值。 而如果每个线程希望能单独拥有它，那么就需要使用线程存储了。表面上看起来这是一个全局变量，所有线程都可以使用它，而它的值在每一个线程中又是单独存储 的。这就是线程存储的意义。

下面说一下线程存储的具体用法。

l 创建一个类型为 pthread_key_t 类型的变量。

l 调用 pthread_key_create() 来创建该变量。该函数有两个参数，第一个参数就是上面声明的 pthread_key_t 变量，第二个参数是一个清理函数，用来在线程释放该线程存储的时候被调用。该函数指针可以设成 NULL ，这样系统将调用默认的清理函数。

l 当线程中需要存储特殊值的时候，可以调用 pthread_setspecific() 。该函数有两个参数，第一个为前面声明的 pthread_key_t 变量，第二个为 void* 变量，这样你可以存储任何类型的值。

l 如果需要取出所存储的值，调用 pthread_getspecific() 。该函数的参数为前面提到的 pthread_key_t 变量，该函数返回 void * 类型的值。

下面是前面提到的函数的原型：

int pthread_setspecific(pthread_key_t key, const void *value);

void *pthread_getspecific(pthread_key_t key);

int pthread_key_create(pthread_key_t *key, void (*destructor)(void*));

下面是一个如何使用线程存储的例子：

 

#include <malloc.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
static pthread_key_t log_key;

void write_log(const char* msg){
	FILE* fp = (FILE*)pthread_getspecific(log_key);
	fprintf(fp, "get a msg:%s\n", msg);
}

void* thread_func(void* args){
	static int cnt = 0; 
	char fn[32]; 
	//要在当前目录创建log子目录，否则创建文件会失败
	//sprintf(fn, "log/thread.%d.log", (unsigned int)pthread_self());
	sprintf(fn, "log/thread.%d.log", ++cnt);
	FILE* fp = fopen(fn, "w");		
	if(!fp){
		fprintf(stderr, "open file %s error !\n", fn);
		return NULL;
	}
	pthread_setspecific(log_key, fp);
	char msg[64]; 
	sprintf(msg, "I am %s\n", fn);
	write_log(msg);
}
void close_log_file(void* log_file){
	fclose((FILE*)log_file);
}
int main(){
	const int n = 10; 
	pthread_t pids[n]; 
	pthread_key_create(&log_key, close_log_file);
	for(int i = 0; i < n; i++){
		pthread_create(pids+i, NULL, thread_func, NULL);
	}
	for(int i = 0; i < n; i++){
		pthread_join(pids[i], NULL);
	}
	return 0; 
}