网络服务器软件开发/中间件开发，关注ACE/ICE/boost
C++博客 首页 新随笔 联系 聚合 管理
  152 Posts :: 3 Stories :: 172 Comments :: 0 Trackbacks
自己造的一个线程类
#ifndef THREAD_H
#define THREAD_H

#include <stdio.h>//printf

#ifdef WIN32
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#define THREAD_API         DWORD WINAPI
#else
#include <pthread.h>
#define THREAD_API         void*          
#endif

class CThread
{
private:

#ifdef WIN32
    HANDLE m_hThreadHandle;
    DWORD  m_dwThreadId;
#else
    pthread_t  m_dwThreadId;
#endif

    bool      m_bStopFlag;//是否已经“取消”该线程
    bool      m_bStarted;//线程已经启动

private:
    struct CThreadArg 
    {
        CThread* m_pThread;//保存this指针
        void*    m_pArg;//保存start成员函数里面的arg参数指针
    };

    CThreadArg* m_pThrArg;
public:
    CThread();    
    ~CThread();
    int Start(void* arg = NULL);
    int  Wait();//成功返回0；失败-1

    void Terminate(void);//强制退出

    void  SetStopFlag();    //协作性退出
    bool  GetStopFlag();    


    unsigned int GetID(void);
    
    
protected:    
    void SetThreadArg(CThreadArg* pThrArg = NULL);
    CThreadArg* GetThreadArg(void);

    virtual void Run(void* arg) = 0;//1.子类重载此函数，在run里面完成所有功能.2.子类要检查GetStopFlag()，以配合对该线程的“停止”;或者采用强制终止Terminate()

private:
    static THREAD_API Routine(void* arg);    
};


#endif




 下面是Thread.cpp

#include "Thread.h"

CThread::CThread():m_dwThreadId(0),m_bStopFlag(false),m_bStarted(false)
{
#ifdef WIN32
    m_hThreadHandle = NULL;
    SetThreadArg(NULL);
#endif
}

CThread::~CThread()
{
#ifdef WIN32
    if (m_hThreadHandle != NULL)
    {
        CloseHandle(m_hThreadHandle);
    }
#endif

    if (m_pThrArg != NULL)
    {
        delete m_pThrArg;
        m_pThrArg = NULL;
    }
}

int CThread::Start(void* arg /* = NULL */)
{
    if (m_bStarted)
    {
        return 0;
    }
    else
    {
        m_bStarted = true;
    }
    
    CThreadArg *pThrArg = new CThreadArg();
    pThrArg->m_pArg = arg;
    pThrArg->m_pThread = this;


#ifdef WIN32        
    m_hThreadHandle = CreateThread(NULL,NULL,Routine,(void*)pThrArg,NULL,&m_dwThreadId);
    if (m_hThreadHandle != NULL)
    {
        m_pThrArg = pThrArg;
        return 0;
    }
    else
    {
        delete pThrArg;
        return -1;
    }

#else
    int ret = pthread_create(&m_dwThreadId,NULL,Routine,(void*)pThrArg);
    if (ret == 0)
    {
        m_pThrArg = pThrArg;
        return 0;
    }
    else
    {
        delete pThrArg;
        return -1;
    }        
#endif        
}

int CThread::Wait()
{
#ifdef WIN32
    ::WaitForSingleObject(m_hThreadHandle,INFINITE);
    BOOL bRet = CloseHandle(m_hThreadHandle);
    if (bRet)
    {
        m_hThreadHandle = NULL;
        return 0;
    }
    else
        return -1;

#else
    int nRet = pthread_join(m_dwThreadId,NULL);
    if(nRet == 0)
        return 0;
    else
        return -1;        
#endif        
}

void CThread::SetStopFlag()
{
    m_bStopFlag = true;
    return;
}

bool CThread::GetStopFlag()
{
    return m_bStopFlag;
}

void CThread::Terminate()
{
#ifdef WIN32
    if (GetCurrentThreadId() == m_dwThreadId)
    {
        ExitThread(0);
        return;
    }
    else
    {        
        TerminateThread(m_hThreadHandle,0);
        DWORD dwRet = 0;
        GetExitCodeThread(m_hThreadHandle,&dwRet);
        
        while (dwRet == STILL_ACTIVE)
        {
            printf("still active \n");
            Sleep(100);
            TerminateThread(m_hThreadHandle,0);
            GetExitCodeThread(m_hThreadHandle,&dwRet);
        }
    }
#else
    if (pthread_self() == m_dwThreadId)
    {
        pthread_exit(NULL);
        return;
    }
    else
    {
        pthread_cancel(m_dwThreadId);
    }
    
#endif
    
    SetStopFlag();
    return;
}

