首页 > 代码库 > (转载)Linux 多线程条件变量同步

(转载)Linux 多线程条件变量同步

条件变量是线程同步的另一种方式,实际上,条件变量是信号量的底层实现,这也就意味着,使用条件变量可以拥有更大的自由度,同时也就需要更加小心的进行同步操作。条件变量使用的条件本身是需要使用互斥量进行保护的,线程在改变条件状态之前必须首先锁住互斥量,其他线程在获得互斥量之前不会察觉到这种改变,因为互斥量必须在锁定之后才能计算条件。

模型

#include<pthread.h>
pthread_cond_t cond              //准备条件变量
pthread_cond_t cond = PTHREAD_COND_INITIALIZER;     //初始化静态的条件变量
pthread_cond_init()         //初始化一个动态的条件变量
pthread_cond_wait()         //等待条件变量变为真
pthread_cond_timedwait()    //等待条件变量变为真,等待有时间限制。
pthread_cond_signal()       //至少唤醒一个等待该条件的线程
pthread_cond_broadcast()    //唤醒等待该条件的所有线程
pthread_cond_destroy()      //销毁一个条件变量

pthread_cond_init()

//初始化一个动态的条件变量
//成功返回0,失败返回error number
int pthread_cond_init(pthread_cond_t *restrict cond, const pthread_condattr_t *restrict attr);

cond:条件变量指针,这里使用了restrict关键字
attr:条件变量属性指针,默认属性赋NULL

pthread_cond_wait() / pthread_cond_timedwait()

//等待条件变量为真。收到pthread_cond_broadcast()或pthread_cond_signal()就唤醒
//成功返回0,失败返回error number
int pthread_cond_wait(pthread_cond_t *restrict cond, pthread_mutex_t *restrict mutex);
int pthread_cond_timedwait(pthread_cond_t *restrict cond, pthread_mutex_t *restrict mutex, const struct timespec *restrict abstime);

条件变量的使在多线程的程序中,因为各个线程都可以访问大部分的进程资源,所以我们为了保证公有资源的使用是可以控制的,在一个线程开始使用公有资源之前要尝试获取互斥锁,在使用完毕之后要释放互斥锁,这样就能保证每个公共资源在一个时刻都只能被一个线程使用,在一定程度上得到了控制,但这并不能解决同步的问题,考虑如下两个线程:

Q=create_queue();
pthread_t mutex
//线程A,入队
while(1){
    lock(mutex);
    in_queue(Q);
    unlock(mutex);
}

//线程B,出队
while(1){
    lock(mutex);
    out_queue(Q);
    unlock(mutex);
}

上述代码可以实现两个线程的互斥,即同一时刻只有一个线程在使用公有资源-队列。但如果线程B获取了锁,但队列中是空的,它的out_queue()也就是没有意义的,所以我们这里更需要一种方法将两个线程进行同步:只有当队列中有数据的时候才进行出队。
我们设计这样一种逻辑:

//线程B,出队
while(1){
    lock(mutex);
    if(队列是空,线程不应该执行){
        释放锁;
        continue;
    }
    out_queue(Q);
    unlock(mutex);
}

这个程序就解决了上述的问题,即便线程B抢到了互斥锁,但是如果队列是空的,他就释放锁让两个线程重新抢锁,希望这次线程A能抢到并往里放一个数据。
但这个逻辑还有一个问题,就是多线程并发的问题,很有可能发生的一种情况是:线程B抢到了锁,发现没有数据,释放锁->线程A立即抢到了锁并往里放了一个数据->线程B执行continue,显然,这种情况下是不应该continue的,因为线程B想要的条件在释放锁之后立即就被满足了,它错过了条件。
So,我们想一种反过来的逻辑:

//线程B,出队
while(1){
    lock(mutex);
    if(队列是空,线程不应该执行){
        continue;
        释放锁;
    }
    out_queue(Q);
    unlock(mutex);
}

显然这种方法有个致命的问题:一旦continue了,线程B自己获得锁就没有被释放,这样线程A不可能抢到锁,而B继续加锁就会形成死锁!
Finaly,我们希望看到一个函数fcn,如果条件不满足,能同时释放锁+停止执行线程,如果条件满足,自己当时获得的锁还在

//线程B,出队
while(1){
    lock(mutex);
    fcn(当前线程不应该执行,mutex)     //if(当前线程不应该执行){释放锁“同时” continue;}
    out_queue(Q);
    unlock(mutex);
}

