Linux系統編程—信號量

你們知道，互斥鎖能夠用於線程間同步，可是，每次只能有一個線程搶到互斥鎖，這樣限制了程序的併發行。若是咱們但願容許多個線程同時訪問同一個資源，那麼使用互斥鎖是沒有辦法實現的，只能互斥鎖會將整個共享資源鎖住，只容許一個線程訪問。

這種現象，使得線程依次輪流運行，也就是線程從並行執行變成了串行執行，這樣與直接使用單進程無異。

因而，Linux系統提出了信號量的概念。這是一種相對比較折中的處理方式，它既能保證線程間同步，數據不混亂，又能提升線程的併發性。注意，這裏提到的信號量，與咱們所學的信號沒有一點關係，就好比Java與JavaScript沒有任何關係同樣。

主要應用函數：

​ sem_init函數
​ sem_destroy函數
​ sem_wait函數
​ sem_trywait函數
​ sem_timedwait函數
​ sem_post函數
以上6 個函數的返回值都是：成功返回0， 失敗返回-1，同時設置errno。

細心的讀者可能留意到，它們沒有pthread前綴，這說明信號量不只能夠用在線程間，也能夠用在進程間。

sem_t數據類型，其本質還是結構體。可是相似於文件描述符同樣，咱們在應用期間可簡單將它看做爲整數，而忽略實現細節。

使用方法：sem_t sem; 咱們約定，信號量sem不能小於0。使用時，注意包含頭文件 <semaphore.h>。

相似於互斥鎖，信號量也有相似加鎖和解鎖的操做，加鎖使用sem_wait函數，解鎖使用sem_post函數。這兩個函數有以下特性：

1. 調用sem_post時，若是信號量大於0，則信號量減一；
2. 當信號量等於0時，調用sem_post時將形成線程阻塞；
3. 調用sem_post時，將信號量加一，同時喚醒阻塞在信號量上的線程。

上面提到的對線程的加一減一操做，因爲sem_t的實現對用戶隱藏，因此這兩個操做只能經過函數來實現，而不能直接使用++、--符號來操做。

sem_init函數

函數原型：
int sem_init(sem_t *sem, int pshared, unsigned int value);

函數做用：
初始化一個信號量；

參數說明：
sem：信號量 ；
pshared：取0時，信號量用於線程間同步；取非0（通常爲1）時則用於進程間同步；
value：指定信號量初值，而信號量的初值，決定了容許同時佔用信號量的線程的個數。

sem_destroy函數

函數原型：
int sem_destroy(sem_t *sem);

函數做用：
銷燬一個信號量

sem_wait函數

函數原型：
int sem_wait(sem_t *sem);

函數做用：
給信號量值加一

sem_post函數

函數原型：
int sem_post(sem_t *sem);

函數做用：
給信號量值減一

sem_trywait函數

函數原型：
int sem_trywait(sem_t *sem);

函數做用：
嘗試對信號量加鎖，與pthread_mutex_trylock相似；

sem_timedwait函數

函數原型：
int sem_timedwait(sem_t *sem, const struct timespec *abs_timeout);

函數做用：
限時嘗試對信號量加鎖

參數說明：
sem：信號量；
abs_timeout：與pthread_cond_timedwait同樣，採用的是絕對時間。

用法以下（例如超時時間設爲1秒）：

time_t cur = time(NULL); 獲取當前時間。
struct timespec t; 定義timespec 結構體變量t
t.tv_sec = cur+1; 定時1秒
t.tv_nsec = t.tv_sec +100;
sem_timedwait(&sem, &t); 傳參

生產者消費者信號量模型

/*信號量實現 生產者 消費者問題*/

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

#define NUM 5               

int queue[NUM];                                     //全局數組實現環形隊列
sem_t blank_number, product_number;                 //空格子信號量, 產品信號量

void *producer(void *arg)
{
    int i = 0;

    while (1) {
        sem_wait(&blank_number);                    //生產者將空格子數--,爲0則阻塞等待
        queue[i] = rand() % 1000 + 1;               //生產一個產品
        printf("----Produce---%d\n", queue[i]);        
        sem_post(&product_number);                  //將產品數++

        i = (i+1) % NUM;                            //藉助下標實現環形
        sleep(rand()%3);
    }
}

void *consumer(void *arg)
{
    int i = 0;

    while (1) {
        sem_wait(&product_number);                  //消費者將產品數--,爲0則阻塞等待
        printf("-Consume---%d\n", queue[i]);
        queue[i] = 0;                               //消費一個產品 
        sem_post(&blank_number);                    //消費掉之後,將空格子數++

        i = (i+1) % NUM;
        sleep(rand()%3);
    }
}

int main(int argc, char *argv[])
{
    pthread_t pid, cid;

    sem_init(&blank_number, 0, NUM);                //初始化空格子信號量爲5
    sem_init(&product_number, 0, 0);                //產品數爲0

    pthread_create(&pid, NULL, producer, NULL);
    pthread_create(&cid, NULL, consumer, NULL);

    pthread_join(pid, NULL);
    pthread_join(cid, NULL);

    sem_destroy(&blank_number);
    sem_destroy(&product_number);

    return 0;
}

運行結果：

更多精彩內容，請關注公衆號良許Linux，公衆內回覆1024可免費得到5T技術資料，包括：Linux，C/C++，Python，樹莓派，嵌入式，Java，人工智能，等等。公衆號內回覆進羣，邀請您進高手如雲技術交流羣。

---

公衆號：良許Linux

有收穫？但願老鐵們來個三連擊，給更多的人看到這篇文章