execution_context.hpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414
  1. //
  2. // execution_context.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_EXECUTION_CONTEXT_HPP
  11. #define BOOST_ASIO_EXECUTION_CONTEXT_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. #include <cstddef>
  17. #include <stdexcept>
  18. #include <typeinfo>
  19. #include <boost/asio/detail/noncopyable.hpp>
  20. #include <boost/asio/detail/variadic_templates.hpp>
  21. #include <boost/asio/detail/push_options.hpp>
  22. namespace boost {
  23. namespace asio {
  24. class execution_context;
  25. class io_context;
  26. #if !defined(GENERATING_DOCUMENTATION)
  27. template <typename Service> Service& use_service(execution_context&);
  28. template <typename Service> Service& use_service(io_context&);
  29. template <typename Service> void add_service(execution_context&, Service*);
  30. template <typename Service> bool has_service(execution_context&);
  31. #endif // !defined(GENERATING_DOCUMENTATION)
  32. namespace detail { class service_registry; }
  33. /// A context for function object execution.
  34. /**
  35. * An execution context represents a place where function objects will be
  36. * executed. An @c io_context is an example of an execution context.
  37. *
  38. * @par The execution_context class and services
  39. *
  40. * Class execution_context implements an extensible, type-safe, polymorphic set
  41. * of services, indexed by service type.
  42. *
  43. * Services exist to manage the resources that are shared across an execution
  44. * context. For example, timers may be implemented in terms of a single timer
  45. * queue, and this queue would be stored in a service.
  46. *
  47. * Access to the services of an execution_context is via three function
  48. * templates, use_service(), add_service() and has_service().
  49. *
  50. * In a call to @c use_service<Service>(), the type argument chooses a service,
  51. * making available all members of the named type. If @c Service is not present
  52. * in an execution_context, an object of type @c Service is created and added
  53. * to the execution_context. A C++ program can check if an execution_context
  54. * implements a particular service with the function template @c
  55. * has_service<Service>().
  56. *
  57. * Service objects may be explicitly added to an execution_context using the
  58. * function template @c add_service<Service>(). If the @c Service is already
  59. * present, the service_already_exists exception is thrown. If the owner of the
  60. * service is not the same object as the execution_context parameter, the
  61. * invalid_service_owner exception is thrown.
  62. *
  63. * Once a service reference is obtained from an execution_context object by
  64. * calling use_service(), that reference remains usable as long as the owning
  65. * execution_context object exists.
  66. *
  67. * All service implementations have execution_context::service as a public base
  68. * class. Custom services may be implemented by deriving from this class and
  69. * then added to an execution_context using the facilities described above.
  70. *
  71. * @par The execution_context as a base class
  72. *
  73. * Class execution_context may be used only as a base class for concrete
  74. * execution context types. The @c io_context is an example of such a derived
  75. * type.
  76. *
  77. * On destruction, a class that is derived from execution_context must perform
  78. * <tt>execution_context::shutdown()</tt> followed by
  79. * <tt>execution_context::destroy()</tt>.
  80. *
  81. * This destruction sequence permits programs to simplify their resource
  82. * management by using @c shared_ptr<>. Where an object's lifetime is tied to
  83. * the lifetime of a connection (or some other sequence of asynchronous
  84. * operations), a @c shared_ptr to the object would be bound into the handlers
  85. * for all asynchronous operations associated with it. This works as follows:
  86. *
  87. * @li When a single connection ends, all associated asynchronous operations
  88. * complete. The corresponding handler objects are destroyed, and all @c
  89. * shared_ptr references to the objects are destroyed.
  90. *
  91. * @li To shut down the whole program, the io_context function stop() is called
  92. * to terminate any run() calls as soon as possible. The io_context destructor
  93. * calls @c shutdown() and @c destroy() to destroy all pending handlers,
  94. * causing all @c shared_ptr references to all connection objects to be
  95. * destroyed.
  96. */
  97. class execution_context
  98. : private noncopyable
  99. {
  100. public:
  101. class id;
  102. class service;
  103. public:
  104. /// Constructor.
  105. BOOST_ASIO_DECL execution_context();
  106. /// Destructor.
  107. BOOST_ASIO_DECL ~execution_context();
  108. protected:
  109. /// Shuts down all services in the context.
  110. /**
  111. * This function is implemented as follows:
  112. *
  113. * @li For each service object @c svc in the execution_context set, in
  114. * reverse order of the beginning of service object lifetime, performs @c
  115. * svc->shutdown().
  116. */
  117. BOOST_ASIO_DECL void shutdown();
  118. /// Destroys all services in the context.
  119. /**
  120. * This function is implemented as follows:
  121. *
  122. * @li For each service object @c svc in the execution_context set, in
  123. * reverse order * of the beginning of service object lifetime, performs
  124. * <tt>delete static_cast<execution_context::service*>(svc)</tt>.
  125. */
  126. BOOST_ASIO_DECL void destroy();
  127. public:
  128. /// Fork-related event notifications.
  129. enum fork_event
  130. {
  131. /// Notify the context that the process is about to fork.
  132. fork_prepare,
  133. /// Notify the context that the process has forked and is the parent.
  134. fork_parent,
  135. /// Notify the context that the process has forked and is the child.
  136. fork_child
  137. };
  138. /// Notify the execution_context of a fork-related event.
  139. /**
  140. * This function is used to inform the execution_context that the process is
  141. * about to fork, or has just forked. This allows the execution_context, and
  142. * the services it contains, to perform any necessary housekeeping to ensure
  143. * correct operation following a fork.
  144. *
  145. * This function must not be called while any other execution_context
  146. * function, or any function associated with the execution_context's derived
  147. * class, is being called in another thread. It is, however, safe to call
  148. * this function from within a completion handler, provided no other thread
  149. * is accessing the execution_context or its derived class.
  150. *
  151. * @param event A fork-related event.
  152. *
  153. * @throws boost::system::system_error Thrown on failure. If the notification
  154. * fails the execution_context object should no longer be used and should be
  155. * destroyed.
  156. *
  157. * @par Example
  158. * The following code illustrates how to incorporate the notify_fork()
  159. * function:
  160. * @code my_execution_context.notify_fork(execution_context::fork_prepare);
  161. * if (fork() == 0)
  162. * {
  163. * // This is the child process.
  164. * my_execution_context.notify_fork(execution_context::fork_child);
  165. * }
  166. * else
  167. * {
  168. * // This is the parent process.
  169. * my_execution_context.notify_fork(execution_context::fork_parent);
  170. * } @endcode
  171. *
  172. * @note For each service object @c svc in the execution_context set,
  173. * performs <tt>svc->notify_fork();</tt>. When processing the fork_prepare
  174. * event, services are visited in reverse order of the beginning of service
  175. * object lifetime. Otherwise, services are visited in order of the beginning
  176. * of service object lifetime.
  177. */
  178. BOOST_ASIO_DECL void notify_fork(fork_event event);
  179. /// Obtain the service object corresponding to the given type.
  180. /**
  181. * This function is used to locate a service object that corresponds to the
  182. * given service type. If there is no existing implementation of the service,
  183. * then the execution_context will create a new instance of the service.
  184. *
  185. * @param e The execution_context object that owns the service.
  186. *
  187. * @return The service interface implementing the specified service type.
  188. * Ownership of the service interface is not transferred to the caller.
  189. */
  190. template <typename Service>
  191. friend Service& use_service(execution_context& e);
  192. /// Obtain the service object corresponding to the given type.
  193. /**
  194. * This function is used to locate a service object that corresponds to the
  195. * given service type. If there is no existing implementation of the service,
  196. * then the io_context will create a new instance of the service.
  197. *
  198. * @param ioc The io_context object that owns the service.
  199. *
  200. * @return The service interface implementing the specified service type.
  201. * Ownership of the service interface is not transferred to the caller.
  202. *
  203. * @note This overload is preserved for backwards compatibility with services
  204. * that inherit from io_context::service.
  205. */
  206. template <typename Service>
  207. friend Service& use_service(io_context& ioc);
  208. #if defined(GENERATING_DOCUMENTATION)
  209. /// Creates a service object and adds it to the execution_context.
  210. /**
  211. * This function is used to add a service to the execution_context.
  212. *
  213. * @param e The execution_context object that owns the service.
  214. *
  215. * @param args Zero or more arguments to be passed to the service
  216. * constructor.
  217. *
  218. * @throws boost::asio::service_already_exists Thrown if a service of the
  219. * given type is already present in the execution_context.
  220. */
  221. template <typename Service, typename... Args>
  222. friend Service& make_service(execution_context& e, Args&&... args);
  223. #elif defined(BOOST_ASIO_HAS_VARIADIC_TEMPLATES)
  224. template <typename Service, typename... Args>
  225. friend Service& make_service(execution_context& e,
  226. BOOST_ASIO_MOVE_ARG(Args)... args);
  227. #else // defined(BOOST_ASIO_HAS_VARIADIC_TEMPLATES)
  228. template <typename Service>
  229. friend Service& make_service(execution_context& e);
  230. #define BOOST_ASIO_PRIVATE_MAKE_SERVICE_DEF(n) \
  231. template <typename Service, BOOST_ASIO_VARIADIC_TPARAMS(n)> \
  232. friend Service& make_service(execution_context& e, \
  233. BOOST_ASIO_VARIADIC_MOVE_PARAMS(n)); \
  234. /**/
  235. BOOST_ASIO_VARIADIC_GENERATE(BOOST_ASIO_PRIVATE_MAKE_SERVICE_DEF)
  236. #undef BOOST_ASIO_PRIVATE_MAKE_SERVICE_DEF
  237. #endif // defined(BOOST_ASIO_HAS_VARIADIC_TEMPLATES)
  238. /// (Deprecated: Use make_service().) Add a service object to the
  239. /// execution_context.
  240. /**
  241. * This function is used to add a service to the execution_context.
  242. *
  243. * @param e The execution_context object that owns the service.
  244. *
  245. * @param svc The service object. On success, ownership of the service object
  246. * is transferred to the execution_context. When the execution_context object
  247. * is destroyed, it will destroy the service object by performing: @code
  248. * delete static_cast<execution_context::service*>(svc) @endcode
  249. *
  250. * @throws boost::asio::service_already_exists Thrown if a service of the
  251. * given type is already present in the execution_context.
  252. *
  253. * @throws boost::asio::invalid_service_owner Thrown if the service's owning
  254. * execution_context is not the execution_context object specified by the
  255. * @c e parameter.
  256. */
  257. template <typename Service>
  258. friend void add_service(execution_context& e, Service* svc);
  259. /// Determine if an execution_context contains a specified service type.
  260. /**
  261. * This function is used to determine whether the execution_context contains a
  262. * service object corresponding to the given service type.
  263. *
  264. * @param e The execution_context object that owns the service.
  265. *
  266. * @return A boolean indicating whether the execution_context contains the
  267. * service.
  268. */
  269. template <typename Service>
  270. friend bool has_service(execution_context& e);
  271. private:
  272. // The service registry.
  273. boost::asio::detail::service_registry* service_registry_;
  274. };
  275. /// Class used to uniquely identify a service.
  276. class execution_context::id
  277. : private noncopyable
  278. {
  279. public:
  280. /// Constructor.
  281. id() {}
  282. };
  283. /// Base class for all io_context services.
  284. class execution_context::service
  285. : private noncopyable
  286. {
  287. public:
  288. /// Get the context object that owns the service.
  289. execution_context& context();
  290. protected:
  291. /// Constructor.
  292. /**
  293. * @param owner The execution_context object that owns the service.
  294. */
  295. BOOST_ASIO_DECL service(execution_context& owner);
  296. /// Destructor.
  297. BOOST_ASIO_DECL virtual ~service();
  298. private:
  299. /// Destroy all user-defined handler objects owned by the service.
  300. virtual void shutdown() = 0;
  301. /// Handle notification of a fork-related event to perform any necessary
  302. /// housekeeping.
  303. /**
  304. * This function is not a pure virtual so that services only have to
  305. * implement it if necessary. The default implementation does nothing.
  306. */
  307. BOOST_ASIO_DECL virtual void notify_fork(
  308. execution_context::fork_event event);
  309. friend class boost::asio::detail::service_registry;
  310. struct key
  311. {
  312. key() : type_info_(0), id_(0) {}
  313. const std::type_info* type_info_;
  314. const execution_context::id* id_;
  315. } key_;
  316. execution_context& owner_;
  317. service* next_;
  318. };
  319. /// Exception thrown when trying to add a duplicate service to an
  320. /// execution_context.
  321. class service_already_exists
  322. : public std::logic_error
  323. {
  324. public:
  325. BOOST_ASIO_DECL service_already_exists();
  326. };
  327. /// Exception thrown when trying to add a service object to an
  328. /// execution_context where the service has a different owner.
  329. class invalid_service_owner
  330. : public std::logic_error
  331. {
  332. public:
  333. BOOST_ASIO_DECL invalid_service_owner();
  334. };
  335. namespace detail {
  336. // Special derived service id type to keep classes header-file only.
  337. template <typename Type>
  338. class service_id
  339. : public execution_context::id
  340. {
  341. };
  342. // Special service base class to keep classes header-file only.
  343. template <typename Type>
  344. class execution_context_service_base
  345. : public execution_context::service
  346. {
  347. public:
  348. static service_id<Type> id;
  349. // Constructor.
  350. execution_context_service_base(execution_context& e)
  351. : execution_context::service(e)
  352. {
  353. }
  354. };
  355. template <typename Type>
  356. service_id<Type> execution_context_service_base<Type>::id;
  357. } // namespace detail
  358. } // namespace asio
  359. } // namespace boost
  360. #include <boost/asio/detail/pop_options.hpp>
  361. #include <boost/asio/impl/execution_context.hpp>
  362. #if defined(BOOST_ASIO_HEADER_ONLY)
  363. # include <boost/asio/impl/execution_context.ipp>
  364. #endif // defined(BOOST_ASIO_HEADER_ONLY)
  365. #endif // BOOST_ASIO_EXECUTION_CONTEXT_HPP