168 lines
5.3 KiB
C++
168 lines
5.3 KiB
C++
//
|
|
// use_awaitable.hpp
|
|
// ~~~~~~~~~~~~~~~~~
|
|
//
|
|
// Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef ASIO_USE_AWAITABLE_HPP
|
|
#define ASIO_USE_AWAITABLE_HPP
|
|
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
# pragma once
|
|
#endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
|
|
|
|
#include "asio/detail/config.hpp"
|
|
|
|
#if defined(ASIO_HAS_CO_AWAIT) || defined(GENERATING_DOCUMENTATION)
|
|
|
|
#include "asio/awaitable.hpp"
|
|
#include "asio/detail/handler_tracking.hpp"
|
|
|
|
#if defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
# if defined(ASIO_HAS_SOURCE_LOCATION)
|
|
# include "asio/detail/source_location.hpp"
|
|
# endif // defined(ASIO_HAS_SOURCE_LOCATION)
|
|
#endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
|
|
#include "asio/detail/push_options.hpp"
|
|
|
|
namespace asio {
|
|
|
|
/// A @ref completion_token that represents the currently executing coroutine.
|
|
/**
|
|
* The @c use_awaitable_t class, with its value @c use_awaitable, is used to
|
|
* represent the currently executing coroutine. This completion token may be
|
|
* passed as a handler to an asynchronous operation. For example:
|
|
*
|
|
* @code awaitable<void> my_coroutine()
|
|
* {
|
|
* std::size_t n = co_await my_socket.async_read_some(buffer, use_awaitable);
|
|
* ...
|
|
* } @endcode
|
|
*
|
|
* When used with co_await, the initiating function (@c async_read_some in the
|
|
* above example) suspends the current coroutine. The coroutine is resumed when
|
|
* the asynchronous operation completes, and the result of the operation is
|
|
* returned.
|
|
*/
|
|
template <typename Executor = any_io_executor>
|
|
struct use_awaitable_t
|
|
{
|
|
/// Default constructor.
|
|
ASIO_CONSTEXPR use_awaitable_t(
|
|
#if defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
# if defined(ASIO_HAS_SOURCE_LOCATION)
|
|
detail::source_location location = detail::source_location::current()
|
|
# endif // defined(ASIO_HAS_SOURCE_LOCATION)
|
|
#endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
)
|
|
#if defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
# if defined(ASIO_HAS_SOURCE_LOCATION)
|
|
: file_name_(location.file_name()),
|
|
line_(location.line()),
|
|
function_name_(location.function_name())
|
|
# else // defined(ASIO_HAS_SOURCE_LOCATION)
|
|
: file_name_(0),
|
|
line_(0),
|
|
function_name_(0)
|
|
# endif // defined(ASIO_HAS_SOURCE_LOCATION)
|
|
#endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
{
|
|
}
|
|
|
|
/// Constructor used to specify file name, line, and function name.
|
|
ASIO_CONSTEXPR use_awaitable_t(const char* file_name,
|
|
int line, const char* function_name)
|
|
#if defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
: file_name_(file_name),
|
|
line_(line),
|
|
function_name_(function_name)
|
|
#endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
{
|
|
#if !defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
(void)file_name;
|
|
(void)line;
|
|
(void)function_name;
|
|
#endif // !defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
}
|
|
|
|
/// Adapts an executor to add the @c use_awaitable_t completion token as the
|
|
/// default.
|
|
template <typename InnerExecutor>
|
|
struct executor_with_default : InnerExecutor
|
|
{
|
|
/// Specify @c use_awaitable_t as the default completion token type.
|
|
typedef use_awaitable_t default_completion_token_type;
|
|
|
|
/// Construct the adapted executor from the inner executor type.
|
|
template <typename InnerExecutor1>
|
|
executor_with_default(const InnerExecutor1& ex,
|
|
typename constraint<
|
|
conditional<
|
|
!is_same<InnerExecutor1, executor_with_default>::value,
|
|
is_convertible<InnerExecutor1, InnerExecutor>,
|
|
false_type
|
|
>::type::value
|
|
>::type = 0) ASIO_NOEXCEPT
|
|
: InnerExecutor(ex)
|
|
{
|
|
}
|
|
};
|
|
|
|
/// Type alias to adapt an I/O object to use @c use_awaitable_t as its
|
|
/// default completion token type.
|
|
#if defined(ASIO_HAS_ALIAS_TEMPLATES) \
|
|
|| defined(GENERATING_DOCUMENTATION)
|
|
template <typename T>
|
|
using as_default_on_t = typename T::template rebind_executor<
|
|
executor_with_default<typename T::executor_type> >::other;
|
|
#endif // defined(ASIO_HAS_ALIAS_TEMPLATES)
|
|
// || defined(GENERATING_DOCUMENTATION)
|
|
|
|
/// Function helper to adapt an I/O object to use @c use_awaitable_t as its
|
|
/// default completion token type.
|
|
template <typename T>
|
|
static typename decay<T>::type::template rebind_executor<
|
|
executor_with_default<typename decay<T>::type::executor_type>
|
|
>::other
|
|
as_default_on(ASIO_MOVE_ARG(T) object)
|
|
{
|
|
return typename decay<T>::type::template rebind_executor<
|
|
executor_with_default<typename decay<T>::type::executor_type>
|
|
>::other(ASIO_MOVE_CAST(T)(object));
|
|
}
|
|
|
|
#if defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
const char* file_name_;
|
|
int line_;
|
|
const char* function_name_;
|
|
#endif // defined(ASIO_ENABLE_HANDLER_TRACKING)
|
|
};
|
|
|
|
/// A @ref completion_token object that represents the currently executing
|
|
/// coroutine.
|
|
/**
|
|
* See the documentation for asio::use_awaitable_t for a usage example.
|
|
*/
|
|
#if defined(GENERATING_DOCUMENTATION)
|
|
constexpr use_awaitable_t<> use_awaitable;
|
|
#elif defined(ASIO_HAS_CONSTEXPR)
|
|
constexpr use_awaitable_t<> use_awaitable(0, 0, 0);
|
|
#elif defined(ASIO_MSVC)
|
|
__declspec(selectany) use_awaitable_t<> use_awaitable(0, 0, 0);
|
|
#endif
|
|
|
|
} // namespace asio
|
|
|
|
#include "asio/detail/pop_options.hpp"
|
|
|
|
#include "asio/impl/use_awaitable.hpp"
|
|
|
|
#endif // defined(ASIO_HAS_CO_AWAIT) || defined(GENERATING_DOCUMENTATION)
|
|
|
|
#endif // ASIO_USE_AWAITABLE_HPP
|