io_context_strand.hpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387
  1. //
  2. // io_context_strand.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
  6. //
  7. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. #ifndef BOOST_ASIO_IO_CONTEXT_STRAND_HPP
  11. #define BOOST_ASIO_IO_CONTEXT_STRAND_HPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include <boost/asio/detail/config.hpp>
  16. #if !defined(BOOST_ASIO_NO_EXTENSIONS) \
  17. && !defined(BOOST_ASIO_NO_TS_EXECUTORS)
  18. #include <boost/asio/async_result.hpp>
  19. #include <boost/asio/detail/handler_type_requirements.hpp>
  20. #include <boost/asio/detail/strand_service.hpp>
  21. #include <boost/asio/detail/wrapped_handler.hpp>
  22. #include <boost/asio/io_context.hpp>
  23. #include <boost/asio/detail/push_options.hpp>
  24. namespace boost {
  25. namespace asio {
  26. /// Provides serialised handler execution.
  27. /**
  28. * The io_context::strand class provides the ability to post and dispatch
  29. * handlers with the guarantee that none of those handlers will execute
  30. * concurrently.
  31. *
  32. * @par Order of handler invocation
  33. * Given:
  34. *
  35. * @li a strand object @c s
  36. *
  37. * @li an object @c a meeting completion handler requirements
  38. *
  39. * @li an object @c a1 which is an arbitrary copy of @c a made by the
  40. * implementation
  41. *
  42. * @li an object @c b meeting completion handler requirements
  43. *
  44. * @li an object @c b1 which is an arbitrary copy of @c b made by the
  45. * implementation
  46. *
  47. * if any of the following conditions are true:
  48. *
  49. * @li @c s.post(a) happens-before @c s.post(b)
  50. *
  51. * @li @c s.post(a) happens-before @c s.dispatch(b), where the latter is
  52. * performed outside the strand
  53. *
  54. * @li @c s.dispatch(a) happens-before @c s.post(b), where the former is
  55. * performed outside the strand
  56. *
  57. * @li @c s.dispatch(a) happens-before @c s.dispatch(b), where both are
  58. * performed outside the strand
  59. *
  60. * then @c a() happens-before @c b()
  61. *
  62. * Note that in the following case:
  63. * @code async_op_1(..., s.wrap(a));
  64. * async_op_2(..., s.wrap(b)); @endcode
  65. * the completion of the first async operation will perform @c s.dispatch(a),
  66. * and the second will perform @c s.dispatch(b), but the order in which those
  67. * are performed is unspecified. That is, you cannot state whether one
  68. * happens-before the other. Therefore none of the above conditions are met and
  69. * no ordering guarantee is made.
  70. *
  71. * @note The implementation makes no guarantee that handlers posted or
  72. * dispatched through different @c strand objects will be invoked concurrently.
  73. *
  74. * @par Thread Safety
  75. * @e Distinct @e objects: Safe.@n
  76. * @e Shared @e objects: Safe.
  77. *
  78. * @par Concepts:
  79. * Dispatcher.
  80. */
  81. class io_context::strand
  82. {
  83. private:
  84. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  85. struct initiate_dispatch;
  86. struct initiate_post;
  87. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  88. public:
  89. /// Constructor.
  90. /**
  91. * Constructs the strand.
  92. *
  93. * @param io_context The io_context object that the strand will use to
  94. * dispatch handlers that are ready to be run.
  95. */
  96. explicit strand(boost::asio::io_context& io_context)
  97. : service_(boost::asio::use_service<
  98. boost::asio::detail::strand_service>(io_context))
  99. {
  100. service_.construct(impl_);
  101. }
  102. /// Destructor.
  103. /**
  104. * Destroys a strand.
  105. *
  106. * Handlers posted through the strand that have not yet been invoked will
  107. * still be dispatched in a way that meets the guarantee of non-concurrency.
  108. */
  109. ~strand()
  110. {
  111. }
  112. /// Obtain the underlying execution context.
  113. boost::asio::io_context& context() const noexcept
  114. {
  115. return service_.get_io_context();
  116. }
  117. /// Inform the strand that it has some outstanding work to do.
  118. /**
  119. * The strand delegates this call to its underlying io_context.
  120. */
  121. void on_work_started() const noexcept
  122. {
  123. context().get_executor().on_work_started();
  124. }
  125. /// Inform the strand that some work is no longer outstanding.
  126. /**
  127. * The strand delegates this call to its underlying io_context.
  128. */
  129. void on_work_finished() const noexcept
  130. {
  131. context().get_executor().on_work_finished();
  132. }
  133. /// Request the strand to invoke the given function object.
  134. /**
  135. * This function is used to ask the strand to execute the given function
  136. * object on its underlying io_context. The function object will be executed
  137. * inside this function if the strand is not otherwise busy and if the
  138. * underlying io_context's executor's @c dispatch() function is also able to
  139. * execute the function before returning.
  140. *
  141. * @param f The function object to be called. The executor will make
  142. * a copy of the handler object as required. The function signature of the
  143. * function object must be: @code void function(); @endcode
  144. *
  145. * @param a An allocator that may be used by the executor to allocate the
  146. * internal storage needed for function invocation.
  147. */
  148. template <typename Function, typename Allocator>
  149. void dispatch(Function&& f, const Allocator& a) const
  150. {
  151. decay_t<Function> tmp(static_cast<Function&&>(f));
  152. service_.dispatch(impl_, tmp);
  153. (void)a;
  154. }
  155. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  156. /// (Deprecated: Use boost::asio::dispatch().) Request the strand to invoke
  157. /// the given handler.
  158. /**
  159. * This function is used to ask the strand to execute the given handler.
  160. *
  161. * The strand object guarantees that handlers posted or dispatched through
  162. * the strand will not be executed concurrently. The handler may be executed
  163. * inside this function if the guarantee can be met. If this function is
  164. * called from within a handler that was posted or dispatched through the same
  165. * strand, then the new handler will be executed immediately.
  166. *
  167. * The strand's guarantee is in addition to the guarantee provided by the
  168. * underlying io_context. The io_context guarantees that the handler will only
  169. * be called in a thread in which the io_context's run member function is
  170. * currently being invoked.
  171. *
  172. * @param handler The handler to be called. The strand will make a copy of the
  173. * handler object as required. The function signature of the handler must be:
  174. * @code void handler(); @endcode
  175. */
  176. template <typename LegacyCompletionHandler>
  177. auto dispatch(LegacyCompletionHandler&& handler)
  178. -> decltype(
  179. async_initiate<LegacyCompletionHandler, void ()>(
  180. declval<initiate_dispatch>(), handler, this))
  181. {
  182. return async_initiate<LegacyCompletionHandler, void ()>(
  183. initiate_dispatch(), handler, this);
  184. }
  185. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  186. /// Request the strand to invoke the given function object.
  187. /**
  188. * This function is used to ask the executor to execute the given function
  189. * object. The function object will never be executed inside this function.
  190. * Instead, it will be scheduled to run by the underlying io_context.
  191. *
  192. * @param f The function object to be called. The executor will make
  193. * a copy of the handler object as required. The function signature of the
  194. * function object must be: @code void function(); @endcode
  195. *
  196. * @param a An allocator that may be used by the executor to allocate the
  197. * internal storage needed for function invocation.
  198. */
  199. template <typename Function, typename Allocator>
  200. void post(Function&& f, const Allocator& a) const
  201. {
  202. decay_t<Function> tmp(static_cast<Function&&>(f));
  203. service_.post(impl_, tmp);
  204. (void)a;
  205. }
  206. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  207. /// (Deprecated: Use boost::asio::post().) Request the strand to invoke the
  208. /// given handler and return immediately.
  209. /**
  210. * This function is used to ask the strand to execute the given handler, but
  211. * without allowing the strand to call the handler from inside this function.
  212. *
  213. * The strand object guarantees that handlers posted or dispatched through
  214. * the strand will not be executed concurrently. The strand's guarantee is in
  215. * addition to the guarantee provided by the underlying io_context. The
  216. * io_context guarantees that the handler will only be called in a thread in
  217. * which the io_context's run member function is currently being invoked.
  218. *
  219. * @param handler The handler to be called. The strand will make a copy of the
  220. * handler object as required. The function signature of the handler must be:
  221. * @code void handler(); @endcode
  222. */
  223. template <typename LegacyCompletionHandler>
  224. auto post(LegacyCompletionHandler&& handler)
  225. -> decltype(
  226. async_initiate<LegacyCompletionHandler, void ()>(
  227. declval<initiate_post>(), handler, this))
  228. {
  229. return async_initiate<LegacyCompletionHandler, void ()>(
  230. initiate_post(), handler, this);
  231. }
  232. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  233. /// Request the strand to invoke the given function object.
  234. /**
  235. * This function is used to ask the executor to execute the given function
  236. * object. The function object will never be executed inside this function.
  237. * Instead, it will be scheduled to run by the underlying io_context.
  238. *
  239. * @param f The function object to be called. The executor will make
  240. * a copy of the handler object as required. The function signature of the
  241. * function object must be: @code void function(); @endcode
  242. *
  243. * @param a An allocator that may be used by the executor to allocate the
  244. * internal storage needed for function invocation.
  245. */
  246. template <typename Function, typename Allocator>
  247. void defer(Function&& f, const Allocator& a) const
  248. {
  249. decay_t<Function> tmp(static_cast<Function&&>(f));
  250. service_.post(impl_, tmp);
  251. (void)a;
  252. }
  253. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  254. /// (Deprecated: Use boost::asio::bind_executor().) Create a new handler that
  255. /// automatically dispatches the wrapped handler on the strand.
  256. /**
  257. * This function is used to create a new handler function object that, when
  258. * invoked, will automatically pass the wrapped handler to the strand's
  259. * dispatch function.
  260. *
  261. * @param handler The handler to be wrapped. The strand will make a copy of
  262. * the handler object as required. The function signature of the handler must
  263. * be: @code void handler(A1 a1, ... An an); @endcode
  264. *
  265. * @return A function object that, when invoked, passes the wrapped handler to
  266. * the strand's dispatch function. Given a function object with the signature:
  267. * @code R f(A1 a1, ... An an); @endcode
  268. * If this function object is passed to the wrap function like so:
  269. * @code strand.wrap(f); @endcode
  270. * then the return value is a function object with the signature
  271. * @code void g(A1 a1, ... An an); @endcode
  272. * that, when invoked, executes code equivalent to:
  273. * @code strand.dispatch(boost::bind(f, a1, ... an)); @endcode
  274. */
  275. template <typename Handler>
  276. #if defined(GENERATING_DOCUMENTATION)
  277. unspecified
  278. #else
  279. detail::wrapped_handler<strand, Handler, detail::is_continuation_if_running>
  280. #endif
  281. wrap(Handler handler)
  282. {
  283. return detail::wrapped_handler<io_context::strand, Handler,
  284. detail::is_continuation_if_running>(*this, handler);
  285. }
  286. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  287. /// Determine whether the strand is running in the current thread.
  288. /**
  289. * @return @c true if the current thread is executing a handler that was
  290. * submitted to the strand using post(), dispatch() or wrap(). Otherwise
  291. * returns @c false.
  292. */
  293. bool running_in_this_thread() const noexcept
  294. {
  295. return service_.running_in_this_thread(impl_);
  296. }
  297. /// Compare two strands for equality.
  298. /**
  299. * Two strands are equal if they refer to the same ordered, non-concurrent
  300. * state.
  301. */
  302. friend bool operator==(const strand& a, const strand& b) noexcept
  303. {
  304. return a.impl_ == b.impl_;
  305. }
  306. /// Compare two strands for inequality.
  307. /**
  308. * Two strands are equal if they refer to the same ordered, non-concurrent
  309. * state.
  310. */
  311. friend bool operator!=(const strand& a, const strand& b) noexcept
  312. {
  313. return a.impl_ != b.impl_;
  314. }
  315. private:
  316. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  317. struct initiate_dispatch
  318. {
  319. template <typename LegacyCompletionHandler>
  320. void operator()(LegacyCompletionHandler&& handler,
  321. strand* self) const
  322. {
  323. // If you get an error on the following line it means that your
  324. // handler does not meet the documented type requirements for a
  325. // LegacyCompletionHandler.
  326. BOOST_ASIO_LEGACY_COMPLETION_HANDLER_CHECK(
  327. LegacyCompletionHandler, handler) type_check;
  328. detail::non_const_lvalue<LegacyCompletionHandler> handler2(handler);
  329. self->service_.dispatch(self->impl_, handler2.value);
  330. }
  331. };
  332. struct initiate_post
  333. {
  334. template <typename LegacyCompletionHandler>
  335. void operator()(LegacyCompletionHandler&& handler,
  336. strand* self) const
  337. {
  338. // If you get an error on the following line it means that your
  339. // handler does not meet the documented type requirements for a
  340. // LegacyCompletionHandler.
  341. BOOST_ASIO_LEGACY_COMPLETION_HANDLER_CHECK(
  342. LegacyCompletionHandler, handler) type_check;
  343. detail::non_const_lvalue<LegacyCompletionHandler> handler2(handler);
  344. self->service_.post(self->impl_, handler2.value);
  345. }
  346. };
  347. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  348. boost::asio::detail::strand_service& service_;
  349. mutable boost::asio::detail::strand_service::implementation_type impl_;
  350. };
  351. } // namespace asio
  352. } // namespace boost
  353. #include <boost/asio/detail/pop_options.hpp>
  354. #endif // !defined(BOOST_ASIO_NO_EXTENSIONS)
  355. // && !defined(BOOST_ASIO_NO_TS_EXECUTORS)
  356. #endif // BOOST_ASIO_IO_CONTEXT_STRAND_HPP