关闭

使用QueueUserWorkerItem实现的线程池封装

标签: window线程池
179人阅读 评论(0) 收藏 举报
分类:

此线程池所依赖的线程类,请参看《一个Windows C++的线程类实现》:

http://blog.csdn.net/huyiyang2010/archive/2010/08/10/5801597.aspx

 

SystemThreadPool.h

 

  1. #define __SYSTEM_THREAD_POOL__  
  2. #include "Thread.h"  
  3. #include <list>  
  4. #include <windows.h>  
  5. class CThreadPoolExecutor  
  6. {  
  7. public:  
  8.     CThreadPoolExecutor(void);  
  9.     ~CThreadPoolExecutor(void);  
  10.     /** 
  11.       初始化线程池,创建minThreads个线程 
  12.     **/  
  13.     bool Init(unsigned int maxTaskse);  
  14.     /** 
  15.       执行任务,若当前任务列表没有满,将此任务插入到任务列表,返回true 
  16.       否则返回false 
  17.     **/  
  18.     bool Execute(Runnable * pRunnable);  
  19.     /** 
  20.       终止线程池,先制止塞入任务, 
  21.       然后等待直到任务列表为空, 
  22.       然后设置最小线程数量为0, 
  23.       等待直到线程数量为空, 
  24.       清空垃圾堆中的任务 
  25.     **/  
  26.     void Terminate();  
  27.     /** 
  28.       返回线程池中当前的线程数量 
  29.     **/  
  30.     unsigned int GetThreadPoolSize();  
  31. private:  
  32.     static unsigned int WINAPI StaticThreadFunc(void * arg);  
  33. private:  
  34.     typedef std::list<Runnable *> Tasks;  
  35.     typedef Tasks::iterator TasksItr;  
  36.     Tasks m_Tasks;  
  37.     CRITICAL_SECTION m_csTasksLock;  
  38.     volatile bool m_bRun;  
  39.     volatile bool m_bEnableInsertTask;  
  40.     volatile unsigned int m_maxTasks;  
  41. };  
  42. #endif  

 

 

SytemThreadPool.cpp

 

  1. #include "SystemThreadPool.h"  
  2. CThreadPoolExecutor::CThreadPoolExecutor(void) :   
  3. m_bRun(false),  
  4. m_bEnableInsertTask(false)  
  5. {  
  6.     InitializeCriticalSection(&m_csTasksLock);  
  7. }  
  8. CThreadPoolExecutor::~CThreadPoolExecutor(void)  
  9. {  
  10.     Terminate();  
  11.     DeleteCriticalSection(&m_csTasksLock);  
  12. }  
  13. bool CThreadPoolExecutor::Init(unsigned int maxTasks)  
  14. {  
  15.     if(maxTasks == 0)  
  16.     {  
  17.         return false;  
  18.     }  
  19.     m_maxTasks = maxTasks;  
  20.     m_bRun = true;  
  21.     m_bEnableInsertTask = true;  
  22.     return true;  
  23. }  
  24. bool CThreadPoolExecutor::Execute(Runnable * pRunnable)  
  25. {  
  26.     if(!m_bEnableInsertTask)  
  27.     {  
  28.         return false;  
  29.     }  
  30.     if(NULL == pRunnable)  
  31.     {  
  32.         return false;  
  33.     }  
  34.     EnterCriticalSection(&m_csTasksLock);  
  35.     if(m_Tasks.size() >= m_maxTasks)  
  36.     {  
  37.         LeaveCriticalSection(&m_csTasksLock);  
  38.         return false;  
  39.     }  
  40.     m_Tasks.push_back(pRunnable);  
  41.     LeaveCriticalSection(&m_csTasksLock);  
  42.     bool ret = QueueUserWorkItem((LPTHREAD_START_ROUTINE)StaticThreadFunc, this, WT_EXECUTEINPERSISTENTIOTHREAD);  
  43.     if(!ret)  
  44.     {  
  45.         EnterCriticalSection(&m_csTasksLock);  
  46.         m_Tasks.remove(pRunnable);  
  47.         LeaveCriticalSection(&m_csTasksLock);  
  48.     }  
  49.     return ret;  
  50. }  
  51. unsigned int CThreadPoolExecutor::GetThreadPoolSize()  
  52. {  
  53.     return m_Tasks.size();  
  54. }  
  55. void CThreadPoolExecutor::Terminate()  
  56. {  
  57.     m_bEnableInsertTask = false;  
  58.     m_bRun = false;  
  59.     while(m_Tasks.size() != 0)  
  60.     {  
  61.         Sleep(1);  
  62.     }  
  63. }  
  64. unsigned int WINAPI CThreadPoolExecutor::StaticThreadFunc(void * arg)  
  65. {  
  66.     CThreadPoolExecutor * pThreadPool = (CThreadPoolExecutor *)arg;  
  67.     Runnable * pRunnable = NULL;  
  68.     EnterCriticalSection(&pThreadPool->m_csTasksLock);  
  69.     pRunnable = pThreadPool->m_Tasks.front();  
  70.     if(NULL != pRunnable)  
  71.     {  
  72.         pThreadPool->m_Tasks.pop_front();  
  73.     }  
  74.     LeaveCriticalSection(&pThreadPool->m_csTasksLock);  
  75.     if(NULL != pRunnable)  
  76.     {  
  77.         pRunnable->Run();  
  78.     }  
  79.     return 0;  
  80. }  

 

 

用法:

 

#include "Thread.h"
#include "SystemThreadPool.h"

class R : public Runnable
{
public:
    ~R()
    {
    }
    void Run()
    {
        printf("Hello World/n");
    }
};

int _tmain(int argc, _TCHAR* argv[])
{
    CThreadPoolExecutor * pExecutor = new CThreadPoolExecutor();
    pExecutor->Init(50);
    R r;
    for(int i=0;i<100;i++)
    {
        while(!pExecutor->Execute(&r))
        {
        }
    }
    pExecutor->Terminate();
    delete pExecutor;
    getchar();
    return 0;
}

 

测试结果:

 

机器:

Intel(R) Core(TM)2 Duo CPU

E8400 @ 3.00GHz

2G内存

对于100个任务并且每个任务包含10000000个循环,任务中无等待:

线程池耗时:2203时间片

 

 

from:http://blog.csdn.net/huyiyang2010/article/details/5820548

1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:289817次
    • 积分:4273
    • 等级:
    • 排名:第7089名
    • 原创:56篇
    • 转载:547篇
    • 译文:0篇
    • 评论:18条
    最新评论