在Linux 多執行緒程式設計中,執行緒同步是一個非常重要的問題。如果執行緒之間沒有正確地同步,就會導致程式出現一些意外的問題,例如:
接下來將介紹互斥鎖、條件變數、號誌、讀寫鎖這幾種執行緒同步方法,並使用C語言程式碼範例說明其使用方法。
互斥鎖是一種用於執行緒同步的鎖,用於保護共用資源。只有擁有該鎖的執行緒才能存取共用資源,其他執行緒需要等待鎖被釋放後才能繼續執行。
在Linux環境下,我們可以使用pthread庫提供的互斥鎖函數來實現互斥鎖機制。以下是一些常用的互斥鎖函數:
函數名 | 描述 |
---|---|
pthread_mutex_init |
初始化互斥鎖 |
pthread_mutex_lock |
加鎖互斥鎖 |
pthread_mutex_trylock |
嘗試加鎖互斥鎖 |
pthread_mutex_unlock |
解鎖互斥鎖 |
pthread_mutex_destroy |
銷燬互斥鎖 |
在使用互斥鎖之前,需要先初始化互斥鎖。pthread_mutex_init函數用於初始化一個互斥鎖。函數原型如下:
int pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr);
其中,mutex引數是一個指向pthread_mutex_t結構體的指標,用於指定要初始化的互斥鎖;attr引數是一個指向pthread_mutexattr_t結構體的指標,用於指定互斥鎖的屬性,通常設定為NULL。
以下是一個初始化互斥鎖的例子:
#include <pthread.h>
pthread_mutex_t mutex;
int main()
{
// 初始化互斥鎖
pthread_mutex_init(&mutex, NULL);
// ...
// 銷燬互斥鎖
pthread_mutex_destroy(&mutex);
return 0;
}
加鎖互斥鎖用於保證同一時刻只有一個執行緒能夠存取共用資源。pthread_mutex_lock函數用於加鎖一個互斥鎖。函數原型如下:
int pthread_mutex_lock(pthread_mutex_t *mutex);
其中,mutex引數是一個指向pthread_mutex_t結構體的指標,用於指定要加鎖的互斥鎖。
以下是一個加鎖互斥鎖的例子:
#include <pthread.h>
pthread_mutex_t mutex;
void* thread_func(void* arg)
{
// 加鎖互斥鎖
pthread_mutex_lock(&mutex);
// 存取共用資源
// ...
// 解鎖互斥鎖
pthread_mutex_unlock(&mutex);
return NULL;
}
int main()
{
// 初始化互斥鎖
pthread_mutex_init(&mutex, NULL);
// 建立執行緒
pthread_t tid;
pthread_create(&tid, NULL, thread_func, NULL);
// ...
// 等待執行緒結束
pthread_join(tid, NULL);
// 銷燬互斥鎖
pthread_mutex_destroy(&mutex);
return 0;
}
嘗試加鎖互斥鎖與加鎖互斥鎖的主要區別在於,如果互斥鎖已經被其他執行緒鎖定了,嘗試加鎖互斥鎖將不會阻塞當前執行緒,而是會立即返回一個錯誤程式碼。函數原型如下:
int pthread_mutex_trylock(pthread_mutex_t *mutex);
其中,mutex引數是一個指向pthread_mutex_t結構體的指標,用於指定要嘗試加鎖的互斥鎖。
以下是一個嘗試加鎖互斥鎖的例子:
#include <pthread.h>
pthread_mutex_t mutex;
void* thread_func(void* arg)
{
// 嘗試加鎖互斥鎖
int ret = pthread_mutex_trylock(&mutex);
if (ret == 0) {
// 存取共用資源
// ...
// 解鎖互斥鎖
pthread_mutex_unlock(&mutex);
} else {
// 互斥鎖已經被其他執行緒鎖定了
// ...
}
return NULL;
}
int main()
{
// 初始化互斥鎖
pthread_mutex_init(&mutex, NULL);
// 建立執行緒
pthread_t tid;
pthread_create(&tid, NULL, thread_func, NULL);
// ...
// 等待執行緒結束
pthread_join(tid, NULL);
// 銷燬互斥鎖
pthread_mutex_destroy(&mutex);
return 0;
}
解鎖互斥鎖用於釋放已經鎖定的互斥鎖。pthread_mutex_unlock函數用於解鎖一個互斥鎖。函數原型如下:
int pthread_mutex_unlock(pthread_mutex_t *mutex);
其中,mutex引數是一個指向pthread_mutex_t結構體的指標,用於指定要解鎖的互斥鎖。
以下是一個解鎖互斥鎖的例子:
#include <pthread.h>
pthread_mutex_t mutex;
void* thread_func(void* arg)
{
// 加鎖互斥鎖
pthread_mutex_lock(&mutex);
// 存取共用資源
//
// 解鎖互斥鎖
pthread_mutex_unlock(&mutex);
return NULL;
}
在不再需要使用互斥鎖時,需要將互斥鎖銷燬。pthread_mutex_destroy函數用於銷燬一個互斥鎖。函數原型如下:
int pthread_mutex_destroy(pthread_mutex_t *mutex);
其中,mutex引數是一個指向pthread_mutex_t結構體的指標,用於指定要銷燬的互斥鎖。
以下是一個銷燬互斥鎖的例子:
#include <pthread.h>
pthread_mutex_t mutex;
int main()
{
// 初始化互斥鎖
pthread_mutex_init(&mutex, NULL);
// ...
// 銷燬互斥鎖
pthread_mutex_destroy(&mutex);
return 0;
}
下面是一個簡單的範例程式,演示瞭如何使用互斥鎖來同步兩個執行緒的存取。
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
pthread_mutex_t mutex;
int shared_data = 0;
void *thread_func(void *arg)
{
int i;
for (i = 0; i < 1000000; i++) {
pthread_mutex_lock(&mutex);
shared_data++;
pthread_mutex_unlock(&mutex);
}
return NULL;
}
int main()
{
pthread_t thread1, thread2;
pthread_mutex_init(&mutex, NULL);
pthread_create(&thread1, NULL, thread_func, NULL);
pthread_create(&thread2, NULL, thread_func, NULL);
pthread_join(thread1, NULL);
pthread_join(thread2, NULL);
pthread_mutex_destroy(&mutex);
printf("Shared data: %d\n", shared_data);
return 0;
}
在這個程式中,thread_func
函數是兩個執行緒執行的函數,它會對shared_data
變數進行1000000次加一操作。
為了確保多個執行緒不會同時存取shared_data
變數,我們使用了一個互斥鎖。當一個執行緒要存取shared_data
變數時,它會呼叫pthread_mutex_lock
函數來加鎖。如果鎖已經被其他執行緒持有,那麼這個執行緒就會被阻塞,直到鎖被釋放為止。當執行緒完成對shared_data
變數的操作後,它會呼叫pthread_mutex_unlock
函數來釋放鎖。
在這個程式執行完畢後,我們可以通過列印shared_data
變數的值來檢查程式是否正確地同步了兩個執行緒的存取。如果程式正確地同步了執行緒的存取,那麼shared_data
變數的值應該是2000000。