www.pudn.com > WQDemo.rar > WorkQueue.h

#ifndef WORK_QUEUE_H 
#define WORK_QUEUE_H 
#include  
 
class CWorkQueue; 
 
 
/*------------------------------------------------------------------------ 
WorkItemBase	 
 
this is the basic WorkItem that the Work Queue Use its interface 
This class should be inherited and these virtual abstract functions  
implemented. 
 
  DoWork() 
 
  virtual abstract function is the function that is called when the 
  work item turn has came to be poped out of the queue and be processed. 
 
 
  Abort () 
 
  This function is called, when the Destroy function is called, for each of the WorkItems 
  That are left in the queue. 
 
   
------------------------------------------------------------------------*/ 
class WorkItemBase 
{ 
   virtual void   DoWork(void* pThreadContext)    = 0; 
   virtual void   Abort () = 0; 
   friend CWorkQueue; 
}; 
 
 
typedef std::queue           WorkItemQueue,*PWorkItemQueue; 
 
 
/*------------------------------------------------------------------------ 
CWorkQueue   
 
  这是一个工作队列类，也就是线程池，这个类主要是创建线程队列，然后等待，直到 
  有工作项目插入进来，它就唤醒相应的线程执行相应工作项目的代码，然后继续进入 
  等待状态 
------------------------------------------------------------------------*/ 
 
class  CWorkQueue  
{ 
public: 
 
   virtual ~CWorkQueue(){}; 
    
   bool Create(const unsigned int		nNumberOfThreads,  
               void*					*pThreadDataArray             = NULL); 
    
   bool InsertWorkItem(WorkItemBase* pWorkItem); 
    
   void Destroy(); 
 
private: 
    
   static unsigned long __stdcall ThreadFunc( void* pParam ); 
   WorkItemBase* RemoveWorkItem(); 
    
    
    
   enum{ 
      ABORT_EVENT_INDEX = 0, 
      SEMAPHORE_INDEX, 
      NUMBER_OF_SYNC_OBJ, 
   }; 
 
   PHANDLE					m_phThreads; 
   unsigned int				m_nNumberOfThreads; 
   void*					m_pThreadDataArray; 
    
   HANDLE					m_phSincObjectsArray[NUMBER_OF_SYNC_OBJ]; 
 
   CRITICAL_SECTION			m_CriticalSection;	 
   PWorkItemQueue           m_pWorkItemQueue; 
}; 
 
 
 
 
#endif // WORK_QUEUE_H