완성 원본 샘플

  • 아티클

참고 항목

이 항목의 샘플 코드 대신 cpp-async GitHub 리포지토리에 프로덕션 준비 버전의 작업 완료 원본 구현에 대한 소스 코드가 있습니다.

이 항목에서는 .NET의 TaskCompletionSource와 유사한 자체 완성 소스 클래스를 작성하고 사용하는 방법을 보여줍니다.

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의 한 가지 사용 사례를 보여줍니다. Windows 콘솔 애플리케이션(C++/WinRT) 프로젝트 템플릿을 기반으로 Visual Studio에서 새 프로젝트를 만들고 다음 코드 목록을 main.cpp에 붙여넣습니다(이전 섹션의 목록을 기반으로 complete_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를 캡슐화하고 값을 반환

다음 예제에서는 간단한 앱 클래스를 사용하여 completion_source를 캡슐화하고 완료되면 값을 반환합니다. Windows 콘솔 애플리케이션(C++/WinRT) 프로젝트 템플릿을 기반으로 Visual Studio에서 새 프로젝트를 만들고 다음 코드 목록을 main.cpp에 붙여넣습니다(이전 섹션의 목록을 기반으로 complete_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());
}