您的位置:首页 > 其它

Chapter11-"windows线程池" 之 内核对象触发调用回调函数

2013-01-26 23:58 387 查看
     众所周知,一般可以调用WaitForSingleObject、WaitForMultiObject来实现线程同步。但是当多个线程等待同一个内核对象变为触发态时,这并不是一种明智的做法,这样会浪费大量的系统资源。而windows线程池很有效提供系统性能。

     利用windows线程池实现,当内核对象变为触发态时,自动调用一个回调函数,一如既往的需要四步

创建对应的回调函数,回调函数必须符合如下原型
VOID CALLBACK WaitCallback(
_Inout_      PTP_CALLBACK_INSTANCE Instance,
_Inout_opt_  PVOID Context,
_Inout_      PTP_WAIT Wait,
_In_         TP_WAIT_RESULT WaitResult // 表示WaitCallback函数被执行的原因
// WAIT_OBJECT_0 表示等待的内核对象变为触发态
而执行WaitCallback
// WAIT_TIMEOUT 表示因为超时了
而执行WaitCallback函数
// WAIT_ABANDONED_0 表示等待的内核对象被"抛弃"了。
)

调用CreateThreadpoolWait函数创建对应的等待项
PTP_WAIT WINAPI CreateThreadpoolWait(
_In_         PTP_WAIT_CALLBACK pfnwa,//对应上面的WaitCallback函数
_Inout_opt_  PVOID pv, //对应WaitCallback函数的第二个参数Context
_In_opt_     PTP_CALLBACK_ENVIRON pcbe // WaitCallback函数的运行环境
);

调用SetThreadpoolWait函数项线程池注册一个由CreateThreadpoolWait函数创建的 PTP_WAIT
VOID WINAPI SetThreadpoolWait(
_Inout_   PTP_WAIT pwa, // 由CreateThreadpoolWait函数创建的 PTP_WAIT
_In_opt_  HANDLE h, // 等待的内核对象
_In_opt_  PFILETIME pftTimeout // 最长的等待时间
);

完成之后,调用 CloseThreadpoolWait 函数释放资源。
#include <windows.h>
#include <iostream>
#include <time.h>

VOID CALLBACK WaitCallback(
_Inout_      PTP_CALLBACK_INSTANCE Instance,
_Inout_opt_  PVOID Context,
_Inout_      PTP_WAIT Wait,
_In_         TP_WAIT_RESULT WaitResult
)
{
printf("Wait: %s\n", Context);
}

void   main()
{
HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
PTP_WAIT ptpWait[10];
char *p[10];
ULARGE_INTEGER ulRelativeStartTime;
ulRelativeStartTime.QuadPart = (LONGLONG) -(50000000); // 5秒
FILETIME ftRelativeStartTime;
ftRelativeStartTime.dwHighDateTime = ulRelativeStartTime.HighPart;
ftRelativeStartTime.dwLowDateTime  = ulRelativeStartTime.LowPart;

for (int i = 0; i < 10; i++)
{
p[i] = new char[2];
p[i][0] = '0' + i;
p[i][1] = 0;
ptpWait[i] = CreateThreadpoolWait(WaitCallback, p[i], NULL);
SetThreadpoolWait(ptpWait[i], hEvent, &ftRelativeStartTime);
}

//如果在"5秒"内按下任意键,则后面的 SetEvent 函数会将 hEvent 变为触发态,并且没有 PTP_WAIT 过期,
//    从10个 PTP_WORK 中任意选出一个激活,接着就会触发对应的WaitCallback函数
//如果超过"5秒"才按下任意键,则所有的 PTP_WORK 都已过期,10 个 PTP_WAIT都会被激活,
//    接着对应的 10 个 WaitCallback 函数会被执行。
getchar();
SetEvent(hEvent);
getchar();

//释放资源
for (int i = 0; i < 10; i++)
{
CloseThreadpoolWait(ptpWait[i]);

delete []p[i];
p[i] = NULL;
}

printf("all the wait have been closed!\n");
}
《windows核心编程》(笔记)系列文章是本人看《windows核心编程》时的一些学习笔记,有疏忽之处,欢迎各位网友指正。QQ邮箱:job.zhanghui@qq.com
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: