Linux_线程同步_条件变量

pthread_cond_t

条件变量用于在线程之间同步共享数据的值，条件变量提供了一种线程间的通知机制：当某个共享数据达到某个值时，唤醒等待这个共享数据的线程。

条件变量使我们可以睡眠等待某种条件出现。主要包括两个动作：一个或者多个线程等待"条件变量的条件成立"而挂起；另一个线程使"条件成立"信号。

创建：

#include <pthread.h>

int pthread_cond_init(pthread_cond_t *restrict cond, pthread_condattr_t *restrict attr);

返回值：
成功，返回 0；出错，返回错误编号

销毁：

int pthread_cond_destroy(pthread_cond_t *restrict cond);

返回值：
成功，返回 0；出错，返回错误编号

尽管POSIX标准中为条件变量定义了属性，但在Linux中没有实现，因此cond_attr值通常为NULL，且被忽略。

等待：

//阻塞等待
int pthread_cond_wait(pthread_cond_t *restrict cond, pthread_mutex_t *restric mutex);

//超时等待
int pthread_cond_timedwait(pthread_cond_t *restrict cond, pthread_mutex_t *restrict mutex, const struct timespec *restrict timeout);

struct timespec {
     time_t tv_sec;  /** seconds */ 
     ong tv_nsec;    /** nanoseconds */ 
};

返回值：成功，返回 0，出错，返回错误编号

无论哪种等待方式，都必须和一个互斥锁配合，以防止多个线程同时请求。mutex互斥锁必须是普通锁（PTHREAD_MUTEX_TIMED_NP）或者适应锁（PTHREAD_MUTEX_ADAPTIVE_NP），且在调用pthread_cond_wait()前必须由本线程加锁（pthread_mutex_lock()），而在更新条件等待队列以前，mutex保持锁定状态，并在线程挂起进入等待前解锁。在条件满足从而离开pthread_cond_wait()之前，mutex将被重新加锁，以与进入pthread_cond_wait()前的加锁动作对应。

条件变量要和互斥量相联结，以避免出现条件竞争：一个线程预备等待一个条件变量，当它在真正进入等待之前，另一个线程恰好触发了该条件（条件满足信号有可能在测试条件和调用pthread_cond_wait函数（block）之间被发出，从而造成无限制的等待）。

通知操作：

//激活一个等待该条件的线程（存在多个等待线程时按入队顺序激活其中一个）
int pthread_cond_signal(pthread_cond_t *cond);
//激活所有等待线程
int pthread_cond_broadcast(pthread_cond_t *cond);

返回值：成功，返回 0；失败，返回错误编号
一定要在改变条件状态以后再给线程发送信号

等待条件变量信号的用法约定一般是这样的：

…

pthread_mutex_lock(&mutex);

…

pthread_cond_wait (&cond, &mutex);

…

pthread_mutex_unlock (&mutex);

条件变量的 pthread_cond_wait(cond, mutex) 函数内部流程：

unlock(&mutex)：释放锁

lock(&mutex)

将线程自己插入到条件变量的等待队列中

unlock(&mutex)；

当前等待的线程阻塞 <<== 等待其他线程通知唤醒（signal 或 broadcast）

在唤醒后，lock(&mutex)

从等待队列中删除线程自己

案例：一个线程计算结果，多个线程获取数据

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

/** 两个线程定义的共享资源 */
typedef struct {
    int res;
    int counter;            ///< 用于统计获取结果线程的数量
    pthread_cond_t cond;    ///< 条件变量
    pthread_mutex_t mutex;  ///< 互斥锁
}Result;


/** 计算并将结果放置在 Result 中的线程运行函数 */
void *set_fn(void *arg)
{
    Result *r = (Result *)arg;
    int i = 0;
    int sum = 0;

    for(; i <= 100; i++){
        sum += i;
    }

    /** 将结果放置到 Result 中 */
    r->res = sum;

    pthread_mutex_lock(&r->mutex);
    /** 判断获取结果的线程是否达到指定的数量 */
    while(r->counter < 2){
        pthread_mutex_unlock(&r->mutex);
        usleep(100);
        pthread_mutex_lock(&r->mutex);
    }
    pthread_mutex_unlock(&r->mutex);

    /** 通知唤醒等待的那个获取结果的线程 */
    pthread_cond_broadcast(&r->cond);

    return (void *)0;
}

/** 获得结果的线程运行函数 */
void *get_fn(void *arg)
{
    Result *r = (Result *)arg;

    /** 对两个线程共享的判断条件进行保护（加锁） */
    /** 两个线程对判断条件的操作是互斥的 */
    pthread_mutex_lock(&r->mutex);
    /** 有一个线程准备好了，则计数器 +1 */
    r->counter++;

    /** 获取结果的线程等待 */
    pthread_cond_wait(&r->cond, &r->mutex);

    /** 被唤醒后 */
    pthread_mutex_unlock(&r->mutex);

    /** 去获取计算结果 */
    int res = r->res;
    printf("0x%lx get sum is %d\n", pthread_self(), res);

    return (void *)0;
}

int main(void)
{
    int err;
    pthread_t cal, get1, get2;

    Result r;
    r.counter = 0;
    pthread_cond_init(&r.cond, NULL);
    pthread_mutex_init(&r.mutex, NULL);

    /** 启动获取结果的线程 */
    if((err = pthread_create(&get1, NULL, get_fn, (void *)&r)) != 0){
        perror("pthread create error");
    }

    if((err = pthread_create(&get2, NULL, get_fn, (void *)&r)) != 0){
        perror("pthread create error");
    }

    /** 启动计算结果的线程 */
    if((err = pthread_create(&cal, NULL, set_fcal, NULL);
    pthread_join(get1, NULL);
    pthread_join(get2, NULL);

    pthread_join(get2, NULL);
    pthread_cond_destroy(&r.cond);
    pthread_mutex_destroy(&r.mutex);

  return 0;
}