Пример источника завершения
Примечание.
В качестве альтернативы примеру кода в этом разделе существует исходный код для готовой к рабочей версии реализации источника завершения задачи в репозитории cpp-async GitHub.
В этой статье показано, как создать и использовать собственный класс источника завершения, аналогичный TaskCompletionSource, доступному в .NET.
Исходный код для примера completion_source
Приведенный ниже код предлагается в качестве примера. Он иллюстрирует, как написать свою версию такого кода. Например, поддержка отмены и распространения ошибок выходит за рамки этого примера.
#include <winrt/base.h>
#include <windows.h>
template <typename T>
struct completion_source
{
completion_source()
{
m_signal.attach(::CreateEvent(nullptr, true, false, nullptr));
}
void set(T const& value)
{
m_value = value;
::SetEvent(m_signal.get());
}
bool await_ready() const noexcept
{
return ::WaitForSingleObject(m_signal.get(), 0) == 0;
}
void await_suspend(std::experimental::coroutine_handle<> resume)
{
m_wait.attach(winrt::check_pointer(::CreateThreadpoolWait(callback, resume.address(), nullptr)));
::SetThreadpoolWait(m_wait.get(), m_signal.get(), nullptr);
}
T await_resume() const noexcept
{
return m_value;
}
private:
static void __stdcall callback(PTP_CALLBACK_INSTANCE, void* context, PTP_WAIT, TP_WAIT_RESULT) noexcept
{
std::experimental::coroutine_handle<>::from_address(context)();
}
struct wait_traits
{
using type = PTP_WAIT;
static void close(type value) noexcept
{
::CloseThreadpoolWait(value);
}
static constexpr type invalid() noexcept
{
return nullptr;
}
};
winrt::handle m_signal;
winrt::handle_type<wait_traits> m_wait;
T m_value{};
};
Передача завершения в отдельную сопрограмму
В этом разделе показан один вариант использования для completion_source. Создайте в Visual Studio проект на основе шаблона проекта Консольное приложение Windows (C++/WinRT) и вставьте приведенный ниже код в main.cpp
(расширение определения completion_source на основе списка из предыдущего раздела).
// main.cpp
#include "pch.h"
#include <winrt/base.h>
#include <windows.h>
template <typename T>
struct completion_source
{
... // Paste the listing of completion_source here.
}
using namespace std::literals;
using namespace winrt;
using namespace Windows::Foundation;
fire_and_forget CompleteAfterFiveSecondsAsync(completion_source<bool>& completionSource)
{
co_await 5s;
completionSource.set(true);
}
IAsyncAction CompletionSourceExample1Async()
{
completion_source<bool> completionSource;
CompleteAfterFiveSecondsAsync(completionSource);
co_await completionSource;
}
int main()
{
auto asyncAction { CompletionSourceExample1Async() };
puts("waiting");
asyncAction.get();
puts("done");
}
Инкапсуляция completion_source в класс и возврат значения
В следующем примере используется простой класс App для инкапсуляции completion_source и возвращения значения после завершения. Создайте в Visual Studio проект на основе шаблона проекта Консольное приложение Windows (C++/WinRT) и вставьте приведенный ниже код в main.cpp
(расширение определения completion_source на основе списка из предыдущего раздела).
// main.cpp
#include "pch.h"
#include <winrt/base.h>
#include <windows.h>
template <typename T>
struct completion_source
{
... // Paste the listing of completion_source here.
}
using namespace std::literals;
using namespace winrt;
using namespace Windows::Foundation;
struct App
{
completion_source<winrt::hstring> m_completionSource;
IAsyncOperation<winrt::hstring> CompletionSourceExample2Async()
{
co_return co_await m_completionSource;
}
winrt::fire_and_forget CompleteAfterFiveSecondsAsync()
{
co_await 5s;
m_completionSource.set(L"Hello, World!");
}
};
int main()
{
App app;
auto asyncAction{ app.CompletionSourceExample2Async() };
app.CompleteAfterFiveSecondsAsync();
puts("waiting");
auto message = asyncAction.get();
printf("%ls\n", message.c_str());
}