| // |
| // detail/handler_work.hpp |
| // ~~~~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2020 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_DETAIL_HANDLER_WORK_HPP |
| #define ASIO_DETAIL_HANDLER_WORK_HPP |
| |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
| # pragma once |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
| |
| #include "asio/detail/config.hpp" |
| #include "asio/associated_executor.hpp" |
| #include "asio/detail/handler_invoke_helpers.hpp" |
| #include "asio/system_executor.hpp" |
| |
| #include "asio/detail/push_options.hpp" |
| |
| namespace asio { |
| namespace detail { |
| |
| // A helper class template to allow completion handlers to be dispatched |
| // through either the new executors framework or the old invocaton hook. The |
| // primary template uses the new executors framework. |
| template <typename Handler, |
| typename IoExecutor, typename HandlerExecutor |
| = typename associated_executor<Handler, IoExecutor>::type> |
| class handler_work |
| { |
| public: |
| explicit handler_work(Handler& handler) ASIO_NOEXCEPT |
| : io_executor_(), |
| executor_(asio::get_associated_executor(handler, io_executor_)) |
| { |
| } |
| |
| handler_work(Handler& handler, const IoExecutor& io_ex) ASIO_NOEXCEPT |
| : io_executor_(io_ex), |
| executor_(asio::get_associated_executor(handler, io_executor_)) |
| { |
| } |
| |
| static void start(Handler& handler, |
| const IoExecutor& io_ex) ASIO_NOEXCEPT |
| { |
| HandlerExecutor ex(asio::get_associated_executor(handler, io_ex)); |
| ex.on_work_started(); |
| io_ex.on_work_started(); |
| } |
| |
| ~handler_work() |
| { |
| io_executor_.on_work_finished(); |
| executor_.on_work_finished(); |
| } |
| |
| template <typename Function> |
| void complete(Function& function, Handler& handler) |
| { |
| executor_.dispatch(ASIO_MOVE_CAST(Function)(function), |
| asio::get_associated_allocator(handler)); |
| } |
| |
| private: |
| // Disallow copying and assignment. |
| handler_work(const handler_work&); |
| handler_work& operator=(const handler_work&); |
| |
| IoExecutor io_executor_; |
| HandlerExecutor executor_; |
| }; |
| |
| // This specialisation dispatches a handler through the old invocation hook. |
| // The specialisation is not strictly required for correctness, as the |
| // system_executor will dispatch through the hook anyway. However, by doing |
| // this we avoid an extra copy of the handler. |
| template <typename Handler> |
| class handler_work<Handler, system_executor, system_executor> |
| { |
| public: |
| explicit handler_work(Handler&) ASIO_NOEXCEPT {} |
| static void start(Handler&, const system_executor&) ASIO_NOEXCEPT {} |
| ~handler_work() {} |
| |
| template <typename Function> |
| void complete(Function& function, Handler& handler) |
| { |
| asio_handler_invoke_helpers::invoke(function, handler); |
| } |
| |
| private: |
| // Disallow copying and assignment. |
| handler_work(const handler_work&); |
| handler_work& operator=(const handler_work&); |
| }; |
| |
| } // namespace detail |
| } // namespace asio |
| |
| #include "asio/detail/pop_options.hpp" |
| |
| #endif // ASIO_DETAIL_HANDLER_WORK_HPP |