C++如何实现一个线程池_C++高性能并发编程中的线程池设计与实现_技术教程

线程池通过复用线程减少开销，核心组件包括工作线程、任务队列、同步机制和控制状态；使用线程安全队列管理任务，配合互斥锁与条件变量实现等待唤醒；每个线程循环从队列取任务执行，支持优雅关闭；通过enqueue提交任务，可结合future获取结果，优化方向包括无锁队列、优先级调度和动态调整线程数。

在C++高性能并发编程中，线程池的核心目标是复用线程、减少频繁创建销毁的开销，并有效管理任务调度。实现一个高效的线程池需要解决任务队列、线程同步、负载均衡和资源释放等问题。下面是一个简洁、实用且线程安全的线程池设计与实现方案。

线程池的基本结构

一个典型的线程池包含以下几个核心组件：

工作线程集合：固定数量的后台线程，持续从任务队列中取任务执行。
任务队列：存放待执行的任务（通常为函数对象），使用线程安全的队列。
同步机制：使用互斥锁（std::mutex）和条件变量（std::condition_variable）协调线程间的等待与唤醒。
控制状态：标记线程池是否正在运行，用于优雅关闭。

任务队列与线程安全封装

任务队列需要支持多线程下的入队和出队操作。可以封装一个线程安全的队列：

template
class ThreadSafeQueue {
private:
    std::queue queue_;
    mutable std::mutex mtx_;
    std::condition_variable cv_;

public:
    void push(T task) {
        std::lock_guard lock(mtx_);
        queue_.push(std::move(task));
        cv_.notify_one();
    }

    bool try_pop(T& task) {
        std::lock_guard lock(mtx_);
        if (queue_.empty()) return false;
        task = std::move(queue_.front());
        queue_.pop();
        return true;
    }

    void wait_and_pop(T& task) {
        std::unique_lock lock(mtx_);
        cv_.wait(lock, [this] { return !queue_.empty(); });
        task = std::move(queue_.front());
        queue_.pop();
    }
};

线程池的实现

基于上述队列，构建线程池类：

class ThreadPool {
private:
    std::vector workers_;
    ThreadSafeQueue> task_queue_;
    std::atomic stop_{false};

    void worker_thread() {
        while (!stop_) {
            std::function task;
            task_queue_.wait_and_pop(task);
            if (task) task();
        }
    }

public:
    explicit ThreadPool(size_t num_threads) {
        for (size_t i = 0; i < num_threads; ++i) {
            workers_.emplace_back(&ThreadPool::worker_thread, this);
        }
    }

    ~ThreadPool() {
        stop_ = true;
        // 唤醒所有等待线程
        // 可以通过向队列中放入空任务来触发退出
        for (auto& t : workers_) {
            if (t.joinable()) t.join();
        }
    }

    template
    void enqueue(F&& f) {
        task_queue_.push(std::forward(f));
    }
};

这个实现中，每个工作线程在worker_thread中循环等待任务。通过wait_and_pop阻塞等待新任务，一旦收到任务立即执行。

使用示例与性能建议

使用线程池提交任务非常简单：

int main() {
    ThreadPool pool(4); // 创建4个线程

    for (int i = 0; i < 8; ++i) {
        pool.enqueue([i]() {
            std::cout << "Task " << i << " running on thread "
                      << std::this_thread::get_id() << '\n';
            std::this_thread::sleep_for(std::chrono::milliseconds(100));
        });
    }

    // 主线程等待任务完成（实际项目中可加屏障或future）
    std::this_thread::sleep_for(std::chrono::seconds(2));
    return 0;
}

为了进一步提升性能，可以考虑以下优化：