123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965 |
- #ifndef BOOST_ASIO_THREAD_POOL_HPP
- #define BOOST_ASIO_THREAD_POOL_HPP
- #if defined(_MSC_VER) && (_MSC_VER >= 1200)
- # pragma once
- #endif
- #include <boost/asio/detail/config.hpp>
- #include <boost/asio/detail/atomic_count.hpp>
- #include <boost/asio/detail/scheduler.hpp>
- #include <boost/asio/detail/thread_group.hpp>
- #include <boost/asio/execution.hpp>
- #include <boost/asio/execution_context.hpp>
- #include <boost/asio/detail/push_options.hpp>
- namespace boost {
- namespace asio {
- namespace detail {
- struct thread_pool_bits
- {
- static constexpr unsigned int blocking_never = 1;
- static constexpr unsigned int blocking_always = 2;
- static constexpr unsigned int blocking_mask = 3;
- static constexpr unsigned int relationship_continuation = 4;
- static constexpr unsigned int outstanding_work_tracked = 8;
- };
- }
- class thread_pool
- : public execution_context
- {
- public:
- template <typename Allocator, unsigned int Bits>
- class basic_executor_type;
- template <typename Allocator, unsigned int Bits>
- friend class basic_executor_type;
-
- typedef basic_executor_type<std::allocator<void>, 0> executor_type;
- #if !defined(BOOST_ASIO_NO_TS_EXECUTORS)
-
- BOOST_ASIO_DECL thread_pool();
- #endif
-
- BOOST_ASIO_DECL thread_pool(std::size_t num_threads);
-
-
- BOOST_ASIO_DECL ~thread_pool();
-
- executor_type get_executor() noexcept;
-
- executor_type executor() noexcept;
-
-
- BOOST_ASIO_DECL void stop();
-
-
- BOOST_ASIO_DECL void attach();
-
-
- BOOST_ASIO_DECL void join();
-
-
- BOOST_ASIO_DECL void wait();
- private:
- thread_pool(const thread_pool&) = delete;
- thread_pool& operator=(const thread_pool&) = delete;
- struct thread_function;
-
- BOOST_ASIO_DECL detail::scheduler& add_scheduler(detail::scheduler* s);
-
- detail::scheduler& scheduler_;
-
- detail::thread_group threads_;
-
- detail::atomic_count num_threads_;
- };
- template <typename Allocator, unsigned int Bits>
- class thread_pool::basic_executor_type : detail::thread_pool_bits
- {
- public:
-
- basic_executor_type(const basic_executor_type& other) noexcept
- : pool_(other.pool_),
- allocator_(other.allocator_),
- bits_(other.bits_)
- {
- if (Bits & outstanding_work_tracked)
- if (pool_)
- pool_->scheduler_.work_started();
- }
-
- basic_executor_type(basic_executor_type&& other) noexcept
- : pool_(other.pool_),
- allocator_(static_cast<Allocator&&>(other.allocator_)),
- bits_(other.bits_)
- {
- if (Bits & outstanding_work_tracked)
- other.pool_ = 0;
- }
-
- ~basic_executor_type() noexcept
- {
- if (Bits & outstanding_work_tracked)
- if (pool_)
- pool_->scheduler_.work_finished();
- }
-
- basic_executor_type& operator=(const basic_executor_type& other) noexcept;
-
- basic_executor_type& operator=(basic_executor_type&& other) noexcept;
- #if !defined(GENERATING_DOCUMENTATION)
- private:
- friend struct boost_asio_require_fn::impl;
- friend struct boost_asio_prefer_fn::impl;
- #endif
-
-
- constexpr basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED(Bits & ~blocking_mask)>
- require(execution::blocking_t::possibly_t) const
- {
- return basic_executor_type<Allocator, Bits & ~blocking_mask>(
- pool_, allocator_, bits_ & ~blocking_mask);
- }
-
-
- constexpr basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED((Bits & ~blocking_mask) | blocking_always)>
- require(execution::blocking_t::always_t) const
- {
- return basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED((Bits & ~blocking_mask) | blocking_always)>(
- pool_, allocator_, bits_ & ~blocking_mask);
- }
-
-
- constexpr basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED(Bits & ~blocking_mask)>
- require(execution::blocking_t::never_t) const
- {
- return basic_executor_type<Allocator, Bits & ~blocking_mask>(
- pool_, allocator_, (bits_ & ~blocking_mask) | blocking_never);
- }
-
-
- constexpr basic_executor_type require(execution::relationship_t::fork_t) const
- {
- return basic_executor_type(pool_,
- allocator_, bits_ & ~relationship_continuation);
- }
-
-
- constexpr basic_executor_type require(
- execution::relationship_t::continuation_t) const
- {
- return basic_executor_type(pool_,
- allocator_, bits_ | relationship_continuation);
- }
-
-
- constexpr basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED(Bits | outstanding_work_tracked)>
- require(execution::outstanding_work_t::tracked_t) const
- {
- return basic_executor_type<Allocator, Bits | outstanding_work_tracked>(
- pool_, allocator_, bits_);
- }
-
-
- constexpr basic_executor_type<Allocator,
- BOOST_ASIO_UNSPECIFIED(Bits & ~outstanding_work_tracked)>
- require(execution::outstanding_work_t::untracked_t) const
- {
- return basic_executor_type<Allocator, Bits & ~outstanding_work_tracked>(
- pool_, allocator_, bits_);
- }
-
-
- template <typename OtherAllocator>
- constexpr basic_executor_type<OtherAllocator, Bits>
- require(execution::allocator_t<OtherAllocator> a) const
- {
- return basic_executor_type<OtherAllocator, Bits>(
- pool_, a.value(), bits_);
- }
-
-
- constexpr basic_executor_type<std::allocator<void>, Bits>
- require(execution::allocator_t<void>) const
- {
- return basic_executor_type<std::allocator<void>, Bits>(
- pool_, std::allocator<void>(), bits_);
- }
- #if !defined(GENERATING_DOCUMENTATION)
- private:
- friend struct boost_asio_query_fn::impl;
- friend struct boost::asio::execution::detail::mapping_t<0>;
- friend struct boost::asio::execution::detail::outstanding_work_t<0>;
- #endif
-
-
- static constexpr execution::mapping_t query(execution::mapping_t) noexcept
- {
- return execution::mapping.thread;
- }
-
-
- thread_pool& query(execution::context_t) const noexcept
- {
- return *pool_;
- }
-
-
- constexpr execution::blocking_t query(execution::blocking_t) const noexcept
- {
- return (bits_ & blocking_never)
- ? execution::blocking_t(execution::blocking.never)
- : ((Bits & blocking_always)
- ? execution::blocking_t(execution::blocking.always)
- : execution::blocking_t(execution::blocking.possibly));
- }
-
-
- constexpr execution::relationship_t query(
- execution::relationship_t) const noexcept
- {
- return (bits_ & relationship_continuation)
- ? execution::relationship_t(execution::relationship.continuation)
- : execution::relationship_t(execution::relationship.fork);
- }
-
-
- static constexpr execution::outstanding_work_t query(
- execution::outstanding_work_t) noexcept
- {
- return (Bits & outstanding_work_tracked)
- ? execution::outstanding_work_t(execution::outstanding_work.tracked)
- : execution::outstanding_work_t(execution::outstanding_work.untracked);
- }
-
-
- template <typename OtherAllocator>
- constexpr Allocator query(
- execution::allocator_t<OtherAllocator>) const noexcept
- {
- return allocator_;
- }
-
-
- constexpr Allocator query(execution::allocator_t<void>) const noexcept
- {
- return allocator_;
- }
-
-
- std::size_t query(execution::occupancy_t) const noexcept
- {
- return static_cast<std::size_t>(pool_->num_threads_);
- }
- public:
-
-
- bool running_in_this_thread() const noexcept;
-
-
- friend bool operator==(const basic_executor_type& a,
- const basic_executor_type& b) noexcept
- {
- return a.pool_ == b.pool_
- && a.allocator_ == b.allocator_
- && a.bits_ == b.bits_;
- }
-
-
- friend bool operator!=(const basic_executor_type& a,
- const basic_executor_type& b) noexcept
- {
- return a.pool_ != b.pool_
- || a.allocator_ != b.allocator_
- || a.bits_ != b.bits_;
- }
-
- template <typename Function>
- void execute(Function&& f) const
- {
- this->do_execute(static_cast<Function&&>(f),
- integral_constant<bool, (Bits & blocking_always) != 0>());
- }
- public:
- #if !defined(BOOST_ASIO_NO_TS_EXECUTORS)
-
- thread_pool& context() const noexcept;
-
-
- void on_work_started() const noexcept;
-
-
- void on_work_finished() const noexcept;
-
-
- template <typename Function, typename OtherAllocator>
- void dispatch(Function&& f, const OtherAllocator& a) const;
-
-
- template <typename Function, typename OtherAllocator>
- void post(Function&& f, const OtherAllocator& a) const;
-
-
- template <typename Function, typename OtherAllocator>
- void defer(Function&& f, const OtherAllocator& a) const;
- #endif
- private:
- friend class thread_pool;
- template <typename, unsigned int> friend class basic_executor_type;
-
- explicit basic_executor_type(thread_pool& p) noexcept
- : pool_(&p),
- allocator_(),
- bits_(0)
- {
- if (Bits & outstanding_work_tracked)
- pool_->scheduler_.work_started();
- }
-
- basic_executor_type(thread_pool* p,
- const Allocator& a, unsigned int bits) noexcept
- : pool_(p),
- allocator_(a),
- bits_(bits)
- {
- if (Bits & outstanding_work_tracked)
- if (pool_)
- pool_->scheduler_.work_started();
- }
-
- template <typename Function>
- void do_execute(Function&& f, false_type) const;
-
- template <typename Function>
- void do_execute(Function&& f, true_type) const;
-
- thread_pool* pool_;
-
- Allocator allocator_;
-
- unsigned int bits_;
- };
- #if !defined(GENERATING_DOCUMENTATION)
- namespace traits {
- #if !defined(BOOST_ASIO_HAS_DEDUCED_EQUALITY_COMPARABLE_TRAIT)
- template <typename Allocator, unsigned int Bits>
- struct equality_comparable<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- };
- #endif
- #if !defined(BOOST_ASIO_HAS_DEDUCED_EXECUTE_MEMBER_TRAIT)
- template <typename Allocator, unsigned int Bits, typename Function>
- struct execute_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- Function
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef void result_type;
- };
- #endif
- #if !defined(BOOST_ASIO_HAS_DEDUCED_REQUIRE_MEMBER_TRAIT)
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::blocking_t::possibly_t
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits & ~blocking_mask> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::blocking_t::always_t
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<Allocator,
- (Bits & ~blocking_mask) | blocking_always> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::blocking_t::never_t
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits & ~blocking_mask> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::relationship_t::fork_t
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::relationship_t::continuation_t
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::outstanding_work_t::tracked_t
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits | outstanding_work_tracked> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::outstanding_work_t::untracked_t
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- Allocator, Bits & ~outstanding_work_tracked> result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::allocator_t<void>
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- std::allocator<void>, Bits> result_type;
- };
- template <unsigned int Bits,
- typename Allocator, typename OtherAllocator>
- struct require_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::allocator_t<OtherAllocator>
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = false;
- typedef boost::asio::thread_pool::basic_executor_type<
- OtherAllocator, Bits> result_type;
- };
- #endif
- #if !defined(BOOST_ASIO_HAS_DEDUCED_QUERY_STATIC_CONSTEXPR_MEMBER_TRAIT)
- template <typename Allocator, unsigned int Bits, typename Property>
- struct query_static_constexpr_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- Property,
- typename boost::asio::enable_if<
- boost::asio::is_convertible<
- Property,
- boost::asio::execution::outstanding_work_t
- >::value
- >::type
- > : boost::asio::detail::thread_pool_bits
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::execution::outstanding_work_t result_type;
- static constexpr result_type value() noexcept
- {
- return (Bits & outstanding_work_tracked)
- ? execution::outstanding_work_t(execution::outstanding_work.tracked)
- : execution::outstanding_work_t(execution::outstanding_work.untracked);
- }
- };
- template <typename Allocator, unsigned int Bits, typename Property>
- struct query_static_constexpr_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- Property,
- typename boost::asio::enable_if<
- boost::asio::is_convertible<
- Property,
- boost::asio::execution::mapping_t
- >::value
- >::type
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::execution::mapping_t::thread_t result_type;
- static constexpr result_type value() noexcept
- {
- return result_type();
- }
- };
- #endif
- #if !defined(BOOST_ASIO_HAS_DEDUCED_QUERY_MEMBER_TRAIT)
- template <typename Allocator, unsigned int Bits, typename Property>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- Property,
- typename boost::asio::enable_if<
- boost::asio::is_convertible<
- Property,
- boost::asio::execution::blocking_t
- >::value
- >::type
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::execution::blocking_t result_type;
- };
- template <typename Allocator, unsigned int Bits, typename Property>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- Property,
- typename boost::asio::enable_if<
- boost::asio::is_convertible<
- Property,
- boost::asio::execution::relationship_t
- >::value
- >::type
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::execution::relationship_t result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::occupancy_t
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef std::size_t result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::context_t
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef boost::asio::thread_pool& result_type;
- };
- template <typename Allocator, unsigned int Bits>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::allocator_t<void>
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef Allocator result_type;
- };
- template <typename Allocator, unsigned int Bits, typename OtherAllocator>
- struct query_member<
- boost::asio::thread_pool::basic_executor_type<Allocator, Bits>,
- boost::asio::execution::allocator_t<OtherAllocator>
- >
- {
- static constexpr bool is_valid = true;
- static constexpr bool is_noexcept = true;
- typedef Allocator result_type;
- };
- #endif
- }
- namespace execution {
- template <>
- struct is_executor<thread_pool> : false_type
- {
- };
- }
- #endif
- }
- }
- #include <boost/asio/detail/pop_options.hpp>
- #include <boost/asio/impl/thread_pool.hpp>
- #if defined(BOOST_ASIO_HEADER_ONLY)
- # include <boost/asio/impl/thread_pool.ipp>
- #endif
- #endif
|