unsigned int CThread::GetID(void)
{
    return (unsigned int) m_dwThreadId;
}

void CThread::SetThreadArg(CThreadArg* pThrArg /* = NULL */)
{
    m_pThrArg = pThrArg;
    return;
}

CThread::CThreadArg* CThread::GetThreadArg()
{
    return m_pThrArg;
}

THREAD_API CThread::Routine(void* arg)
{

    CThreadArg* pThrArg = (CThreadArg*)arg;

#ifndef WIN32
    pthread_setcancelstate(PTHREAD_CANCEL_ENABLE,NULL);
    pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS,NULL);
#endif

    pThrArg->m_pThread->Run(pThrArg->m_pArg);    
    return NULL;
}

 自己用的，如果你有好的想法，告诉我！
posted on 2009-01-19 12:54 true 阅读(2750) 评论(12)  编辑 收藏 引用 所属分类: linux

Feedback

# re: 自己造的一个线程类 2009-01-19 14:13 eXile
Win32最好不要使用CreateThread, 使用 _beginthreadex  回复  更多评论
  

# re: 自己造的一个线程类[未登录] 2009-01-19 20:58 true
@eXile
嗯，这个网上有这种说法，不过我目前主要在linux，至于win下没那么讲究。  回复  更多评论
  

# re: 自己造的一个线程类 2009-01-21 12:38 星绽紫辉
具体可查看《Windows核心编程》,里面讲到CreateThread可能不会释放tid线程块，而C运行库的_beginthreadex在内部调用了CreateThread，很好地解决了释放问题。  回复  更多评论
  

# re: 自己造的一个线程类 2009-02-18 18:15 陈头
linux下使用，竟然不能使用工厂模式，
创建线程工厂有错误。原因是Run方法，郁闷  回复  更多评论
  

# re: 自己造的一个线程类 2009-02-18 18:22 true
陈：
Run是纯虚函数，必须重载，然后才能实例化。  回复  更多评论
  

# re: 自己造的一个线程类 2009-03-10 15:13 陈头
请问如何显示的析构 new 出来的线程类呢?
  回复  更多评论
  

# re: 自己造的一个线程类 2009-03-10 15:28 true
举个例子：
class CThreadTest : public CThread
{
protected:
void Run(void* arg)
{
printf("arg = %s\n",(char*)arg);
sleep(10);
printf("after sleep 10s");
}

};
步骤：
1.char* pArg = "hello";CThreadTest *pTest = new CThreadTest
2.pTest->Start(pArg);
3.pTest->Wait();
4.delete pTest;
你可以在CThreadTest 里面实现，delete的时候，自动终止线程  回复  更多评论
  

# re: 自己造的一个线程类 2009-03-14 23:21 zengfanmaio

pthread_create第二个参数是空，默认有PTHREAD_CREATE_JOINABLE属性。因此，析构的时候，也要调用一下pthread_join（对应Win32的CloseHandle）。如果设置了PTHREAD_CREATE_DETACHED就不需要。：）  回复  更多评论
  

# re: 自己造的一个线程类 2009-03-16 08:34 true
@zengfanmaio
好久不见！！  回复  更多评论
  

# re: 自己造的一个线程类 2009-03-24 19:20 老陈头
谢谢  回复  更多评论
  

# re: 自己造的一个线程类[未登录] 2009-09-11 20:16 欲三更
谢谢分享，很通用的设计，VCL类库里的TThread类基本就是这个样子。

说一点个人的想法：我自己写的类也差不多是这样，不过有一点差别：我个人认为“线程”和“线程要跑的任务”是两个东西，前者是内核对象（抱歉，我总是改不了用win32本位思考）的抽象，跟具体功能没关系，后者是功能的抽象，跟具体程序有关系。

所以我倾向于把static THREAD_API Routine(void* arg)分离出来，成为一个ITask接口。

这样做还有一个好处，当Thread跑完了Task的代码后可以阻塞在那里等待跑别的Task。

另外我对protobuf也蛮感兴趣的，但是我有点看不懂那个东西，功力太浅：）  回复  更多评论
  

# re: 自己造的一个线程类 2009-09-11 20:34 true
@ 欲三更
线程的抽象问题，也算是见仁见智，你描述的方式类似boost的实现(好像是)，而博文中的抽象是基于“主动对象”这个概念的，我个人觉得，“主动对象”更容易理解。
libprotobuf是个好东东，多看几遍就好了：），有啥问题可以一起交流。关于数据格式，还可以看一下json，感觉不错。
  回复  更多评论
  


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理