1. 程式人生 > >Linux系統程式設計——執行緒私有資料

Linux系統程式設計——執行緒私有資料

在多執行緒程式中,經常要用全域性變數來實現多個函式間的資料共享。由於資料空間是共享的,因此全域性變數也為所有執行緒共有。

測試程式碼如下:

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

int key = 100; //全域性變數

void *helloworld_one(void *arg)
{
	printf("the message is %s\n",(char *)arg);
	key = 10;
	printf("key=%d, the child id is %lu\n", key, pthread_self());
	
	return NULL;
}

void *helloworld_two(void *arg)
{
	printf("the message is %s\n", (char *)arg);
	sleep(1);
	printf("key=%d, the child id is %lu\n", key, pthread_self());
	
	return NULL;
}

int main(int argc, char *argv[])
{
	pthread_t thread_id_one;
	pthread_t thread_id_two;

	//建立執行緒
	pthread_create(&thread_id_one, NULL, helloworld_one, "helloworld_one");
	pthread_create(&thread_id_two, NULL, helloworld_two, "helloworld_two");
	
	//等待執行緒結束,回收資源
	pthread_join(thread_id_one, NULL);
	pthread_join(thread_id_two, NULL);
	
	return 0;
}

執行結果如下:

由執行結果可以看出,其中一個執行緒對全域性變數的修改將影響到另一個執行緒的訪問

但有時應用程式設計中必要提供執行緒私有的全域性變數,這個變數僅線上程中有效,但卻可以跨過多個函式訪問。比如在程式裡可能需要每個執行緒維護一個連結串列,而會使用相同的函式來操作這個連結串列,最簡單的方法就是使用同名而不同變數地址的執行緒相關資料結構。這樣的資料結構可以由 Posix 執行緒庫維護,成為執行緒私有資料 (Thread-specific Data,或稱為 TSD)。

1)建立執行緒私有資料

所需標頭檔案:

#include <pthread.h> 

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

功能:

建立一個型別為 pthread_key_t 型別的私有資料變數( key )。

引數:

key:在分配( malloc )執行緒私有資料之前,需要建立和執行緒私有資料相關聯的鍵( key ),這個鍵的功能是獲得對執行緒私有資料的訪問權。

destructor:清理函式名字( 如:fun )。當執行緒退出時,如果執行緒私有資料地址不是非 NULL,此函式會自動被呼叫。該函式指標可以設成 NULL ,這樣系統將呼叫預設的清理函式。

回撥函式其定義如下:

void fun(void *arg)

{// arg 為 key 值
}

返回值:

成功:0

失敗:非 0

2)登出執行緒私有資料

int pthread_key_delete(pthread_key_t key);

功能:

登出執行緒私有資料。這個函式並不會檢查當前是否有執行緒正使用執行緒私有資料( key ),也不會呼叫清理函式 destructor() ,而只是將執行緒私有資料( key )釋放以供下一次呼叫 pthread_key_create() 使用。

引數:

key:待登出的私有資料。

返回值:

成功:0

失敗:非 0

3)設定執行緒私有資料的關聯

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

功能:

設定執行緒私有資料( key ) 和 value 關聯,注意,是 value 的值(不是所指的內容)和 key 相關聯。

引數:

key:執行緒私有資料。

value:和 key 相關聯的指標。

返回值:

成功:0

失敗:非 0

4)讀取執行緒私有資料所關聯的值

void *pthread_getspecific(pthread_key_t key);

功能:

讀取執行緒私有資料( key )所關聯的值。

引數:

key:執行緒私有資料。

返回值:

成功:執行緒私有資料( key )所關聯的值。

失敗:NULL

示例程式碼如下:

// this is the test code for pthread_key 
#include <stdio.h> 
#include <pthread.h> 

pthread_key_t key;	// 私有資料,全域性變數

void echomsg(void *t) 
{ 
	printf("[destructor] thread_id = %lu, param = %p\n", pthread_self(), t); 
} 

void *child1(void *arg) 
{ 
	int i = 10;
	
	pthread_t tid = pthread_self(); //執行緒號
	printf("\nset key value %d in thread %lu\n", i, tid); 
	
	pthread_setspecific(key, &i); // 設定私有資料
	
	printf("thread one sleep 2 until thread two finish\n\n");
	sleep(2); 
	printf("\nthread %lu returns %d, add is %p\n",
		tid, *((int *)pthread_getspecific(key)), pthread_getspecific(key) ); 
} 

void *child2(void *arg) 
{ 
	int temp = 20;
	
	pthread_t tid = pthread_self();  //執行緒號
	printf("\nset key value %d in thread %lu\n", temp, tid); 
	
	pthread_setspecific(key, &temp); //設定私有資料
	
	sleep(1); 
	printf("thread %lu returns %d, add is %p\n", 
		tid, *((int *)pthread_getspecific(key)), pthread_getspecific(key)); 
} 

int main(void) 
{ 
	pthread_t tid1,tid2; 
	pthread_key_create(&key, echomsg); // 建立
	
	pthread_create(&tid1, NULL, child1, NULL); 
	pthread_create(&tid2, NULL, child2, NULL); 
	pthread_join(tid1, NULL);
	pthread_join(tid2, NULL);
	
	pthread_key_delete(key); // 登出
	
	return 0; 
} 

執行結果如下:

從執行結果來看,各執行緒對自己的私有資料操作互不影響。也就是說,雖然 key 是同名且全域性,但訪問的記憶體空間並不是同一個。