io_context_strand.hpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398
  1. //
  2. // io_context_strand.hpp
  3. // ~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2024 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. /// Copy constructor.
  103. /**
  104. * Creates a copy such that both strand objects share the same underlying
  105. * state.
  106. */
  107. strand(const strand& other) noexcept
  108. : service_(other.service_),
  109. impl_(other.impl_)
  110. {
  111. }
  112. /// Destructor.
  113. /**
  114. * Destroys a strand.
  115. *
  116. * Handlers posted through the strand that have not yet been invoked will
  117. * still be dispatched in a way that meets the guarantee of non-concurrency.
  118. */
  119. ~strand()
  120. {
  121. }
  122. /// Obtain the underlying execution context.
  123. boost::asio::io_context& context() const noexcept
  124. {
  125. return service_.get_io_context();
  126. }
  127. /// Inform the strand that it has some outstanding work to do.
  128. /**
  129. * The strand delegates this call to its underlying io_context.
  130. */
  131. void on_work_started() const noexcept
  132. {
  133. context().get_executor().on_work_started();
  134. }
  135. /// Inform the strand that some work is no longer outstanding.
  136. /**
  137. * The strand delegates this call to its underlying io_context.
  138. */
  139. void on_work_finished() const noexcept
  140. {
  141. context().get_executor().on_work_finished();
  142. }
  143. /// Request the strand to invoke the given function object.
  144. /**
  145. * This function is used to ask the strand to execute the given function
  146. * object on its underlying io_context. The function object will be executed
  147. * inside this function if the strand is not otherwise busy and if the
  148. * underlying io_context's executor's @c dispatch() function is also able to
  149. * execute the function before returning.
  150. *
  151. * @param f The function object to be called. The executor will make
  152. * a copy of the handler object as required. The function signature of the
  153. * function object must be: @code void function(); @endcode
  154. *
  155. * @param a An allocator that may be used by the executor to allocate the
  156. * internal storage needed for function invocation.
  157. */
  158. template <typename Function, typename Allocator>
  159. void dispatch(Function&& f, const Allocator& a) const
  160. {
  161. decay_t<Function> tmp(static_cast<Function&&>(f));
  162. service_.dispatch(impl_, tmp);
  163. (void)a;
  164. }
  165. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  166. /// (Deprecated: Use boost::asio::dispatch().) Request the strand to invoke
  167. /// the given handler.
  168. /**
  169. * This function is used to ask the strand to execute the given handler.
  170. *
  171. * The strand object guarantees that handlers posted or dispatched through
  172. * the strand will not be executed concurrently. The handler may be executed
  173. * inside this function if the guarantee can be met. If this function is
  174. * called from within a handler that was posted or dispatched through the same
  175. * strand, then the new handler will be executed immediately.
  176. *
  177. * The strand's guarantee is in addition to the guarantee provided by the
  178. * underlying io_context. The io_context guarantees that the handler will only
  179. * be called in a thread in which the io_context's run member function is
  180. * currently being invoked.
  181. *
  182. * @param handler The handler to be called. The strand will make a copy of the
  183. * handler object as required. The function signature of the handler must be:
  184. * @code void handler(); @endcode
  185. */
  186. template <typename LegacyCompletionHandler>
  187. auto dispatch(LegacyCompletionHandler&& handler)
  188. -> decltype(
  189. async_initiate<LegacyCompletionHandler, void ()>(
  190. declval<initiate_dispatch>(), handler, this))
  191. {
  192. return async_initiate<LegacyCompletionHandler, void ()>(
  193. initiate_dispatch(), handler, this);
  194. }
  195. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  196. /// Request the strand to invoke the given function object.
  197. /**
  198. * This function is used to ask the executor to execute the given function
  199. * object. The function object will never be executed inside this function.
  200. * Instead, it will be scheduled to run by the underlying io_context.
  201. *
  202. * @param f The function object to be called. The executor will make
  203. * a copy of the handler object as required. The function signature of the
  204. * function object must be: @code void function(); @endcode
  205. *
  206. * @param a An allocator that may be used by the executor to allocate the
  207. * internal storage needed for function invocation.
  208. */
  209. template <typename Function, typename Allocator>
  210. void post(Function&& f, const Allocator& a) const
  211. {
  212. decay_t<Function> tmp(static_cast<Function&&>(f));
  213. service_.post(impl_, tmp);
  214. (void)a;
  215. }
  216. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  217. /// (Deprecated: Use boost::asio::post().) Request the strand to invoke the
  218. /// given handler and return immediately.
  219. /**
  220. * This function is used to ask the strand to execute the given handler, but
  221. * without allowing the strand to call the handler from inside this function.
  222. *
  223. * The strand object guarantees that handlers posted or dispatched through
  224. * the strand will not be executed concurrently. The strand's guarantee is in
  225. * addition to the guarantee provided by the underlying io_context. The
  226. * io_context guarantees that the handler will only be called in a thread in
  227. * which the io_context's run member function is currently being invoked.
  228. *
  229. * @param handler The handler to be called. The strand will make a copy of the
  230. * handler object as required. The function signature of the handler must be:
  231. * @code void handler(); @endcode
  232. */
  233. template <typename LegacyCompletionHandler>
  234. auto post(LegacyCompletionHandler&& handler)
  235. -> decltype(
  236. async_initiate<LegacyCompletionHandler, void ()>(
  237. declval<initiate_post>(), handler, this))
  238. {
  239. return async_initiate<LegacyCompletionHandler, void ()>(
  240. initiate_post(), handler, this);
  241. }
  242. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  243. /// Request the strand to invoke the given function object.
  244. /**
  245. * This function is used to ask the executor to execute the given function
  246. * object. The function object will never be executed inside this function.
  247. * Instead, it will be scheduled to run by the underlying io_context.
  248. *
  249. * @param f The function object to be called. The executor will make
  250. * a copy of the handler object as required. The function signature of the
  251. * function object must be: @code void function(); @endcode
  252. *
  253. * @param a An allocator that may be used by the executor to allocate the
  254. * internal storage needed for function invocation.
  255. */
  256. template <typename Function, typename Allocator>
  257. void defer(Function&& f, const Allocator& a) const
  258. {
  259. decay_t<Function> tmp(static_cast<Function&&>(f));
  260. service_.post(impl_, tmp);
  261. (void)a;
  262. }
  263. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  264. /// (Deprecated: Use boost::asio::bind_executor().) Create a new handler that
  265. /// automatically dispatches the wrapped handler on the strand.
  266. /**
  267. * This function is used to create a new handler function object that, when
  268. * invoked, will automatically pass the wrapped handler to the strand's
  269. * dispatch function.
  270. *
  271. * @param handler The handler to be wrapped. The strand will make a copy of
  272. * the handler object as required. The function signature of the handler must
  273. * be: @code void handler(A1 a1, ... An an); @endcode
  274. *
  275. * @return A function object that, when invoked, passes the wrapped handler to
  276. * the strand's dispatch function. Given a function object with the signature:
  277. * @code R f(A1 a1, ... An an); @endcode
  278. * If this function object is passed to the wrap function like so:
  279. * @code strand.wrap(f); @endcode
  280. * then the return value is a function object with the signature
  281. * @code void g(A1 a1, ... An an); @endcode
  282. * that, when invoked, executes code equivalent to:
  283. * @code strand.dispatch(boost::bind(f, a1, ... an)); @endcode
  284. */
  285. template <typename Handler>
  286. #if defined(GENERATING_DOCUMENTATION)
  287. unspecified
  288. #else
  289. detail::wrapped_handler<strand, Handler, detail::is_continuation_if_running>
  290. #endif
  291. wrap(Handler handler)
  292. {
  293. return detail::wrapped_handler<io_context::strand, Handler,
  294. detail::is_continuation_if_running>(*this, handler);
  295. }
  296. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  297. /// Determine whether the strand is running in the current thread.
  298. /**
  299. * @return @c true if the current thread is executing a handler that was
  300. * submitted to the strand using post(), dispatch() or wrap(). Otherwise
  301. * returns @c false.
  302. */
  303. bool running_in_this_thread() const noexcept
  304. {
  305. return service_.running_in_this_thread(impl_);
  306. }
  307. /// Compare two strands for equality.
  308. /**
  309. * Two strands are equal if they refer to the same ordered, non-concurrent
  310. * state.
  311. */
  312. friend bool operator==(const strand& a, const strand& b) noexcept
  313. {
  314. return a.impl_ == b.impl_;
  315. }
  316. /// Compare two strands for inequality.
  317. /**
  318. * Two strands are equal if they refer to the same ordered, non-concurrent
  319. * state.
  320. */
  321. friend bool operator!=(const strand& a, const strand& b) noexcept
  322. {
  323. return a.impl_ != b.impl_;
  324. }
  325. private:
  326. #if !defined(BOOST_ASIO_NO_DEPRECATED)
  327. struct initiate_dispatch
  328. {
  329. template <typename LegacyCompletionHandler>
  330. void operator()(LegacyCompletionHandler&& handler,
  331. strand* self) const
  332. {
  333. // If you get an error on the following line it means that your
  334. // handler does not meet the documented type requirements for a
  335. // LegacyCompletionHandler.
  336. BOOST_ASIO_LEGACY_COMPLETION_HANDLER_CHECK(
  337. LegacyCompletionHandler, handler) type_check;
  338. detail::non_const_lvalue<LegacyCompletionHandler> handler2(handler);
  339. self->service_.dispatch(self->impl_, handler2.value);
  340. }
  341. };
  342. struct initiate_post
  343. {
  344. template <typename LegacyCompletionHandler>
  345. void operator()(LegacyCompletionHandler&& handler,
  346. strand* self) const
  347. {
  348. // If you get an error on the following line it means that your
  349. // handler does not meet the documented type requirements for a
  350. // LegacyCompletionHandler.
  351. BOOST_ASIO_LEGACY_COMPLETION_HANDLER_CHECK(
  352. LegacyCompletionHandler, handler) type_check;
  353. detail::non_const_lvalue<LegacyCompletionHandler> handler2(handler);
  354. self->service_.post(self->impl_, handler2.value);
  355. }
  356. };
  357. #endif // !defined(BOOST_ASIO_NO_DEPRECATED)
  358. boost::asio::detail::strand_service& service_;
  359. mutable boost::asio::detail::strand_service::implementation_type impl_;
  360. };
  361. } // namespace asio
  362. } // namespace boost
  363. #include <boost/asio/detail/pop_options.hpp>
  364. #endif // !defined(BOOST_ASIO_NO_EXTENSIONS)
  365. // && !defined(BOOST_ASIO_NO_TS_EXECUTORS)
  366. #endif // BOOST_ASIO_IO_CONTEXT_STRAND_HPP