Linux多線程編程入門(3)

3、條件變量

　　前一節中我們講述瞭如何使用互斥鎖來實現線程間數據的共享和通信，互斥鎖一個明顯的缺點是它只有兩種狀態：鎖定和非鎖定。而條件變量通過允許線程阻塞和等待另一個線程發送信號的方法彌補了互斥鎖的不足，它常和互斥鎖一起使用。使用時，條件變量被用來阻塞一個線程，當條件不滿足時，線程往往解開相應的互斥鎖並等待條件發生變化。一旦其它的某個線程改變了條件變量，它將通知相應的條件變量喚醒一個或多個正被此條件變量阻塞的線程。這些線程將重新鎖定互斥鎖並重新測試條件是否滿足。一般說來，條件變量被用來進行線承間的同步。

　　條件變量的結構爲pthread_cond_t，函數pthread_cond_init（）被用來初始化一個條件變量。它的原型爲：

　　extern int pthread_cond_init __P ((pthread_cond_t *__cond,__const pthread_condattr_t *__cond_attr));

　　其中cond是一個指向結構pthread_cond_t的指針，cond_attr是一個指向結構pthread_condattr_t的指針。結構 pthread_condattr_t是條件變量的屬性結構，和互斥鎖一樣我們可以用它來設置條件變量是進程內可用還是進程間可用，默認值是 PTHREAD_ PROCESS_PRIVATE，即此條件變量被同一進程內的各個線程使用。注意初始化條件變量只有未被使用時才能重新初始化或被釋放。釋放一個條件變量的函數爲pthread_cond_ destroy（pthread_cond_t cond）。　

　　函數pthread_cond_wait（）使線程阻塞在一個條件變量上。它的函數原型爲：

　　extern int pthread_cond_wait __P ((pthread_cond_t *__cond,pthread_mutex_t *__mutex));

　　線程解開mutex指向的鎖並被條件變量cond阻塞。線程可以被函數pthread_cond_signal和函數 pthread_cond_broadcast喚醒，但是要注意的是，條件變量只是起阻塞和喚醒線程的作用，具體的判斷條件還需用戶給出，例如一個變量是否爲0等等，這一點我們從後面的例子中可以看到。線程被喚醒後，它將重新檢查判斷條件是否滿足，如果還不滿足，一般說來線程應該仍阻塞在這裏，被等待被下一次喚醒。這個過程一般用while語句實現。

　　另一個用來阻塞線程的函數是pthread_cond_timedwait（），它的原型爲：
　　extern int pthread_cond_timedwait __P ((pthread_cond_t *__cond,pthread_mutex_t *__mutex, __const struct timespec *__abstime));

　　它比函數pthread_cond_wait（）多了一個時間參數，經歷abstime段時間後，即使條件變量不滿足，阻塞也被解除。

　　函數pthread_cond_signal（）的原型爲：

　　extern int pthread_cond_signal __P ((pthread_cond_t *__cond));

　　它用來釋放被阻塞在條件變量cond上的一個線程。多個線程阻塞在此條件變量上時，哪一個線程被喚醒是由線程的調度策略所決定的。要注意的是，必須用保護條件變量的互斥鎖來保護這個函數，否則條件滿足信號又可能在測試條件和調用pthread_cond_wait函數之間被髮出，從而造成無限制的等待。下面是使用函數pthread_cond_wait（）和函數pthread_cond_signal（）的一個簡單的例子。

 

  pthread_mutex_t count_lock; pthread_cond_t count_nonzero; unsigned count; decrement_count() { pthread_mutex_lock (&count_lock); while(count == 0) pthread_cond_wait( &count_nonzero, &count_lock); count = count-1; pthread_mutex_unlock (&count_lock); } increment_count() { pthread_mutex_lock(&count_lock); if(count==0) pthread_cond_signal(&count_nonzero); count = count+1; pthread_mutex_unlock(&count_lock); } 

 

      count值爲0時，decrement函數在pthread_cond_wait處被阻塞，並打開互斥鎖count_lock。此時，當調用到函數 increment_count時，pthread_cond_signal（）函數改變條件變量，告知decrement_count（）停止阻塞。讀者可以試着讓兩個線程分別運行這兩個函數，看看會出現什麼樣的結果。

　　函數pthread_cond_broadcast（pthread_cond_t *cond）用來喚醒所有被阻塞在條件變量cond上的線程。這些線程被喚醒後將再次競爭相應的互斥鎖，所以必須小心使用這個函數。

 

 

4、信號量

