Обычный метод состоит в том, чтобы сохранить все дескрипторы потоков, а затем ждать каждого дескриптора. Когда дескриптор сигнализирует , поток завершен, поэтому он удаляется из набора потоков. Я использую std::set<HANDLE>
, чтобы следить за ручками потоков. В Windows существует два метода ожидания для нескольких объектов:
- Перебор набора и вызов
WaitForSingleObject
с таймаутом на каждом
- Преобразовать набор в массив или вектор и вызвать
WaitForMultipleObjects
Первое звучит неэффективно, но на самом деле оно является наиболее прямым и наименее подверженным ошибкам из двух. Если вам нужно дождаться всех потоков, используйте следующий цикл:
std::set<HANDLE> thread_handles; // contains the handle of each worker thread
while (!thread_handles.empty()) {
std::set<HANDLE> threads_left;
for (std::set<HANDLE>::iterator cur_thread=thread_handles.begin(),
last=thread_handles.end();
cur_thread != last; ++cur_thread)
{
DWORD rc = ::WaitForSingleObject(*cur_thread, some_timeout);
if (rc == WAIT_OBJECT_0) {
::CloseHandle(*cur_thread); // necessary with _beginthreadex
} else if (rc == WAIT_TIMEOUT) {
threads_left.add(cur_thread); // wait again
} else {
// this shouldn't happen... try to close the handle and hope
// for the best!
::CloseHandle(*cur_thread); // necessary with _beginthreadex
}
}
std::swap(threads_left, thread_handles);
}
Использование WaitForMultipleObjects
для ожидания завершения потоков немного сложнее, чем кажется. Следующее будет ждать всех потоков; однако, он только ждет потоков WAIT_MAXIMUM_OBJECTS
одновременно. Другим вариантом является цикл по каждой странице потоков. Я оставлю это упражнение читателю;)
DWORD large_timeout = (5 * 60 * 1000); // five minutes
std::set<HANDLE> thread_handles; // contains the handle of each worker thread
std::vector<HANDLE> ary; // WaitForMultipleObjects wants an array...
while (!thread_handles.empty()) {
ary.assign(thread_handles.begin(), thread_handles.end());
DWORD rc = ::WaitForMultipleObjects(std::min(ary.size(), WAIT_MAXIMUM_OBJECTS),
&ary[0], FALSE, large_timeout);
if (rc == WAIT_FAILED) {
// handle a failure case... this is usually something pretty bad
break;
} else if (rc == WAIT_TIMEOUT) {
// no thread exited in five minutes... this can be tricky since one of
// the threads beyond the first WAIT_MAXIMUM_OBJECTS may have terminated
} else {
long idx = (rc - WAIT_OBJECT_0);
if (idx > 0 && idx < ary.size()) {
// the object at `idx` was signaled, this means that the
// thread has terminated.
thread_handles.erase(ary[idx]);
::CloseHandle(ary[idx]); // necessary with _beginthreadex
}
}
}
Это не совсем красиво, но должно работать. Если вы уверены, что все ваши потоки завершатся, и не против их ожидания, вы можете использовать WaitForMultipleObjects(ary.size(), &ary[0], TRUE, INFINITE)
. Это, как правило, не очень безопасно, так как бегущая нить приведет к тому, что ваше приложение будет блокироваться на неопределенный срок и , оно будет работать только если ary.size()
меньше MAXIMUM_WAIT_OBJECTS
.
Конечно, другой вариант - найти реализацию пула потоков и использовать ее вместо этого. Написание многопоточного кода на самом деле не очень весело, особенно если вам нужно поддерживать его на свободе. Попробуйте вместо этого использовать что-то вроде boost::thread_group
.