OK,这个就是pthread_cond_wait()的原理了,只不过它把continue变成了"休眠"这种由OS负责的操作,可以大大的节约资源。
当然,线程执行的条件是很难当作参数传入一个函数的,POSIX多线程的模型使用系统提供的"条件变量"+"我们自己定义的具体条件" 来确定一个线程是否应该执行接下来的内容。"条件变量"只有,所以一种典型的多线程同步的结构如下

//线程B,出队
while(1){
    lock(mutex);
    while(条件不满足)
        pthread_cond_wait(cond,mutex)
        //获得互斥锁可以同时保护while里的条件和cond的判断,二者总是用一把锁保护,并一同释放  
        //cond为假,就休眠同时释放锁,等待被cond为真唤醒,把自己获得的锁拿回来           
        //拿回自己的锁再检查线程执行条件,条件不满足继续循环,直到条件满足跳出循环
        //这个函数是带着"线程的执行条件为真"+"cond为真"走出循环的
        //这个函数返回后cond被重新设置为0
    out_queue(Q);
    unlock(mutex);
}

pthread_cond_braoadcast()/pthread_cond_signal()

//使条件变量为真并唤醒wait中的线程,前者唤醒所有wait的,后者唤醒一个
//成功返回0,失败返回error number
int pthread_cond_broadcast(pthread_cond_t *cond);
int pthread_cond_signal(pthread_cond_t *cond);

pthread_cond_destroy()

//销毁条件变量
//成功返回0,失败返回error number
int pthread_cond_destroy(pthread_cond_t *cond);

例子-线程池

\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\//thread.h
#ifndef __THREAD_H__
#define __THREAD_H__
#define THREAD_NUM 3
#define TASK_NUM 100


typedef struct{//每个节点的封装格式,fcn是用户自定义函数,arg是用户自定义函数参数指针,保证通用性声明为void
    void* (*fcn)(void* arg);
    void* arg;
}task_t;
typedef struct{        //用户自定义函数的参数结构体
    int x;
}argfcn_t;

//#define LQ_DATA_T task_t*
#define LQ_DATA_T task_t

#endif  //__THREAD_H__
\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\//lqueue.c
#include"thread.h"
#include"lqueue.h"
#include<stdlib.h>
...
\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\//thread_pool.c
#include<stdio.h>   
#include<stdlib.h>
#include<string.h>
#include<pthread.h>
#include"thread.h"
#include"lqueue.h"

//互斥量和条件变量
pthread_mutex_t lock;
pthread_cond_t cond;


//全局的表
lqueue_t* Q;

//每个线程的任务,必须是这个形式的函数指针
void* do_task(void* p){   
    task_t data;
    int ret=0;
    while(1){
        pthread_mutex_lock(&lock);
        while(is_empty_lqueue(Q)){          //大家收到广播,因为延迟,可能醒了好几个,要判断一下是不是自己
            pthread_cond_wait(&cond,&lock);     //先抢到锁再醒
        }
        ret=out_lqueue(Q,&data);
        pthread_mutex_unlock(&lock);
        data.fcn(data.arg);
    }
}

//创建线程池
void create_pool(void){
    //初始化队列
    Q=create_lqueue();
    //初始化互斥量
    pthread_mutex_init(&lock,NULL);
    //初始化条件变量
    pthread_cond_init(&cond,NULL);
    int i=THREAD_NUM;
    pthread_t tid[THREAD_NUM];
    while(i--)
        pthread_create(&tid[i],NULL,do_task,NULL);
}



//准备函数
void* fcn(void* parg){                //用户自定义的需要线程执行的函数
    argfcn_t* i=(argfcn_t*)parg;
    printf("this is task1\n");
    printf("task1:%d\n",i->x);
}

//添加任务
void pool_add_task(void*(*pfcn)(void*parg),void*arg){
    task_t task;
    task.fcn=pfcn;
    task.arg=arg;

    in_lqueue(Q,task);
    pthread_cond_signal(&cond); //添加了一个任务,用signal更好
}

int main(int argc, const char *argv[])
{
    //创建线程池
    create_pool();

    //准备参数
    argfcn_t argfcn;
    argfcn.x=5;

    //添加任务
    pool_add_task(fcn,(void*)&argfcn);  
    pool_add_task(fcn,(void*)&argfcn);  
    pool_add_task(fcn,(void*)&argfcn);  
    pause();    
    return 0;
}

(转载)Linux 多线程条件变量同步