线程池

#pragma once
#include <windows.h>

#ifndef XC_API
#define XC_API extern "C"
#endif

typedef  unsigned int(__stdcall *WORKERPROC)(void * lpParameter, DWORD userData);
typedef struct _ThreadPool
{
HANDLE hIoCompletionPort;
HANDLE hExitEvent;
DWORD  dwUserData;
HANDLE hThread[MAXIMUM_WAIT_OBJECTS];
LONG  dwItemCount;  //线程池中还未完成的数量
}ThreadPool, *HThreadPool;

XC_API HThreadPool WINAPI XThreadPool_Create();
XC_API void WINAPI XThreadPool_Destroy(HThreadPool hThreadPool);
XC_API void WINAPI XThreadPool_PostItem(HThreadPool hThreadPool, WORKERPROC userWorker, LPVOID lpParam, DWORD userData);
XC_API void WINAPI XThreadPool_SuspendAll(HThreadPool hThreadPool);
XC_API void WINAPI XThreadPool_ResumeAll(HThreadPool hThreadPool);
XC_API DWORD WINAPI XThread

#include "ThreadPool.h"
#include <process.h>

unsigned int WINAPI Worker(void* lpParam)
{
HThreadPool pThreadPool = (HThreadPool)lpParam;

DWORD dwUserData;
WORKERPROC userCallBack = NULL;
LPVOID      lParam;
while (WaitForSingleObject(pThreadPool->hExitEvent, 0) != WAIT_OBJECT_0)
{
if (GetQueuedCompletionStatus(
pThreadPool->hIoCompletionPort,
&dwUserData,
(PULONG)&userCallBack,
(LPOVERLAPPED*)&lParam,
INFINITE
))
{
if (userCallBack)
{
userCallBack(lParam, dwUserData);
InterlockedDecrement(&pThreadPool->dwItemCount);
}

}
}
return 0;
}

void WINAPI XThreadPool_PostItem(HThreadPool hThreadPool, WORKERPROC userWorker, LPVOID lpParam, DWORD userData)
{
if (hThreadPool)
{
PostQueuedCompletionStatus(
hThreadPool->hIoCompletionPort,
userData,
(ULONG_PTR)userWorker,
(LPOVERLAPPED)lpParam
);
InterlockedIncrement(&hThreadPool->dwItemCount);
}
}
void WINAPI XThreadPool_SuspendAll(HThreadPool hThreadPool)
{
if (hThreadPool)
{
for (int i = 0; i < MAXIMUM_WAIT_OBJECTS; i++)
{
SuspendThread(hThreadPool->hThread[i]);
}
}
}

DWORD WINAPI XThreadPool_GetWorkItem(HThreadPool hThreadPool)
{
if (hThreadPool)
{
return hThreadPool->dwItemCount;
}
return 0;
}

void WINAPI XThreadPool_ResumeAll(HThreadPool hThreadPool)
{
if (hThreadPool)
{
for (int i = 0; i < MAXIMUM_WAIT_OBJECTS; i++)
{
ResumeThread(hThreadPool->hThread[i]);
}
}
}

HThreadPool WINAPI XThreadPool_Create()
{
HThreadPool pThreadPool = new ThreadPool;
pThreadPool->hIoCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, 0, NULL, 0);
pThreadPool->hExitEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
pThreadPool->dwItemCount = 0;

for (int i = 0; i < MAXIMUM_WAIT_OBJECTS; i++)
{
pThreadPool->hThread[i] = (HANDLE)_beginthreadex(
NULL, 0,
Worker, (void*)pThreadPool, 0, NULL);
}
return pThreadPool;
}

void WINAPI XThreadPool_Destroy(HThreadPool hThreadPool)
{
if (hThreadPool)
{
SetEvent(hThreadPool->hExitEvent);
for (int i = 0; i < MAXIMUM_WAIT_OBJECTS; i++)
{
// 通知所有的完成端口操作退出
PostQueuedCompletionStatus(hThreadPool->hIoCompletionPort, 0, 0, NULL);
}
WaitForMultipleObjects(MAXIMUM_WAIT_OBJECTS, hThreadPool->hThread, TRUE, INFINITE);
for (int j = 0; j < MAXIMUM_WAIT_OBJECTS; j++)
{
CloseHandle(hThreadPool->hThread[j]);
}
CloseHandle(hThreadPool->hIoCompletionPort);

delete hThreadPool;
hThreadPool = NULL;
}
}