. /include/boost/corosio /detail /scheduler.hpp
71.43% Lines (5/7) 66.67% Functions (2/3)
[ ] prev/next file n p prev/next uncovered line
TLA Baseline Branch
Line Hits Code Line Hits Code
1   // 1   //
2   // Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com) 2   // Copyright (c) 2025 Vinnie Falco (vinnie.falco@gmail.com)
3   // Copyright (c) 2026 Steve Gerbino 3   // Copyright (c) 2026 Steve Gerbino
4   // Copyright (c) 2026 Michael Vandeberg 4   // Copyright (c) 2026 Michael Vandeberg
5   // 5   //
6   // Distributed under the Boost Software License, Version 1.0. (See accompanying 6   // Distributed under the Boost Software License, Version 1.0. (See accompanying
7   // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) 7   // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
8   // 8   //
9   // Official repository: https://github.com/cppalliance/corosio 9   // Official repository: https://github.com/cppalliance/corosio
10   // 10   //
11   11  
12   #ifndef BOOST_COROSIO_DETAIL_SCHEDULER_HPP 12   #ifndef BOOST_COROSIO_DETAIL_SCHEDULER_HPP
13   #define BOOST_COROSIO_DETAIL_SCHEDULER_HPP 13   #define BOOST_COROSIO_DETAIL_SCHEDULER_HPP
14   14  
15   #include <boost/corosio/detail/config.hpp> 15   #include <boost/corosio/detail/config.hpp>
  16 + #include <boost/corosio/detail/except.hpp>
16 - #include <system_error>  
17   17  
18   #include <boost/capy/continuation.hpp> 18   #include <boost/capy/continuation.hpp>
19 - #include <coroutine> 19 + #include <boost/capy/ex/execution_context.hpp>
20   20  
  21 + #include <coroutine>
21   #include <cstddef> 22   #include <cstddef>
  23 + #include <system_error>
