互斥锁
互斥锁用来保证一段时间内只有一个线程在执行一段代码。
我们先看下面一段代码。这是一个读/写程序,它们公用一个缓冲区,并且我们假定一个缓冲区只能保存一条信息。即缓冲区只有两个状态:有信息或没有信息。
void reader_function ( void );
void writer_function ( void );
char buffer;
int buffer_has_item=0;
pthread_mutex_t mutex;
struct timespec delay;
void main ( void ){
pthread_t reader;
/* 定义延迟时间*/
delay.tv_sec = 2;
delay.tv_nec = 0;
/* 用默认属性初始化一个互斥锁对象*/
pthread_mutex_init (&mutex,NULL);
//创建线程任务
pthread_create(&reader, pthread_attr_default, (void *)&reader_function), NULL);
writer_function( );
}
void writer_function (void){
while(1){
/* 锁定互斥锁*/
pthread_mutex_lock (&mutex);
if (buffer_has_item==0){
buffer=make_new_item( );
buffer_has_item=1;
}
/* 打开互斥锁*/
pthread_mutex_unlock(&mutex);
pthread_delay_np(&delay);
}
}
void reader_function(void){
while(1){
pthread_mutex_lock(&mutex);
if(buffer_has_item==1){
consume_item(buffer);
buffer_has_item=0;
}
pthread_mutex_unlock(&mutex);
pthread_delay_np(&delay);
}
}
这里声明了互斥锁变量mutex,结构pthread_mutex_t
为不公开的数据类型,其中包含一个系统分配的属性对象。
函数pthread_mutex_init
用来生成一个互斥锁。NULL参数表明使用默认属性。
pthread_mutex_lock
声明开始用互斥锁上锁,此后的代码直至调用pthread_mutex_unlock
为止,均被上锁,即同一时间只能被一个线程调用执行。当一个线程执行到pthread_mutex_lock
处时,如果该锁此时被另一个线程使用,那此线程被==阻塞==,即程序将等待到另一个线程释放此互斥锁。在上面的例子中,我们使用了==pthread_delay_np函数,让线程睡眠一段时间==,就是为了防止一个线程始终占据此函数。
- 死锁情况
需要提出的是在使用互斥锁的过程中很有可能会出现死锁:
两个线程试图同时占用两个资源,并按不同的次序锁定相应的互斥锁。
例如两个线程都需要锁定互斥锁1和互斥锁2,a线程先锁定互斥锁1,b线程先锁定互斥锁2,这时就出现了死锁。
此时我们可以使用函数pthread_mutex_trylock
,它是函数pthread_mutex_lock
的非阻塞版本,当它发现死锁不可避免时,它会返回相应的信息,程序员可以针对死锁做出相应的处理。另外不同的互斥锁类型对死锁的处理不一样,但最主要的还是要程序员自己在程序设计注意这一点。
条件变量
我们讲述了如何使用互斥锁来实现线程间数据的共享和通信,互斥锁一个明显的缺点是它只有两种状态:==锁定和非锁定==。
条件变量通过允许线程阻塞和等待另一个线程发送信号的方法弥补了互斥锁的不足,它常和互斥锁一起使用。使用时,条件变量被用来阻塞一个线程,当条件不满足时,线程往往解开相应的互斥锁并等待条件发生变化。一旦其它的某个线程改变了条件变量,它将==通知相应的条件变量唤醒一个或多个正被此条件变量阻塞的线程==。这些线程将重新锁定互斥锁并重新测试条件是否满足。一般说来,条件变量被用来进行线承间的同步。
条件变量的结构为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_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()和函数
/* 使用boost中的条件变量和互斥锁模仿信号量的一个简单的例子。*/
class Semaphore {
public:
Semaphore(long count = 0)
: count_(count) {
}
void Signal()
{
boost::unique_lock lock(mutex_);
++count_;
cv_.notify_one();
}
void Wait()
{
boost::unique_lock lock(mutex_);
cv_.wait(lock, boost::bind(&Semaphore::is_full, this));
--count_;
}
bool is_full()
{
return count_ > 0;
}
private:
boost::mutex mutex_;
boost::condition_variable cv_;
long count_;
};
count值为0时,Wait函数在Wait
处被阻塞,并打开互斥锁lock。此时,当调用到函数Signal
时,notify_one()
函数改变条件变量,告知Wait()
停止阻塞。
函数pthread_cond_broadcast(pthread_cond_t *cond)
用来唤醒所有被阻塞在条件变量cond上的线程。这些线程被唤醒后将再次竞争相应的互斥锁,所以必须小心使用这个函数。
信号量
信号量本质上是一个非负的整数计数器,它被用来控制对公共资源的访问。当公共资源增加时,调用函数sem_post()增加信号量。只有当信号量值大于0时,才能使用公共资源,使用后,函数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不为0时此信号量在进程间共享,否则只能为当前进程的所有线程共享;
- 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
#include
#include
#define MAXSTACK 100
int stack[MAXSTACK][2];
int size=0;
sem_t sem;
/* 从文件1.dat读取数据,每读一次,信号量加一*/
void *ReadData1(void *ptr){
FILE *fp=fopen("1.txt","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 *ptr){
FILE *fp=fopen("2.txt","r");
while(!feof(fp)){
fscanf(fp,"%d %d",&stack[size][0],&stack[size][1]);
sem_post(&sem);
++size;
}
fclose(fp);
}
/*阻塞等待缓冲区有数据,读取数据后,释放空间,继续等待*/
void *HandleData1(void *ptr){
while(1){
sem_wait(&sem);
printf("Plus:%d+%d=%d\n",stack[size][0],stack[size][1],
stack[size][0]+stack[size][1]);
--size;
}
}
void *HandleData2(void *ptr){
while(1){
sem_wait(&sem);
printf("Multiply:%d*%d=%d\n",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,HandleData1,NULL);
pthread_create(&t2,NULL,HandleData2,NULL);
pthread_create(&t3,NULL,ReadData1,NULL);
pthread_create(&t4,NULL,ReadData2,NULL);
/* 防止程序过早退出,让它在此无限期等待*/
pthread_join(t1,NULL);
pthread_join(t2,NULL);
pthread_join(t3,NULL);
pthread_join(t4,NULL);
}
在某种场景下信号量是可以和条件变量互换使用的
信号量和条件变量和线程锁的区别:
- 锁必须在同一个线程中获取或者释放,否则就会死锁,而条件变量和信号量则不必。
- 信号的递增和减少会被系统自动记住,系统内部有一个计数器实现信号量,不必担心会丢失,而唤醒一个条件变量时如果没有相应的线程在等待该条件变量,这次唤醒将丢失。