网络服务器软件开发/中间件开发,关注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 阅读(2697) 评论(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,感觉不错。
  回复  更多评论
  


只有注册用户登录后才能发表评论。
【推荐】超50万行VC++源码: 大型组态工控、电力仿真CAD与GIS源码库
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理