VC++实现线程池的示例代码

源码资源网

VC++实现线程池的示例代码
使用 VC++ 实现线程池的示例代码。这个线程池可以用于管理多个线程以执行任务，适用于需要并发处理的场景。

1. ### 线程池实现示例
   
2. #include <iostream>
   
3. #include <thread>
   
4. #include <vector>
   
5. #include <queue>
   
6. #include <condition_variable>
   
7. #include <mutex>
   
8. #include <functional>
   
9. class ThreadPool {
   
10. public:
    
11.     ThreadPool(size_t numThreads);
    
12.     ~ThreadPool();
    
13.      // 添加任务到线程池
    
14.     void enqueue(std::function<void()> task);
    
15. private:
    
16.     // 工作线程函数
    
17.     void worker();
    
18.      std::vector<std::thread> workers; // 工作线程
    
19.     std::queue<std::function<void()>> tasks; // 任务队列
    
20.     std::mutex queueMutex; // 互斥锁
    
21.     std::condition_variable condition; // 条件变量
    
22.     bool stop; // 停止标志
    
23. };
    
24. // 线程池构造函数
    
25. ThreadPool::ThreadPool(size_t numThreads) : stop(false) {
    
26.     for (size_t i = 0; i < numThreads; ++i) {
    
27.         workers.emplace_back([this] { worker(); });
    
28.     }
    
29. }
    
30. // 线程池析构函数
    
31. ThreadPool::~ThreadPool() {
    
32.     {
    
33.         std::unique_lock<std::mutex> lock(queueMutex);
    
34.         stop = true;
    
35.     }
    
36.     condition.notify_all(); // 唤醒所有线程
    
37.     for (std::thread &worker : workers) {
    
38.         worker.join(); // 等待所有线程结束
    
39.     }
    
40. }
    
41. // 向任务队列添加任务
    
42. void ThreadPool::enqueue(std::function<void()> task) {
    
43.     {
    
44.         std::unique_lock<std::mutex> lock(queueMutex);
    
45.         tasks.emplace(task); // 添加任务
    
46.     }
    
47.     condition.notify_one(); // 唤醒一个线程
    
48. }
    
49. // 工作线程函数
    
50. void ThreadPool::worker() {
    
51.     while (true) {
    
52.         std::function<void()> task;
    
53.          {
    
54.             std::unique_lock<std::mutex> lock(queueMutex);
    
55.             condition.wait(lock, [this] { return stop || !tasks.empty(); }); // 等待任务
    
56.              if (stop && tasks.empty()) {
    
57.                 return; // 如果停止且没有任务，退出线程
    
58.             }
    
59.              task = std::move(tasks.front()); // 获取任务
    
60.             tasks.pop(); // 移除任务
    
61.         }
    
62.          task(); // 执行任务
    
63.     }
    
64. }
    
65. // 示例任务
    
66. void exampleTask(int id) {
    
67.     std::cout << "Task " << id << " is running on thread " << std::this_thread::get_id() << std::endl;
    
68. }
    
69. int main() {
    
70.     ThreadPool pool(4); // 创建一个线程池，包含4个线程
    
71.      // 添加任务到线程池
    
72.     for (int i = 0; i < 10; ++i) {
    
73.         pool.enqueue([i] { exampleTask(i); });
    
74.     }
    
75.      // 稍等一下以确保所有任务完成
    
76.     std::this_thread::sleep_for(std::chrono::seconds(2));
    
77.      return 0;
    
78. }

复制代码

### 代码说明
1. **ThreadPool 类**：
   -  `ThreadPool`  构造函数：创建指定数量的工作线程。
   -  `~ThreadPool`  析构函数：在对象销毁时停止所有线程并等待它们完成。
   -  `enqueue`  方法：添加任务到任务队列。
   -  `worker`  方法：工作线程从任务队列中获取任务并执行。
2. **示例任务**：
   -  `exampleTask`  函数：用于演示任务执行，输出任务ID和线程ID。
3. **主函数**：
   - 创建一个包含4个线程的线程池。
   - 向线程池添加10个任务，并执行这些任务。
### 注意事项
- 线程池可以有效管理多个线程，避免频繁创建和销毁线程带来的开销。
- 在实际应用中，可能需要对任务的执行结果进行处理，或者添加更多的功能，如任务优先级等。
以上是一个简单的线程池实现示例，你可以根据需要进行扩展和优化。