threadpool.h

#pragma once

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

#define COUNT 2  // 一次增加或减少的线程个数

// 任务结构体
typedef struct Task {
    void (*process)(void *arg);  // 处理函数
    void *arg;                   // 处理函数的参数
} Task;

// 线程池结构体
typedef struct ThreadPool {
    // 任务队列
    Task *qTasks;   // 底层数组
    int qCapacity;  // 数组容量
    // 方便队列出入元素
    int qFront;      // 队首
    int qRear;       // 队尾
    int qTaskCount;  // 当前任务数

    pthread_t *workerThreadIDs;  // 工作线程数组
    pthread_t managerThreadID;   // 管理者线程
    int minThreads;              // 最小线程数
    int maxThreads;              // 最大线程数
    int busyCount;               // 忙碌线程计数
    int liveCount;               // 存活线程计数
    int exitCount;               // 退出线程计数
    int shutdown;                // 关闭标志

    pthread_mutex_t poolMutex;  // 保护线程池的互斥锁
    pthread_mutex_t busyMutex;  // 保护busyCount的互斥锁
    pthread_cond_t notEmpty;    // 队列非空条件
    pthread_cond_t notFull;     // 队列未满条件
} ThreadPool;

// 创建线程池
ThreadPool *threadpoolCreate(int min, int max, int queueCapacity);

// 销毁线程池
int threadpoolDestroy(ThreadPool *pool);

// 添加任务到线程池
void threadpoolAddTask(ThreadPool *pool, void (*process)(void *), void *arg);

// 管理者线程函数
void *manager(void *arg);

// 工作线程函数
void *worker(void *arg);

// 单个线程退出
void threadExit(ThreadPool *pool);

// 获取线程池状态
int threadpoolGetBusyCount(ThreadPool *pool);
int threadpoolGetAliveCount(ThreadPool *pool);