开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 查看源码
WorkThreadPool.cpp
资源名称：1731.rar　[点击查看]
上传用户：swkcbjrc
上传日期：2016-04-02
资源大小：45277k
文件大小：4k
源码类别：

游戏

开发平台：

Visual C++

WorkThreadPool.cpp：源码内容
  1. // WorkThreadPool.cpp: implementation of the CWorkThreadPool class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "Thread.h"
  8. #include "threadpoolmodel.h"
  9. #include "WorkThreadPool.h"
  11. #ifdef _DEBUG
  12. #undef THIS_FILE
  13. static char THIS_FILE[]=__FILE__;
  14. #define new DEBUG_NEW
  15. #endif
  17. extern CWorkThreadPool *g_pPool;
  19. //////////////////////////////////////////////////////////////////////
  20. // Construction/Destruction
  21. //////////////////////////////////////////////////////////////////////
  23. CWorkThreadPool::CWorkThreadPool() :
  24. m_exit(false,true),m_signal(false,true)
  25. {
  26. pf = NULL;
  27. m_threadNum = 3;
  29. }
  31. CWorkThreadPool::~CWorkThreadPool()
  32. {
  34. }
  36. int CWorkThreadPool::Run()
  37. {
  39. BOOL bResult=TRUE;
  40. SetCurAllThreadCount(3);
  41. SetMaxThreadCount(5);
  43. for(int i=0;i<m_threadNum;i++)
  44. {
  45. HANDLE hNewThread=NULL;
  46. DWORD dwThreadId=0;
  47. if(CreateNewThread(WORK,this,hNewThread,dwThreadId)==TRUE)
  48. {
  49. TRACE("Work ThreadId=%dn",dwThreadId);
  50. }
  51. else
  52. bResult=FALSE;
  53. //线程池满了
  54. if(TPGetLastError()==ERROR_THREADPOOL_FULL)
  55. return (bResult);
  56. }
  58. return 0;
  59. }
  60. //添加任务
  61. void CWorkThreadPool::RegisterWorkItem(LPVOID item)
  62. {
  63. m_workItems.Add(item);
  64. m_signal.SetEvent();
  65. }
  67. LPVOID CWorkThreadPool::GetNextItem()
  68. {
  69. LPVOID ret = NULL;
  70. m_lock.Lock();
  71. if(m_workItems.GetSize())
  72. ret = m_workItems[0];
  73. m_lock.Unlock();
  74. return ret;
  75. }
  77. void CWorkThreadPool::SetPF(lpCallBack1 *f)
  78. {
  79. pf = f;
  80. }
  82. UINT CWorkThreadPool::Work()
  83. {
  84. //如果全局线程池指针为空，则直接退出本线程
  85.     if(g_pPool==NULL)
  86.     {  
  87. //同时从当前存在线程数中减少一个，设置ERROR_GLOBPOOL_NULL
  88. //全局线程错误
  89.         g_pPool->DecrementCurAllThreadCount( );
  90.         return ERROR_GLOBPOOL_NULL;
  91.     }
  92.     TRACE("Work  Thread Startn");
  93.     //取得当前线程ID
  94.     DWORD dwThreadId=GetCurrentThreadId( );
  95.     DWORD dwIndex=0;
  96.     int nTemp=2;
  98.     //声明一个事件数组用于等待事件
  99.     LPHANDLE pEvent=new HANDLE[nTemp];
  100.     pEvent[0]=g_pPool->GetWorkEvent(); //获取工作事件标志
  101. pEvent[1]=m_exit;
  102.     while(TRUE)
  103.     {
  104.         //等待多事件响应，等待工作事件，死亡事件，全局死亡事件
  105.         //如果有一个事件被置为信号态，则继续运行，否则保持等待
  106.         dwIndex=WaitForMultipleObjects(2,pEvent,FALSE,INFINITE);
  107.         //如果等待返回结果为错误，继续运行
  108.         if(dwIndex==WSA_WAIT_FAILED)
  109.         {
  110.             Sleep(300);
  111.             continue;
  112.         }
  113.         g_pPool->IncrementCurWorkThreadCount( );
  114.         DWORD dwTemp=0;
  115.         dwTemp=dwIndex-WSA_WAIT_EVENT_0;
  116.         switch(dwTemp)
  117.         {
  118. //dwTemp返回值为1表示线程要结束
  119.         case 1://均为死亡
  120.             g_pPool->DecrementCurWorkThreadCount( );
  121.             goto CLEAR;
  122. break;
  123. case 0:
  124. break;
  125.         }
  126.         if(dwTemp!=0)
  127.         {
  128.             g_pPool->DecrementCurWorkThreadCount( );
  129.             continue;
  130.         }
  131.         //线程开始工作
  132.         //取得工作内容
  133. LPVOID pItem = GetNextItem();
  134. if(pItem)
  135. {
  136. RemoveWorkItem(pItem);
  137. pf(pItem);
  138. }else
  139. {
  140. m_signal.ResetEvent();
  141. }
  142.     }
  143. CLEAR:
  144.     if(pEvent!=NULL)
  145.     {
  146.         delete []pEvent;
  147.         pEvent=NULL;
  148.     }
  149.     g_pPool->DecrementCurAllThreadCount( );
  150.     return SUCCESS_THREAD_RETURN;
  154. }
  156. UINT CWorkThreadPool::WORK(LPVOID lp)
  157. {
  158. CWorkThreadPool* pWork = (CWorkThreadPool*)lp;
  159. ASSERT(pWork);
  160. pWork->Work();
  161. return 0;
  162. }
  164. void CWorkThreadPool::Stop()
  165. {
  166. m_signal.ResetEvent();
  167. m_exit.SetEvent();
  169. }
  170. //删除工作队列
  171. void CWorkThreadPool::RemoveWorkItem(LPVOID _pItem)
  172. {
  173. LPVOID pItem = NULL;
  174. m_lock.Lock();
  175. for(int i=0;i<m_workItems.GetSize();i++)
  176. {
  177. pItem = (LPVOID)m_workItems[i];
  178. if(_pItem == pItem)
  179. {
  180. m_workItems.RemoveAt(i);
  181. break;
  182. }
  183. }
  184. m_lock.Unlock();
  185. }
  187. void CWorkThreadPool::SetThreadNum(int num)
  188. {
  189. Stop();
  190. ASSERT(num > 0);
  191. m_threadNum = num;
  192. }