在软件开发过程中，经常需要处理多线程的情况，尤其是在Windows平台下，DLL（动态链接库）的使用非常普遍。本文将介绍一种在DLL中处理多线程等待和消息循环的方法，这种方法可以在不同操作系统上运行，并且无论调用DLL的应用程序类型如何，都能正常工作，包括在多核处理器的机器上。
在MSI-Setup安装过程中，可能会调用一些DLL来进行特定的操作。这些操作可能需要在后台线程中执行，而主线程则需要等待这些操作完成。如果操作完成，可能需要显示一个对话框。问题在于，如何确保在这些后台操作完成之前，主线程不会过早地显示对话框。
为了解决这个问题，需要编写两个函数。第一个函数用于从消息队列中获取消息，第二个函数用于等待特定线程终止。以下是这两个函数的实现方法：
这个函数负责从消息队列中获取消息，并进行处理。以下是C++代码示例：
void CMyTestDialog::PeekMessageLoop() {
MSG msg;
while (PeekMessage(&msg, NULL, NULL, NULL, PM_REMOVE)) {
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
这个函数接受一个线程句柄作为参数，用于等待该线程终止。以下是C++代码示例：
void CMyTestDialog::WaitForThreadToTerminate(HANDLE hThread) {
DWORD dwRet;
do {
dwRet = ::MsgWaitForMultipleObjects(1, &hThread, FALSE, INFINITE, QS_ALLINPUT);
if (dwRet != WAIT_OBJECT_0) {
PeekMessageLoop();
}
} while ((dwRet != WAIT_OBJECT_0) && (dwRet != WAIT_FAILED));
}
void CMyTestDialog::OnButton1() {
m_pUpdateThread = AfxBeginThread(UpdateDeviceContent, (LPVOID)this);
if (m_pUpdateThread) {
WaitForThreadToTerminate(m_pUpdateThread->m_hThread);
}
// 操作完成后执行其他操作
}