【线程】屏障

一、主要函数应用

1. 函数原型：初始化和回收

int pthread_barrier_init(pthread *barrier, const pthread_barrier_t *attr, unsigned int count);
int pthread_barrier_destroy(pthread_barrier_t *barrier);

分析：

• 初始化屏障时，可以使用count参数指定，在允许所有线程运行之前，必须到达屏障的线程数目。 

 

2. 函数原型

#include <pthread.h>
int pthread_barrier_wait(pthread_barrier_t *barrier);

分析：

• 每一个线程调用等待函数，barrier 计数都会加 1，直到计数值等于初始化函数中的参数 count。当最后一个线程让 barrier 计数值达到了 count，所有等待的线程都会被唤醒。
• 所有线程中，只有一个线程会让 pthread_barrier_wait 返回 PTHREAD_BARRIER_SERIAL_THREAD，而其它线程调用此函数会返回 0.

 

1. 测试代码

【题目】程序 barrier 一共有 11 个线程，包含主线程。所有线程会先打印一遍 “hello, I’m thread xx” 然后调用 pthread_barrier_wait 等待，直到 barrier 计数达到了 11 才返回。一旦返回了，就会打印 thread xx returning xx.

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

pthread_barrier_t b;

void* th_fn(void *arg) 
{
    int id = (int)arg;
    int res = 0;
    printf("hello, I'm thread [%d]\n", id);
    res = pthread_barrier_wait(&b);
    printf("thread [%d] returning [%d]\n", id, res);
    return NULL;
}

int main() 
{
    int i, res = 0;
    pthread_t tid[10];
    printf("PTHREAD_BARRIER_SERIAL_THREAD = %d\n", PTHREAD_BARRIER_SERIAL_THREAD);

    pthread_barrier_init(&b, NULL, 11);

    for (i = 0; i < 10; ++i) {
        pthread_create(&tid[i], NULL, th_fn, (void*)i);
    }

    res = pthread_barrier_wait(&b);
    printf("thread [main] returning [%d]\n", res);

    for (i = 0; i < 10; ++i) {
        pthread_join(tid[i], NULL);
    }

    pthread_barrier_destroy(&b);
    return 0;
}

输出结果：

 

参考资料

1. 88-屏障 barrier