io_context_strand.hpp 13 KB

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