使用condition variable实现线程之间同步

发现了,这个condition variable真是个好东西啊。

 

先上代码:

#include <pthread.h>
#include <syslog.h>
#include <fcntl.h>
#include <sys/resource.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <signal.h>
#include <stdlib.h>
#include <stdio.h>
#include <termios.h>
#include <unistd.h>

struct msg {
    struct msg *m_next;
    /* ... more stuff here ... */
    int         num;
};
struct msg *workq;
pthread_cond_t qready = PTHREAD_COND_INITIALIZER;
pthread_mutex_t qlock = PTHREAD_MUTEX_INITIALIZER;

struct msg *
dequeue_msg(void)
{
    struct msg *mp;

    pthread_mutex_lock(&qlock);
    while (workq == NULL)
        pthread_cond_wait(&qready, &qlock);
    mp = workq;
    workq = mp->m_next;
    pthread_mutex_unlock(&qlock);
    /* now process the message mp */

    return mp;
}

void
enqueue_msg(struct msg *mp)
{
    pthread_mutex_lock(&qlock);
    mp->m_next = workq;
    workq = mp;
    pthread_mutex_unlock(&qlock);
    pthread_cond_signal(&qready);
}

void
cleanup(void *arg)
{
    printf("cleanup: %s/n", (char *)arg);
}

void *
thr_fn1(void *arg)
{
    int         i;
    struct msg* message;
    printf("thread 1 start, enqueue task/n");
    for (i=0; i<10; i++)
    {
        sleep(i);
        message = (struct msg*)malloc(sizeof(struct msg));
        message->num = i;
        enqueue_msg(message);
    }
    pthread_exit((void *)1);
}

void *
thr_fn2(void *arg)
{
    int         i;
    struct msg* message;
    printf("thread 2 start, dequeue task/n");
    for (;;)
    {
        message = dequeue_msg();
        printf("thread 2 get message: %d/n", message->num);

        free(message);
    }
    pthread_exit((void *)2);
}

int
main(void)
{
    int         err;
    pthread_t   tid1, tid2;
    void        *tret;

    err = pthread_create(&tid1, NULL, thr_fn1, (void *)1);
    if (err != 0)
        printf("can't create thread 1: %s/n", strerror(err));
    err = pthread_create(&tid2, NULL, thr_fn2, (void *)1);
    if (err != 0)
        printf("can't create thread 2: %s/n", strerror(err));
    err = pthread_join(tid1, &tret);
      if (err != 0)
        printf("can't join with thread 1: %s/n", strerror(err));
    printf("thread 1 exit code %d/n", (int)tret);
    err = pthread_join(tid2, &tret);
    if (err != 0)
        printf("can't join with thread 2: %s/n", strerror(err));
    printf("thread 2 exit code %d/n", (int)tret);
    exit(0);
}

 

执行的结果是
$./thread
thread 1 start, enqueue task
thread 2 start, dequeue task
thread 2 get message: 0
thread 2 get message: 1
thread 2 get message: 2
thread 2 get message: 3
thread 2 get message: 4
thread 2 get message: 5
thread 2 get message: 6
thread 2 get message: 7
thread 2 get message: 8
thread 1 exit code 1
thread 2 get message: 9

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值