22   24  
23   namespace boost::corosio::detail { 25   namespace boost::corosio::detail {
24   26  
25   class scheduler_op; 27   class scheduler_op;
26   28  
27   /** Define the abstract interface for the event loop scheduler. 29   /** Define the abstract interface for the event loop scheduler.
28   30  
29   Concrete backends (epoll, IOCP, kqueue, select) derive from 31   Concrete backends (epoll, IOCP, kqueue, select) derive from
30   this to implement the reactor/proactor event loop. The 32   this to implement the reactor/proactor event loop. The
31   @ref io_context delegates all scheduling operations here. 33   @ref io_context delegates all scheduling operations here.
32   34  
  35 + The scheduler is a registry service keyed under this abstract
  36 + type, so services created on first use can locate it without
  37 + naming a concrete backend.
  38 +
33   @see io_context 39   @see io_context
34   */ 40   */
35   struct BOOST_COROSIO_DECL scheduler 41   struct BOOST_COROSIO_DECL scheduler
  42 + : capy::execution_context::service
36   { 43   {
ECB 37 - 2253 virtual ~scheduler() = default; 44 + using key_type = scheduler;
  45 +
HITGNC   46 + 2255 ~scheduler() override = default;
38   47  
39   /// Post a coroutine handle for deferred execution. 48   /// Post a coroutine handle for deferred execution.
40   virtual void post(std::coroutine_handle<>) const = 0; 49   virtual void post(std::coroutine_handle<>) const = 0;
41   50  
42   /// Post a scheduler operation for deferred execution. 51   /// Post a scheduler operation for deferred execution.
43   virtual void post(scheduler_op*) const = 0; 52   virtual void post(scheduler_op*) const = 0;
44   53  
45   /// Post a continuation for deferred execution (zero-allocation). 54   /// Post a continuation for deferred execution (zero-allocation).
46   virtual void post(capy::continuation&) const = 0; 55   virtual void post(capy::continuation&) const = 0;
47   56  
48   /// Increment the outstanding work count. 57   /// Increment the outstanding work count.
49   virtual void work_started() noexcept = 0; 58   virtual void work_started() noexcept = 0;
50   59  
51   /// Decrement the outstanding work count. 60   /// Decrement the outstanding work count.
52   virtual void work_finished() noexcept = 0; 61   virtual void work_finished() noexcept = 0;
53   62  
54   /// Check if the calling thread is running the event loop. 63   /// Check if the calling thread is running the event loop.
55   virtual bool running_in_this_thread() const noexcept = 0; 64   virtual bool running_in_this_thread() const noexcept = 0;
56   65  
57   /// Signal the event loop to stop. 66   /// Signal the event loop to stop.
58   virtual void stop() = 0; 67   virtual void stop() = 0;
59   68  
60   /// Check if the event loop has been stopped. 69   /// Check if the event loop has been stopped.
61   virtual bool stopped() const noexcept = 0; 70   virtual bool stopped() const noexcept = 0;
62   71  
63   /// Reset the stopped state so `run()` can be called again. 72   /// Reset the stopped state so `run()` can be called again.
64   virtual void restart() = 0; 73   virtual void restart() = 0;
65   74  
66   /// Run the event loop, blocking until all work completes. 75   /// Run the event loop, blocking until all work completes.
67   virtual std::size_t run() = 0; 76   virtual std::size_t run() = 0;
68   77  
69   /// Run one handler, blocking until one completes. 78   /// Run one handler, blocking until one completes.
70   virtual std::size_t run_one() = 0; 79   virtual std::size_t run_one() = 0;
71   80  
72   /** Run one handler, blocking up to @p usec microseconds. 81   /** Run one handler, blocking up to @p usec microseconds.
73   82  
74   @param usec Maximum wait time in microseconds. 83   @param usec Maximum wait time in microseconds.
75   84  
76   @return The number of handlers executed (0 or 1). 85   @return The number of handlers executed (0 or 1).
77   */ 86   */
78   virtual std::size_t wait_one(long usec) = 0; 87   virtual std::size_t wait_one(long usec) = 0;
79   88  
80   /// Run all ready handlers without blocking. 89   /// Run all ready handlers without blocking.
81   virtual std::size_t poll() = 0; 90   virtual std::size_t poll() = 0;
82   91  
83   /// Run at most one ready handler without blocking. 92   /// Run at most one ready handler without blocking.
84   virtual std::size_t poll_one() = 0; 93   virtual std::size_t poll_one() = 0;
85   94  
86   /** Register the read end of the POSIX signal self-pipe. 95   /** Register the read end of the POSIX signal self-pipe.
87   96  
88   Called once (by the first signal_set to register a signal) so the 97   Called once (by the first signal_set to register a signal) so the
89   backend's event loop watches @p read_fd for readability. When the 98   backend's event loop watches @p read_fd for readability. When the
90   pipe becomes readable the backend drains it and calls 99   pipe becomes readable the backend drains it and calls
91   `posix_signal_service::deliver_signal` for each pending signal, in 100   `posix_signal_service::deliver_signal` for each pending signal, in
92   normal thread context. This keeps the C signal handler 101   normal thread context. This keeps the C signal handler
93   async-signal-safe: it only writes the signal number to the pipe. 102   async-signal-safe: it only writes the signal number to the pipe.
94   103  
95   POSIX backends override this; the default is a no-op (Windows/IOCP 104   POSIX backends override this; the default is a no-op (Windows/IOCP
96   uses synchronous C-runtime signal handling instead). 105   uses synchronous C-runtime signal handling instead).
97   106  
98   @param read_fd The read end of the global signal self-pipe. 107   @param read_fd The read end of the global signal self-pipe.
99   108  
100   @return The error code, empty on success. 109   @return The error code, empty on success.
101   */ 110   */
102   [[nodiscard]] virtual std::error_code 111   [[nodiscard]] virtual std::error_code
MISUBC 103   ✗ register_signal_reader([[maybe_unused]] int read_fd) 112   ✗ register_signal_reader([[maybe_unused]] int read_fd)
104   { 113   {
105   return {}; // LCOV_EXCL_LINE POSIX overrides; the only caller never runs on IOCP 114   return {}; // LCOV_EXCL_LINE POSIX overrides; the only caller never runs on IOCP
106   } 115   }
107   116  
108   /// Decomposed threading configuration applied via @ref configure_threading. 117   /// Decomposed threading configuration applied via @ref configure_threading.
109   struct threading_config 118   struct threading_config
110   { 119   {
111   /// Scheduler mutex/condvar enabled. Off only in the `unsafe` tier. 120   /// Scheduler mutex/condvar enabled. Off only in the `unsafe` tier.
112   bool scheduler_locking = true; 121   bool scheduler_locking = true;
113   /// Per-descriptor (reactor) or ring (uring) I/O lock enabled. 122   /// Per-descriptor (reactor) or ring (uring) I/O lock enabled.
114   /// Off in the `unsafe_io` and `unsafe` tiers. 123   /// Off in the `unsafe_io` and `unsafe` tiers.
115   bool reactor_io_locking = true; 124   bool reactor_io_locking = true;
116   /// A single run thread is guaranteed (a lockless tier): elide 125   /// A single run thread is guaranteed (a lockless tier): elide
117   /// inter-run-thread wakeups. 126   /// inter-run-thread wakeups.
118   bool one_thread = false; 127   bool one_thread = false;
119   }; 128   };
120   129  
121   /// True in the fully-lockless (`unsafe`) tier. The resolver and POSIX 130   /// True in the fully-lockless (`unsafe`) tier. The resolver and POSIX
122   /// file services gate their `operation_not_supported` result on this. 131   /// file services gate their `operation_not_supported` result on this.
123   virtual bool scheduler_locking_disabled() const noexcept = 0; 132   virtual bool scheduler_locking_disabled() const noexcept = 0;
124   133  
125   /// Apply @ref threading_config. 134   /// Apply @ref threading_config.
126   virtual void configure_threading(threading_config) noexcept = 0; 135   virtual void configure_threading(threading_config) noexcept = 0;
127   }; 136   };
  137 +
  138 + /** Return the scheduler registered with the context.
  139 +
  140 + @throws std::logic_error If the context has no backend installed.
  141 + */
  142 + inline scheduler&
HITGNC   143 + 600 get_scheduler(capy::execution_context& ctx)
  144 + {
HITGNC   145 + 600 auto* sched = ctx.find_service<scheduler>();
HITGNC   146 + 600 if (!sched)
MISUNC   147 + ✗ throw_logic_error("no scheduler installed");
HITGNC   148 + 600 return *sched;
  149 + }
128   150  
129   } // namespace boost::corosio::detail 151   } // namespace boost::corosio::detail
130   152  
131   #endif 153   #endif