跨平台(linux/win)启动线程, 并设置线程名称(便于调试)

来源：互联网发布：秀萝捏脸数据编辑：程序博客网时间：2024/06/06 05:54
#ifndef THREAD_H#define THREAD_H#include <string>#include <iostream>#ifdef WIN32#else#include <pthread.h>#endifusing namespace std;class CThread{public:    CThread(const string &strThreadName = "");    virtual ~CThread();    virtual unsigned          ThreadProc() = 0;    const string&             GetThreadName();    void                      SetThreadName(const string &strThreadName);    void                      WaitThisThread();#ifdef WIN32    static unsigned __stdcall s_ThreadProc(void* self);    HANDLE                    StartThread();private:    void                      SetNameInternal(DWORD dwThreadID, const char* cThreadName);public:    HANDLE                    m_handle;#else    static void *             s_ThreadProc(void* self);    pthread_t                StartThread();public:    pthread_t                m_id;#endifprivate:    string                   m_strThreadName;};#endif // THREAD_H#include "stdafx.h"  // win文件包含在预编译头文件里#include "thread.h"#ifdef WIN32#else#include <sys/prctl.h>#endifCThread::CThread(const string &strThreadName):m_strThreadName(strThreadName),#ifdef WIN32    m_handle(0)#else    m_id(0)#endif{}CThread::~CThread(){}const string& CThread::GetThreadName(){    return m_strThreadName;}void CThread::SetThreadName(const string &strThreadName){    m_strThreadName = strThreadName;}#ifdef WIN32 // win实现方式const DWORD MS_VC_EXCEPTION = 0x406D1388;// https://msdn.microsoft.com/zh-cn/library/xcb2z8hs.aspx#pragma pack(push,8)typedef struct tagTHREADNAME_INFO{    DWORD dwType; // Must be 0x1000.    LPCSTR szName; // Pointer to name (in user addr space).    DWORD dwThreadID; // Thread ID (-1=caller thread).    DWORD dwFlags; // Reserved for future use, must be zero.} THREADNAME_INFO;#pragma pack(pop)void CThread::SetNameInternal(DWORD dwThreadID, const char* cThreadName){    THREADNAME_INFO info;    info.dwType = 0x1000;    info.szName = cThreadName;    info.dwThreadID = dwThreadID;    info.dwFlags = 0;    __try    {        RaiseException( MS_VC_EXCEPTION, 0, sizeof(info)/sizeof(ULONG_PTR), (ULONG_PTR*)&info );    }    __except(EXCEPTION_EXECUTE_HANDLER)    {    }}unsigned __stdcall CThread::s_ThreadProc(void* self){    // self obj    CThread *_self = (CThread*)self;    // run proc    return _self->ThreadProc();}HANDLE CThread::StartThread(){    // 线程id    DWORD nThreadId;    // 启动线程    m_handle = (HANDLE)_beginthreadex(NULL, 0, s_ThreadProc, (CThread *)this, 0, (unsigned int*)&nThreadId);        // 设置线程名    SetNameInternal(nThreadId, m_strThreadName.c_str());    return m_handle;}void CThread::WaitThisThread(){    ::WaitForSingleObject( m_handle, INFINITE );}#else // linux实现方式void * __stdcall CThread::s_ThreadProc(void* self){    // self obj    CThread *_self = (CThread*)self;    // set thread name    prctl(PR_SET_NAME, _self->m_strThreadName.c_str());    // run proc    _self->ThreadProc();    return NULL;}pthread_t CThread::StartThread(){    pthread_create(&m_id, NULL, s_ThreadProc, this);    return m_id;}void CThread::WaitThisThread(){    pthread_join(m_id,NULL);}#endif
0 0