linux线程池的用处

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

#define THREAD_POOL_SIZE 5

typedef struct {
    void (*function)(void *);
    void *argument;
} task_t;

typedef struct {
    task_t *tasks;
    int task_count;
    int task_size;
    pthread_mutex_t lock;
    pthread_cond_t cond;
} thread_pool_t;

thread_pool_t pool;

void *worker(void *arg) {
    while (1) {
        pthread_mutex_lock(&pool.lock);
        while (pool.task_count == 0) {
            pthread_cond_wait(&pool.cond, &pool.lock);
        }
        task_t task = pool.tasks[--pool.task_count];
        pthread_mutex_unlock(&pool.lock);

        task.function(task.argument);
    }
    return NULL;
}

void thread_pool_init() {
    pool.tasks = malloc(sizeof(task_t) * THREAD_POOL_SIZE);
    pool.task_count = 0;
    pool.task_size = THREAD_POOL_SIZE;
    pthread_mutex_init(&pool.lock, NULL);
    pthread_cond_init(&pool.cond, NULL);

    for (int i = 0; i < THREAD_POOL_SIZE; i++) {
        pthread_t thread;
        pthread_create(&thread, NULL, worker, NULL);
        pthread_detach(thread);
    }
}

void thread_pool_add_task(void (*function)(void *), void *argument) {
    pthread_mutex_lock(&pool.lock);
    if (pool.task_count == pool.task_size) {
        pool.tasks = realloc(pool.tasks, sizeof(task_t) * (pool.task_size * 2));
        pool.task_size *= 2;
    }
    pool.tasks[pool.task_count++] = (task_t){function, argument};
    pthread_cond_signal(&pool.cond);
    pthread_mutex_unlock(&pool.lock);
}

void example_task(void *arg) {
    int *num = (int *)arg;
    printf("Task %d is running\n", *num);
}

int main() {
    thread_pool_init();

    for (int i = 0; i < 10; i++) {
        int *num = malloc(sizeof(int));
        *num = i;
        thread_pool_add_task(example_task, num);
    }

    sleep(1);
    return 0;
}