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函数。这两个函数有如下特性：

调用sem_post时，如果信号量大于0，则信号量减一；
当信号量等于0时，调用sem_post时将造成线程阻塞；
调用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

有收获？希望老铁们来个三连击，给更多的人看到这篇文章