io_context_strand.hpp 13 KB

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