工作线程基类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 = 1);
    //获取线程队列大小
    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=0;i<num;i++)
    {
        CWinThread* pwt = AfxBeginThread(WorkThread,this,THREAD_PRIORITY_NORMAL,0,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 0;
}

void CTaskSvc::svc()
{
}

void CTaskSvc::Close()
{
    int cnt = vec_threads.size();
    if (cnt > 0)
    {
        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，需要先退出工作线程。