VC++ATL封装后的线程启动工具类和调用示例

源码资源网

VC++ATL封装后的线程启动工具类和调用示例
该封装提供了两种线程启动方式：支持全局函数和类成员函数调用，线程安全且符合ATL编程规范。
AtlThreadUtils.h

1. 
   
2. // AtlThreadUtils.h
   
3. #pragma once
   
4. #include <Windows.h>
   
5. #include <process.h>
   
6. 
   
7. class CAtlThreadUtils {
   
8. public:
   
9.     static HRESULT BeginThread(
   
10.         LPTHREAD_START_ROUTINE pfnThreadProc,
    
11.         LPVOID pParam,
    
12.         DWORD dwFlags = 0,
    
13.         LPDWORD pdwThreadId = nullptr)
    
14.     {
    
15.         unsigned threadId = 0;
    
16.         HANDLE hThread = (HANDLE)_beginthreadex(
    
17.             NULL, 0,
    
18.             (unsigned(__stdcall*)(void*))pfnThreadProc,
    
19.             pParam, dwFlags, &threadId);
    
20.         
    
21.         if (!hThread) return E_FAIL;
    
22.         
    
23.         if (pdwThreadId) *pdwThreadId = threadId;
    
24.         CloseHandle(hThread);
    
25.         return S_OK;
    
26.     }
    
27. 
    
28.     template<typename T>
    
29.     static HRESULT BeginThreadEx(
    
30.         DWORD (T::*pfnMember)(),
    
31.         T* pInstance,
    
32.         DWORD dwFlags = 0)
    
33.     {
    
34.         struct ThreadData {
    
35.             T* pObj;
    
36.             DWORD (T::*pfn)();
    
37.         };
    
38.         
    
39.         ThreadData* pData = new ThreadData{pInstance, pfnMember};
    
40.         return BeginThread(
    
41.             [](LPVOID p) -> DWORD {
    
42.                 ThreadData* pData = (ThreadData*)p;
    
43.                 (pData->pObj->*(pData->pfn))();
    
44.                 delete pData;
    
45.                 return 0;
    
46.             },
    
47.             pData,
    
48.             dwFlags);
    
49.     }
    
50. };
    

复制代码

Main.cpp

1. // Main.cpp
   
2. #include "AtlThreadUtils.h"
   
3. #include <iostream>
   
4. 
   
5. class CMyTask {
   
6. public:
   
7.     DWORD WorkerProc() {
   
8.         std::cout << "Thread ID: " << GetCurrentThreadId()
   
9.                  << " is running\n";
   
10.         return 0;
    
11.     }
    
12. };
    
13. 
    
14. DWORD __stdcall GlobalThreadProc(LPVOID) {
    
15.     std::cout << "Global thread function executed\n";
    
16.     return 0;
    
17. }
    
18. 
    
19. int main() {
    
20.     // 方式1：调用全局函数
    
21.     CAtlThreadUtils::BeginThread(GlobalThreadProc, nullptr);
    
22. 
    
23.     // 方式2：调用成员函数
    
24.     CMyTask task;
    
25.     CAtlThreadUtils::BeginThreadEx(&CMyTask::WorkerProc, &task);
    
26. 
    
27.     Sleep(1000); // 等待线程执行
    
28.     return 0;
    
29. }
    

复制代码