用boost自制线程池

元始天尊

初学boost，看到asio库如此好用，就做了个简单的线程池

1. // test.cpp : 定义控制台应用程序的入口点。
   
2. //
   
3. 
   
4. 
   
5. #include <iostream>
   
6. #include <boost/asio.hpp>
   
7. #include <boost/bind.hpp>
   
8. #include <boost/thread.hpp>
   
9. #include <boost/thread/detail/thread_group.hpp>
   
10. #include <boost/scoped_ptr.hpp>
    
11. #include <deque>
    
12. using namespace std;
    
13. 
    
14. std::deque<int> task;
    
15. 
    
16. boost::mutex output;//执行函数用的资源锁
    
17. 
    
18. 
    
19. //任务函数
    
20. void func(int task)//taskindex任务数据索引，用于寻找指定的任务
    
21. {
    
22.         boost::mutex::scoped_lock mio(output);
    
23.         cout << task << "th task run" << endl;
    
24. #include <stdlib.h>
    
25.         Sleep((rand() % 10) * 100);//增大随机性，看到交叉执行的情况
    
26.         cout << task << "th task end" << endl;
    
27. }
    
28. 
    
29. template<typename TASKTYPE>
    
30. class ThreadPool
    
31. {
    
32. public:
    
33.         //用于将task里的任务分配各给threadnum个活动线程，分配时要进行线程同步
    
34.         void wrapper(boost::asio::io_service* service)
    
35.         {
    
36.                 while (true)
    
37.                 {
    
38.                         mu.lock();
    
39.                         if (task.empty())
    
40.                         {
    
41.                                 mu.unlock();
    
42.                                 return;
    
43.                         }
    
44.                         service->post(boost::bind(func, task.front()));
    
45.                         task.pop_front();
    
46.                         mu.unlock();
    
47.                         service->run();
    
48.                         service->reset();
    
49.                 }
    
50.         }
    
51. 
    
52.         ThreadPool(int _threadnum, std::deque<TASKTYPE>& _task) :threadnum(_threadnum), task(_task)
    
53.         {
    
54.                 ppio = new boost::asio::io_service*[threadnum];
    
55.                 if (!ppio)
    
56.                         throw "Error";
    
57.                 for (int i = 0; i < threadnum; i++)
    
58.                 {
    
59.                         ppio[i] = new boost::asio::io_service;
    
60.                         if (!ppio[i])
    
61.                                 throw "Error";
    
62.                 }
    
63.         }
    
64. 
    
65.         bool Exec()
    
66.         {
    
67.                 try
    
68.                 {
    
69.                         //建立线程组
    
70.                         boost::thread_group tg;
    
71.                         for (int i = 0; i < threadnum; i++)
    
72.                         {
    
73.                                 tg.create_thread(boost::bind(&ThreadPool::wrapper, this, ppio[i]));
    
74.                         }
    
75.                         //等待所有任务执行完毕
    
76.                         tg.join_all();
    
77.                 }
    
78.                 catch (...)
    
79.                 {
    
80.                         //确保释放new
    
81.                         return false;
    
82.                 }
    
83.                 return true;
    
84.         }
    
85. 
    
86.         virtual ~ThreadPool()
    
87.         {
    
88.                 for (int i = 0; i < threadnum; i++)
    
89.                         delete ppio[i];
    
90.                 delete[]ppio;
    
91.         }
    
92. private:
    
93.         int threadnum;
    
94.         boost::mutex mu;//分配任务用锁
    
95.         std::deque<TASKTYPE>& task;//要分配的总任务
    
96.         boost::asio::io_service** ppio;//耗费的线程资源      PS：如果Io_service不是noncopyable，我也不用这么费劲用指针做，连vector都不让用
    
97. };
    
98. 
    
99. void main()
    
100. {
     
101. //用法示例：
     
102.         int tasknum = 1000;
     
103.         for (int i = 0; i < tasknum; i++)
     
104.         {
     
105.                 task.push_back(i);//新增1000个任务
     
106.         }
     
107.         //使用3线程完成任务
     
108.         ThreadPool<int> newtaskset(3, task);
     
109.         newtaskset.Exec();
     
110.         cin >> tasknum;//等待运行
     
111. }
     

复制代码

有了这个就可以做我的dns多线程查询工具了！！！