【linux】线程同步

互斥锁

主要应用函数

• pthread_mutex_init

  • 静态初始化：如果互斥锁 mutex 是静态分配的（定义在全局，或加了static关键字修饰），可以直接使用宏进行初始化。 pthead_mutex_t muetx = PTHREAD_MUTEX_INITIALIZER;

  • 动态初始化：局部变量应采用动态初始化。pthread_mutex_init(&mutex, NULL)

• pthread_mutex_lock
  函数加锁。可理解为将mutex–（或-1）
  int pthread_mutex_lock(pthread_mutex_t *mutex);

• pthread_mutex_unlock
  函数解锁。可理解为将mutex ++（或+1）
  int pthread_mutex_unlock(pthread_mutex_t *mutex);

• pthread_mutex_t 类型主要用于定义互斥量

读写锁

• 允许更高的并行性
• 写独占，读共享
• 写比读优先

主要应用函数

• pthread_rwlock_init函数
• pthread_rwlock_destroy函数
• pthread_rwlock_rdlock函数
• pthread_rwlock_wrlock函数
• pthread_rwlock_tryrdlock函数
• pthread_rwlock_trywrlock函数
• pthread_rwlock_unlock函数

条件变量

条件变量本身不是锁。但它也可以造成线程阻塞。通常与互斥锁配合使用。

主要应用函数

• pthread_cond_init函数

• pthread_cond_destroy函数

• pthread_cond_wait函数阻塞等待一个条件变量
  int pthread_cond_wait(pthread_cond_t *restrict cond, pthread_mutex_t *restrict mutex);

  • 阻塞等待条件变量cond（参1）满足
  • 释放已掌握的互斥锁（解锁互斥量）相当于pthread_mutex_unlock(&mutex);
  • 当被唤醒，pthread_cond_wait函数返回时，解除阻塞并重新申请获取互斥锁pthread_mutex_lock(&mutex);

• pthread_cond_timedwait函数

• pthread_cond_signal函数：唤醒至少一个阻塞在条件变量上的线程

• pthread_cond_broadcast函数：唤醒全部阻塞在条件变量上的线程

• pthread_cond_t类型 用于定义条件变量

生产者消费者问题代码

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

struct msg { //利用链表存储，一个msg代表一个节点
	int val;
	msg* next;
};

struct msg *head, *mp;

pthread_cond_t has_product = PTHREAD_COND_INITIALIZER; //init
pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER; //init

void *producer(void *arg) {
	while(1) {
		mp = (struct msg*)malloc(sizeof(struct msg));
		mp->val = rand() % 1000 + 1; // 产生随机数，代表生产一个产品的代号
		printf("producer:%d\n", mp->val);
		
		pthread_mutex_lock(&lock); //上锁
		mp->next = head; //利用头插法将节点插入链表
		head = mp; //
		pthread_mutex_unlock(&lock);//解锁
		
		pthread_cond_signal(&has_product); //唤醒阻塞在has_product条件变量上的线程
		
		sleep(rand() % 3);
		
	}
	
}

void *consumer(void *arg) {
	while(1) {
		pthread_mutex_lock(&lock); // 上锁
		while(head == NULL) {  // 当没有产品时，阻塞条件变量，并且解锁，这两步是原子操作。注意这里使用while而不是if，防止多个消费者共同抢占资源的情况
			pthread_cond_wait(&has_product, &lock); 
		}
		
		mp = head;//将头结点删去
		head = head->next; 
		pthread_mutex_unlock(&lock); //解锁
		
		printf("consumer:%d\n", mp->val);
		free(mp); //释放节点资源
		sleep(rand() %  3);
	}
	
}

int main() {
	pthread_t pid, cid;
	srand(time(NULL)); //随机数种子
	
	pthread_create(&pid, NULL, producer, NULL); //创建子线程
	pthread_create(&cid, NULL, consumer, NULL);
	
	pthread_join(pid, NULL); //释放子线程
	pthread_join(cid, NULL);
	
	return 0;
}