Linux 线程池队列


#include<stdio.h>
#include<unistd.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/wait.h>
#include <pthread.h>
#include <string.h>
#define MAX_NUM 30000200   // 最大素数范围
#define QUEUE_CAPACITY 20   // 队列容量
#define THREAD_NUM 10   // 线程数
#define TASK_NUM 200   // 任务数
#define PRIME_NUM 200   // 待计算的素数个数

// 定义任务队列
struct queue {
    int finished;
    int head;
    int tail;
    int data[QUEUE_CAPACITY];
    int count;
    pthread_mutex_t mutex;
    pthread_cond_t cond;
};

// 定义线性表，用于存储待计算的素数
int prime[PRIME_NUM] = {0};
int prime_cnt = 0;

// 初始化任务队列
void queue_init(struct queue* q) {
    q->finished = 0;
    q->head = 0;
    q->tail = 0;
    q->count = 0;
    pthread_mutex_init(&q->mutex, NULL);
    pthread_cond_init(&q->cond, NULL);
}

// 向任务队列中添加数字
void queue_push(struct queue* q, int num) {
    pthread_mutex_lock(&q->mutex);
    while(q->count >= QUEUE_CAPACITY && q->finished == 0) {
        // 队列已满，等待队列中的数字被取出
        pthread_cond_wait(&q->cond, &q->mutex);
    }

    if(q->finished == 1) {
        pthread_mutex_unlock(&q->mutex);
        return;
    }

    q->data[q->tail] = num;
    q->tail = (q->tail + 1) % QUEUE_CAPACITY;
    q->count++;

    pthread_cond_signal(&q->cond);
    pthread_mutex_unlock(&q->mutex);
}

// 从任务队列中获取数字
int queue_pop(struct queue* q) {
    pthread_mutex_lock(&q->mutex);
    while(q->count == 0 && q->finished == 0) {
        // 队列为空，等待数字到来
        pthread_cond_wait(&q->cond, &q->mutex);
    }

    if(q->count == 0 && q->finished == 1) {
        pthread_mutex_unlock(&q->mutex);
        return -1;
    }

    int num = q->data[q->head];
    q->head = (q->head + 1) % QUEUE_CAPACITY;
    q->count--;

    pthread_cond_signal(&q->cond);
    pthread_mutex_unlock(&q->mutex);

    return num;
}

// 线程执行的任务
void thread_task(struct queue* q) {
    int num;
    while(1) {
        num = queue_pop(q);
        if(num == -1) {
            // 队列为空并且finished为1，退出线程
            break;
        }

        // 判断数字是否为素数，如果是，输出线程号和数字
        int is_prime = 1;
        for(int i = 2; i < num; i++) {
            if(num % i == 0) {
                is_prime = 0;
                break;
            }
        }
        if(is_prime) {
            printf("Thread %ld: %d is a prime\n", pthread_self(), num);
            // 将素数添加到线性表中
            prime[prime_cnt++] = num;
        }
    }
}

int main() {
    pthread_t threads[THREAD_NUM];
    struct queue queue;
    queue_init(&queue);

    // 往任务队列中添加数字
    for(int i = 0; i < TASK_NUM; i++) {
        for(int j = 1; j <= 200; j++) {
            queue_push(&queue, 30000000 + i * 200 + j);
        }
    }
    queue.finished = 1;

    // 创建线程池
    for(int i = 0; i < THREAD_NUM; i++) {
        pthread_create(&threads[i], NULL, (void*)thread_task, NULL);
    }

    // 等待子线程执行完毕
    for(int i = 0; i < THREAD_NUM; i++) {
        pthread_join(threads[i], NULL);
    }

    // 输出所有素数
    printf("All primes:\n");
    for(int i = 0; i < prime_cnt; i++) {
        printf("%d ", prime[i]);
    }
    printf("\n");

    return 0;
}

求帮忙看下为什么输出不了结果


pthread_create(&threads[i], NULL, (void*)thread_task, &queue);

不知道你这个问题是否已经解决, 如果还没有解决的话:

这有个类似的问题, 你可以参考下: https://ask.csdn.net/questions/647148
我还给你找了一篇非常好的博客，你可以看看是否有帮助，链接：Linux中利用线程实现多个客户端和服务器端进行通信
除此之外, 这篇博客: Linux下生产者与消费者实现的问题详解中的 线程属性 部分也许能够解决你的问题, 你可以仔细阅读以下内容或者直接跳转源博客中阅读:
```
int pthread_attr_init(pthread_attr_t **attr);
int pthread_attr_destory(pthread_attr_t *attr);
```
线程属性对象由pthread_attr_init（）接口初始化，由pthread_attr_destory（）来销毁。
您还可以看一下夏曹俊老师的Linux程序设计从入门到实战课程中的函数是线程的载体小节, 巩固相关知识点

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^

Linux 线程池 队列

Linux 线程池队列