　　信號量本質上是一個非負的整數計數器，它被用來控制對公共資源的訪問。當公共資源增加時，調用函數sem_post（）增加信號量。只有當信號量值大於０時，才能使用公共資源，使用後，函數sem_wait（）減少信號量。函數sem_trywait（）和函數pthread_ mutex_trylock（）起同樣的作用，它是函數sem_wait（）的非阻塞版本。下面我們逐個介紹和信號量有關的一些函數，它們都在頭文件 /usr/include/semaphore.h中定義。

　　信號量的數據類型爲結構sem_t，它本質上是一個長整型的數。函數sem_init（）用來初始化一個信號量。它的原型爲：

　　extern int sem_init __P ((sem_t *__sem, int __pshared, unsigned int __value));

　　sem爲指向信號量結構的一個指針；pshared不爲０時此信號量在進程間共享，否則只能爲當前進程的所有線程共享；value給出了信號量的初始值。

　　函數sem_post( sem_t *sem )用來增加信號量的值。當有線程阻塞在這個信號量上時，調用這個函數會使其中的一個線程不在阻塞，選擇機制同樣是由線程的調度策略決定的。

　　函數sem_wait( sem_t *sem )被用來阻塞當前線程直到信號量sem的值大於0，解除阻塞後將sem的值減一，表明公共資源經使用後減少。函數sem_trywait ( sem_t *sem )是函數sem_wait（）的非阻塞版本，它直接將信號量sem的值減一。

　　函數sem_destroy(sem_t *sem)用來釋放信號量sem。

　　下面我們來看一個使用信號量的例子。在這個例子中，一共有4個線程，其中兩個線程負責從文件讀取數據到公共的緩衝區，另兩個線程從緩衝區讀取數據作不同的處理（加和乘運算）。

 

 

//File sem.c #include <stdio.h> #include <pthread.h> #include <semaphore.h> #define MAXSTACK 100 int stack[MAXSTACK][2]; int size = 0; sem_t sem; //從文件1.dat讀取數據，每讀一次，信號量加一 void ReadData1(void) { FILE *fp=fopen("1.dat","r"); while(!feof(fp)) { fscanf(fp,"%d %d",&stack[size][0],&stack[size][1]); sem_post(&sem); ++size; } fclose(fp); } //從文件2.dat讀取數據 void ReadData2(void) { FILE *fp=fopen("2.dat","r"); while(!feof(fp)) { fscanf(fp,"%d %d",&stack[size][0],&stack[size][1]); sem_post(&sem); ++size; } fclose(fp); } //阻塞等待緩衝區有數據，讀取數據後，釋放空間，繼續等待 void HandleData1(void) { while(1) { sem_wait(&sem); printf("Plus:%d+%d=%dn",stack[size][0],stack[size][1], stack[size][0]+stack[size][1]); --size; } } void HandleData2(void) { while(1) { sem_wait(&sem); printf("Multiply:%d*%d=%dn",stack[size][0],stack[size][1], stack[size][0]*stack[size][1]); --size; } } int main(void) { pthread_t t1,t2,t3,t4; sem_init(&sem,0,0); pthread_create(&t1,NULL,(void *)HandleData1,NULL); pthread_create(&t2,NULL,(void *)HandleData2,NULL); pthread_create(&t3,NULL,(void *)ReadData1,NULL); pthread_create(&t4,NULL,(void *)ReadData2,NULL); //防止程序過早退出，讓它在此無限期等待 pthread_join(t1,NULL); }  

   

      在Linux下，我們用命令 gcc -lpthread sem.c -o sem 生成可執行文件sem。我們事先編輯好數據文件1.dat和2.dat，假設它們的內容分別爲1 2 3 4 5 6 7 8 9 10和 -1 -2 -3 -4 -5 -6 -7 -8 -9 -10 ，我們運行sem，得到如下的結果：

 

Multiply: -1*-2=2
Plus: -1+-2=-3
Multiply: 9*10=90
Plus: -9+-10=-19
Multiply: -7*-8=56
Plus: -5+-6=-11
Multiply: -3*-4=12
Plus: 9+10=19
Plus: 7+8=15
Plus: 5+6=11

　　

      從中我們可以看出各個線程間的競爭關係。而數值並未按我們原先的順序顯示出來這是由於size這個數值被各個線程任意修改的緣故。這也往往是多線程編程要注意的問題。

 

小結

　　多線程編程是一個很有意思也很有用的技術，使用多線程技術的網絡螞蟻是目前最常用的下載工具之一，使用多線程技術的grep比單線程的grep要快上幾倍，類似的例子還有很多。希望大家能用多線程技術寫出高效實用的好程序來。