c++实现跨平台的线程池

学无止境

1. #include <iostream>
   
2. #include <queue>
   
3. #include <thread>
   
4. #include <mutex>
   
5. #include <condition_variable>
   
6. 
   
7. class ThreadPool {
   
8. public:
   
9.     ThreadPool(size_t numThreads) : stop(false) {
   
10.         for (size_t i = 0; i < numThreads; ++i) {
    
11.             workers.emplace_back([this] {
    
12.                 while (true) {
    
13.                     std::function<void()> task;
    
14.                     {
    
15.                         std::unique_lock<std::mutex> lock(queueMutex);
    
16.                         condition.wait(lock, [this] { return stop || !tasks.empty(); });
    
17.                         if (stop && tasks.empty()) {
    
18.                             return;
    
19.                         }
    
20.                         task = std::move(tasks.front());
    
21.                         tasks.pop();
    
22.                     }
    
23.                     task();
    
24.                 }
    
25.             });
    
26.         }
    
27.     }
    
28. 
    
29.     template <class F, class... Args>
    
30.     void enqueue(F&& f, Args&&... args) {
    
31.         {
    
32.             std::unique_lock<std::mutex> lock(queueMutex);
    
33.             tasks.emplace([=] { std::forward<F>(f)(std::forward<Args>(args)...); });
    
34.         }
    
35.         condition.notify_one();
    
36.     }
    
37. 
    
38.     ~ThreadPool() {
    
39.         {
    
40.             std::unique_lock<std::mutex> lock(queueMutex);
    
41.             stop = true;
    
42.         }
    
43.         condition.notify_all();
    
44.         for (std::thread& worker : workers) {
    
45.             worker.join();
    
46.         }
    
47.     }
    
48. 
    
49. private:
    
50.     std::vector<std::thread> workers;
    
51.     std::queue<std::function<void()>> tasks;
    
52.     std::mutex queueMutex;
    
53.     std::condition_variable condition;
    
54.     bool stop;
    
55. };
    
56. 
    
57. void printNumber(int number) {
    
58.     std::cout << "Number: " << number << std::endl;
    
59. }
    
60. 
    
61. int main() {
    
62.     ThreadPool pool(4);
    
63. 
    
64.     for (int i = 0; i < 10; ++i) {
    
65.         pool.enqueue(printNumber, i);
    
66.     }
    
67. 
    
68.     return 0;
    
69. }

复制代码