| // |
| // impl/thread_pool.hpp |
| // ~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2015 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_IMPL_THREAD_POOL_HPP |
| #define ASIO_IMPL_THREAD_POOL_HPP |
| |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
| # pragma once |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
| |
| #include "asio/detail/executor_op.hpp" |
| #include "asio/detail/fenced_block.hpp" |
| #include "asio/detail/recycling_allocator.hpp" |
| #include "asio/detail/type_traits.hpp" |
| #include "asio/execution_context.hpp" |
| |
| #include "asio/detail/push_options.hpp" |
| |
| namespace asio { |
| |
| inline thread_pool::executor_type |
| thread_pool::get_executor() ASIO_NOEXCEPT |
| { |
| return executor_type(*this); |
| } |
| |
| inline thread_pool& |
| thread_pool::executor_type::context() const ASIO_NOEXCEPT |
| { |
| return pool_; |
| } |
| |
| inline void |
| thread_pool::executor_type::on_work_started() const ASIO_NOEXCEPT |
| { |
| pool_.scheduler_.work_started(); |
| } |
| |
| inline void thread_pool::executor_type::on_work_finished() |
| const ASIO_NOEXCEPT |
| { |
| pool_.scheduler_.work_finished(); |
| } |
| |
| template <typename Function, typename Allocator> |
| void thread_pool::executor_type::dispatch( |
| ASIO_MOVE_ARG(Function) f, const Allocator& a) const |
| { |
| // Make a local, non-const copy of the function. |
| typedef typename decay<Function>::type function_type; |
| function_type tmp(ASIO_MOVE_CAST(Function)(f)); |
| |
| // Invoke immediately if we are already inside the thread pool. |
| if (pool_.scheduler_.can_dispatch()) |
| { |
| detail::fenced_block b(detail::fenced_block::full); |
| asio_handler_invoke_helpers::invoke(tmp, tmp); |
| return; |
| } |
| |
| // Construct an allocator to be used for the operation. |
| typedef typename detail::get_recycling_allocator<Allocator>::type alloc_type; |
| alloc_type allocator(detail::get_recycling_allocator<Allocator>::get(a)); |
| |
| // Allocate and construct an operation to wrap the function. |
| typedef detail::executor_op<function_type, alloc_type> op; |
| typename op::ptr p = { allocator, 0, 0 }; |
| p.v = p.a.allocate(1); |
| p.p = new (p.v) op(tmp, allocator); |
| |
| ASIO_HANDLER_CREATION((pool_, *p.p, |
| "thread_pool", this, 0, "dispatch")); |
| |
| pool_.scheduler_.post_immediate_completion(p.p, false); |
| p.v = p.p = 0; |
| } |
| |
| template <typename Function, typename Allocator> |
| void thread_pool::executor_type::post( |
| ASIO_MOVE_ARG(Function) f, const Allocator& a) const |
| { |
| // Make a local, non-const copy of the function. |
| typedef typename decay<Function>::type function_type; |
| function_type tmp(ASIO_MOVE_CAST(Function)(f)); |
| |
| // Construct an allocator to be used for the operation. |
| typedef typename detail::get_recycling_allocator<Allocator>::type alloc_type; |
| alloc_type allocator(detail::get_recycling_allocator<Allocator>::get(a)); |
| |
| // Allocate and construct an operation to wrap the function. |
| typedef detail::executor_op<function_type, alloc_type> op; |
| typename op::ptr p = { allocator, 0, 0 }; |
| p.v = p.a.allocate(1); |
| p.p = new (p.v) op(tmp, allocator); |
| |
| ASIO_HANDLER_CREATION((pool_, *p.p, |
| "thread_pool", this, 0, "post")); |
| |
| pool_.scheduler_.post_immediate_completion(p.p, false); |
| p.v = p.p = 0; |
| } |
| |
| template <typename Function, typename Allocator> |
| void thread_pool::executor_type::defer( |
| ASIO_MOVE_ARG(Function) f, const Allocator& a) const |
| { |
| // Make a local, non-const copy of the function. |
| typedef typename decay<Function>::type function_type; |
| function_type tmp(ASIO_MOVE_CAST(Function)(f)); |
| |
| // Construct an allocator to be used for the operation. |
| typedef typename detail::get_recycling_allocator<Allocator>::type alloc_type; |
| alloc_type allocator(detail::get_recycling_allocator<Allocator>::get(a)); |
| |
| // Allocate and construct an operation to wrap the function. |
| typedef detail::executor_op<function_type, alloc_type> op; |
| typename op::ptr p = { allocator, 0, 0 }; |
| p.v = p.a.allocate(1); |
| p.p = new (p.v) op(tmp, allocator); |
| |
| ASIO_HANDLER_CREATION((pool_, *p.p, |
| "thread_pool", this, 0, "defer")); |
| |
| pool_.scheduler_.post_immediate_completion(p.p, true); |
| p.v = p.p = 0; |
| } |
| |
| inline bool |
| thread_pool::executor_type::running_in_this_thread() const ASIO_NOEXCEPT |
| { |
| return pool_.scheduler_.can_dispatch(); |
| } |
| |
| } // namespace asio |
| |
| #include "asio/detail/pop_options.hpp" |
| |
| #endif // ASIO_IMPL_THREAD_POOL_HPP |