工作线程基类TaskSvc

工作线程基类TaskSvc

　　前端时间用ACE写代码，发ACE_Task确实好用。不但能提供数量一定的线程，还能够让这些继承的线程函数自由访问子类的private和protected变量。此外，ACE_Task还提供了ACE_Message_Block的消息块及消息队列，能够自主定义消息，实现数据的异步处理，甚至能在运算有压力的情况下丢弃部分数据。优点虽然多多，但缺点也是有的。为了使用ACE_Task，必须得把ACE库加载起来吧。很多时候，只想用ACE_Task，每次都加载ACE库似乎就变得很讨厌了。

　　由此，觉得自己写一个类似ACE_Task的基类。由于线程用的比较多，这次就先写一个线程基类CTaskSvc。上代码

　　TaskSvc.h

 #pragma once

 #ifndef WHG_CTASKSVC
 #define WHG_CTASKSVC

 class CTaskSvc
 {
 public:
     //Activate用于激活一定数量的工作者线程，默认激活数量为1。返回当前线程队列大小
     UINT    Activate(int num = );
     //获取线程队列大小
     UINT    GetThreadsNum(void);

 protected:
     //只有子类才可以构造父类，拒绝外部访问构造类实例
     CTaskSvc(void);
     ~CTaskSvc(void);
     //子类应重定义工作线程细节
     virtual void svc();
     //Close用于等待线程结束并关闭线程，退出线程由子类控制
     void    Close();

 private:
     //工作者线程访问接口
     static UINT WorkThread(LPVOID param);
     //线程队列
     std::vector<CWinThread*>    vec_threads;
 };

 #endif

　　TaskSvc.cpp

 #include "StdAfx.h"
 #include "TaskSvc.h"

 CTaskSvc::CTaskSvc(void)
 {
 }

 CTaskSvc::~CTaskSvc(void)
 {
     Close();
 }

 UINT CTaskSvc::Activate(int num)
 {
     for (int i=;i<num;i++)
     {
         CWinThread* pwt = AfxBeginThread(WorkThread,this,THREAD_PRIORITY_NORMAL,,CREATE_SUSPENDED);
         if (pwt)
         {
             pwt->m_bAutoDelete = false;
             pwt->ResumeThread();
             vec_threads.push_back(pwt);
         }
     }
     return vec_threads.size();
 }

 UINT CTaskSvc::GetThreadsNum(void)
 {
     return vec_threads.size();
 }

 UINT CTaskSvc::WorkThread(LPVOID param)
 {
     CTaskSvc* pts = (CTaskSvc*)param;
     if (pts)
     {
         pts->svc();
     }
     return ;
 }

 void CTaskSvc::svc()
 {
 }

 void CTaskSvc::Close()
 {
     int cnt = vec_threads.size();
     if (cnt > )
     {
         std::vector<CWinThread*>::iterator iter = vec_threads.begin();
         for (; iter != vec_threads.end(); iter++)
         {
             CWinThread* pwt = *iter;
             WaitForSingleObject(pwt->m_hThread,INFINITE);
             delete pwt;
         }
         vec_threads.clear();
     }
 }

　　调用方法：

1：继承CTaskSvc类

2：添加Svc虚函数

3：在svc内添加线程工作代码

4：释放线程调用Close，需要先退出工作线程。