include/boost/corosio/detail/timer_service.hpp
91.1% Lines (369/405)
97.9% List of functions (47/49)
65.6% Branches (118/180)
Functions (49)
Function
Calls
Lines
Branches
Blocks
<unknown function 91>
:91
–
–
–
–
boost::corosio::detail::timer_service::callback::callback()
:96
1358x
100.0%
–
100.0%
boost::corosio::detail::timer_service::callback::callback(void*, void (*)(void*))
:99
1358x
100.0%
–
100.0%
boost::corosio::detail::timer_service::callback::operator()() const
:108
7150x
100.0%
50.0%
100.0%
boost::corosio::detail::timer_service::timer_service(boost::capy::execution_context&, boost::corosio::detail::scheduler&)
:137
1358x
100.0%
–
100.0%
boost::corosio::detail::timer_service::get_scheduler()
:143
14390x
100.0%
–
100.0%
boost::corosio::detail::timer_service::~timer_service()
:149
2037x
100.0%
–
100.0%
boost::corosio::detail::timer_service::set_on_earliest_changed(boost::corosio::detail::timer_service::callback)
:155
679x
100.0%
–
100.0%
boost::corosio::detail::timer_service::nearest_expiry() const
:168
629579x
100.0%
50.0%
75.0%
boost::corosio::detail::timer_service::refresh_cached_nearest()
:211
698955x
100.0%
100.0%
100.0%
boost::corosio::detail::waiter_node::~waiter_node()
:224
494x
100.0%
–
100.0%
boost::corosio::detail::waiter_node::completion_op::~completion_op()
:228
494x
100.0%
–
100.0%
boost::corosio::detail::waiter_node::completion_op::completion_op()
:235
494x
100.0%
–
100.0%
boost::corosio::detail::waiter_node::waiter_node()
:261
494x
100.0%
–
100.0%
boost::corosio::detail::timer_service::implementation::~implementation()
:267
596x
100.0%
–
100.0%
boost::corosio::detail::try_pop_tl_cache(boost::corosio::detail::timer_service*)
:298
7240x
80.0%
50.0%
62.0%
boost::corosio::detail::try_push_tl_cache(boost::corosio::detail::timer_service::implementation*)
:313
7232x
100.0%
100.0%
100.0%
boost::corosio::detail::try_pop_waiter_tl_cache()
:324
7199x
100.0%
100.0%
100.0%
boost::corosio::detail::try_push_waiter_tl_cache(boost::corosio::detail::waiter_node*)
:336
7183x
100.0%
100.0%
100.0%
boost::corosio::detail::timer_service_invalidate_cache()
:347
679x
100.0%
100.0%
100.0%
boost::corosio::detail::timer_service::implementation::implementation(boost::corosio::detail::timer_service&)
:358
596x
100.0%
50.0%
100.0%
boost::corosio::detail::timer_service::shutdown()
:365
679x
100.0%
69.2%
93.0%
boost::corosio::detail::timer_service::construct()
:422
7240x
63.2%
66.7%
70.0%
boost::corosio::detail::timer_service::destroy(boost::corosio::io_object::implementation*)
:451
7238x
100.0%
–
100.0%
boost::corosio::detail::timer_service::destroy_impl(boost::corosio::detail::timer_service::implementation&)
:457
7238x
73.3%
62.5%
63.0%
boost::corosio::detail::timer_service::create_waiter()
:484
7199x
100.0%
83.3%
80.0%
boost::corosio::detail::timer_service::destroy_waiter(boost::corosio::detail::waiter_node*)
:502
7183x
100.0%
100.0%
100.0%
boost::corosio::detail::timer_service::update_timer(boost::corosio::detail::timer_service::implementation&, std::__1::chrono::time_point<std::__1::chrono::steady_clock, std::__1::chrono::duration<long long, std::__1::ratio<1l, 1000000000l>>>)
:513
6x
93.3%
50.0%
73.0%
boost::corosio::detail::timer_service::insert_waiter(boost::corosio::detail::timer_service::implementation&, boost::corosio::detail::waiter_node*)
:563
7199x
100.0%
75.0%
77.0%
boost::corosio::detail::timer_service::cancel_timer(boost::corosio::detail::timer_service::implementation&)
:583
7826x
87.0%
66.7%
75.0%
boost::corosio::detail::timer_service::cancel_waiter(boost::corosio::detail::waiter_node*)
:623
36x
93.8%
66.7%
66.0%
boost::corosio::detail::timer_service::cancel_one_waiter(boost::corosio::detail::timer_service::implementation&)
:646
2x
73.7%
37.5%
50.0%
boost::corosio::detail::timer_service::process_expired()
:673
691124x
95.0%
90.0%
86.0%
boost::corosio::detail::timer_service::remove_timer_impl(boost::corosio::detail::timer_service::implementation&)
:708
7169x
86.7%
50.0%
70.0%
boost::corosio::detail::timer_service::up_heap(unsigned long)
:735
7183x
100.0%
100.0%
100.0%
boost::corosio::detail::timer_service::down_heap(unsigned long)
:748
6412x
61.5%
50.0%
70.0%
boost::corosio::detail::timer_service::swap_heap(unsigned long, unsigned long)
:768
12801x
100.0%
–
100.0%
boost::corosio::detail::waiter_node::canceller::operator()() const
:780
36x
100.0%
–
100.0%
boost::corosio::detail::waiter_node::completion_op::do_complete(void*, boost::corosio::detail::scheduler_op*, unsigned int, unsigned int)
:786
0
0.0%
0.0%
0.0%
boost::corosio::detail::waiter_node::completion_op::operator()()
:800
7183x
100.0%
50.0%
100.0%
boost::corosio::detail::waiter_node::completion_op::destroy()
:825
8x
100.0%
50.0%
100.0%
boost::corosio::detail::timer_service::implementation::wait(std::__1::coroutine_handle<void>, boost::capy::executor_ref, std::__1::stop_token, std::__1::error_code*, boost::capy::continuation*)
:852
7214x
100.0%
80.0%
100.0%
boost::corosio::detail::timer_service_access::get_timer(boost::corosio::io_context&)
:896
7240x
100.0%
–
100.0%
boost::corosio::detail::timer_service_access::set_timer(boost::corosio::io_context&, boost::corosio::detail::timer_service&)
:901
679x
100.0%
–
100.0%
boost::corosio::detail::timer_service_direct(boost::capy::execution_context&)
:909
7240x
100.0%
–
100.0%
boost::corosio::detail::timer_service_update_expiry(boost::corosio::io_timer::implementation&)
:915
6x
100.0%
–
100.0%
boost::corosio::detail::timer_service_cancel(boost::corosio::io_timer::implementation&)
:922
594x
100.0%
50.0%
66.0%
boost::corosio::detail::timer_service_cancel_one(boost::corosio::io_timer::implementation&)
:929
2x
100.0%
50.0%
66.0%
boost::corosio::detail::get_timer_service(boost::capy::execution_context&, boost::corosio::detail::scheduler&)
:936
679x
100.0%
–
100.0%
| Line | Branch | TLA | Hits | Source Code |
|---|---|---|---|---|
| 1 | // | |||
| 2 | // Copyright (c) 2025 Vinnie Falco ([email protected]) | |||
| 3 | // Copyright (c) 2026 Steve Gerbino | |||
| 4 | // | |||
| 5 | // Distributed under the Boost Software License, Version 1.0. (See accompanying | |||
| 6 | // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | |||
| 7 | // | |||
| 8 | // Official repository: https://github.com/cppalliance/corosio | |||
| 9 | // | |||
| 10 | ||||
| 11 | #ifndef BOOST_COROSIO_DETAIL_TIMER_SERVICE_HPP | |||
| 12 | #define BOOST_COROSIO_DETAIL_TIMER_SERVICE_HPP | |||
| 13 | ||||
| 14 | #include <boost/corosio/timer.hpp> | |||
| 15 | #include <boost/corosio/io_context.hpp> | |||
| 16 | #include <boost/corosio/detail/scheduler_op.hpp> | |||
| 17 | #include <boost/corosio/detail/intrusive.hpp> | |||
| 18 | #include <boost/corosio/detail/thread_local_ptr.hpp> | |||
| 19 | #include <boost/capy/error.hpp> | |||
| 20 | #include <boost/capy/ex/execution_context.hpp> | |||
| 21 | #include <boost/capy/ex/executor_ref.hpp> | |||
| 22 | #include <system_error> | |||
| 23 | ||||
| 24 | #include <atomic> | |||
| 25 | #include <chrono> | |||
| 26 | #include <coroutine> | |||
| 27 | #include <cstddef> | |||
| 28 | #include <limits> | |||
| 29 | #include <mutex> | |||
| 30 | #include <optional> | |||
| 31 | #include <stop_token> | |||
| 32 | #include <utility> | |||
| 33 | #include <vector> | |||
| 34 | ||||
| 35 | namespace boost::corosio::detail { | |||
| 36 | ||||
| 37 | struct scheduler; | |||
| 38 | ||||
| 39 | /* | |||
| 40 | Timer Service | |||
| 41 | ============= | |||
| 42 | ||||
| 43 | Data Structures | |||
| 44 | --------------- | |||
| 45 | waiter_node holds per-waiter state: coroutine handle, executor, | |||
| 46 | error output, stop_token, embedded completion_op. Each concurrent | |||
| 47 | co_await t.wait() allocates one waiter_node. | |||
| 48 | ||||
| 49 | timer_service::implementation holds per-timer state: expiry, | |||
| 50 | heap index, and an intrusive_list of waiter_nodes. Multiple | |||
| 51 | coroutines can wait on the same timer simultaneously. | |||
| 52 | ||||
| 53 | timer_service owns a min-heap of active timers, a free list | |||
| 54 | of recycled impls, and a free list of recycled waiter_nodes. The | |||
| 55 | heap is ordered by expiry time; the scheduler queries | |||
| 56 | nearest_expiry() to set the epoll/timerfd timeout. | |||
| 57 | ||||
| 58 | Optimization Strategy | |||
| 59 | --------------------- | |||
| 60 | 1. Deferred heap insertion — expires_after() stores the expiry | |||
| 61 | but does not insert into the heap. Insertion happens in wait(). | |||
| 62 | 2. Thread-local impl cache — single-slot per-thread cache. | |||
| 63 | 3. Embedded completion_op — eliminates heap allocation per fire/cancel. | |||
| 64 | 4. Cached nearest expiry — atomic avoids mutex in nearest_expiry(). | |||
| 65 | 5. might_have_pending_waits_ flag — skips lock when no wait issued. | |||
| 66 | 6. Thread-local waiter cache — single-slot per-thread cache. | |||
| 67 | ||||
| 68 | Concurrency | |||
| 69 | ----------- | |||
| 70 | stop_token callbacks can fire from any thread. The impl_ | |||
| 71 | pointer on waiter_node is used as a "still in list" marker. | |||
| 72 | */ | |||
| 73 | ||||
| 74 | struct BOOST_COROSIO_SYMBOL_VISIBLE waiter_node; | |||
| 75 | ||||
| 76 | inline void timer_service_invalidate_cache() noexcept; | |||
| 77 | ||||
| 78 | // timer_service class body — member function definitions are | |||
| 79 | // out-of-class (after implementation and waiter_node are complete) | |||
| 80 | class BOOST_COROSIO_DECL timer_service final | |||
| 81 | : public capy::execution_context::service | |||
| 82 | , public io_object::io_service | |||
| 83 | { | |||
| 84 | public: | |||
| 85 | using clock_type = std::chrono::steady_clock; | |||
| 86 | using time_point = clock_type::time_point; | |||
| 87 | ||||
| 88 | /// Type-erased callback for earliest-expiry-changed notifications. | |||
| 89 | class callback | |||
| 90 | { | |||
| 91 | 679x | void* ctx_ = nullptr; | ||
| 92 | 679x | void (*fn_)(void*) = nullptr; | ||
| 93 | ||||
| 94 | public: | |||
| 95 | /// Construct an empty callback. | |||
| 96 | 2037x | callback() = default; | ||
| 97 | ||||
| 98 | /// Construct a callback with the given context and function. | |||
| 99 | 1358x | callback(void* ctx, void (*fn)(void*)) noexcept : ctx_(ctx), fn_(fn) {} | ||
| 100 | ||||
| 101 | /// Return true if the callback is non-empty. | |||
| 102 | explicit operator bool() const noexcept | |||
| 103 | { | |||
| 104 | return fn_ != nullptr; | |||
| 105 | } | |||
| 106 | ||||
| 107 | /// Invoke the callback. | |||
| 108 | 7150x | void operator()() const | ||
| 109 | { | |||
| 110 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7150 times.
|
7150x | if (fn_) | |
| 111 | 7150x | fn_(ctx_); | ||
| 112 | 7150x | } | ||
| 113 | }; | |||
| 114 | ||||
| 115 | struct implementation; | |||
| 116 | ||||
| 117 | private: | |||
| 118 | struct heap_entry | |||
| 119 | { | |||
| 120 | time_point time_; | |||
| 121 | implementation* timer_; | |||
| 122 | }; | |||
| 123 | ||||
| 124 | scheduler* sched_ = nullptr; | |||
| 125 | mutable std::mutex mutex_; | |||
| 126 | std::vector<heap_entry> heap_; | |||
| 127 | 679x | implementation* free_list_ = nullptr; | ||
| 128 | 679x | waiter_node* waiter_free_list_ = nullptr; | ||
| 129 | callback on_earliest_changed_; | |||
| 130 | 679x | bool shutting_down_ = false; | ||
| 131 | // Avoids mutex in nearest_expiry() and empty() | |||
| 132 | 1358x | mutable std::atomic<std::int64_t> cached_nearest_ns_{ | ||
| 133 | 679x | (std::numeric_limits<std::int64_t>::max)()}; | ||
| 134 | ||||
| 135 | public: | |||
| 136 | /// Construct the timer service bound to a scheduler. | |||
| 137 | 3395x | inline timer_service(capy::execution_context&, scheduler& sched) | ||
| 138 | 679x | : sched_(&sched) | ||
| 139 | 2037x | { | ||
| 140 | 1358x | } | ||
| 141 | ||||
| 142 | /// Return the associated scheduler. | |||
| 143 | 14390x | inline scheduler& get_scheduler() noexcept | ||
| 144 | { | |||
| 145 | 14390x | return *sched_; | ||
| 146 | } | |||
| 147 | ||||
| 148 | /// Destroy the timer service. | |||
| 149 | 2037x | ~timer_service() override = default; | ||
| 150 | ||||
| 151 | timer_service(timer_service const&) = delete; | |||
| 152 | timer_service& operator=(timer_service const&) = delete; | |||
| 153 | ||||
| 154 | /// Register a callback invoked when the earliest expiry changes. | |||
| 155 | 679x | inline void set_on_earliest_changed(callback cb) | ||
| 156 | { | |||
| 157 | 679x | on_earliest_changed_ = cb; | ||
| 158 | 679x | } | ||
| 159 | ||||
| 160 | /// Return true if no timers are in the heap. | |||
| 161 | inline bool empty() const noexcept | |||
| 162 | { | |||
| 163 | return cached_nearest_ns_.load(std::memory_order_acquire) == | |||
| 164 | (std::numeric_limits<std::int64_t>::max)(); | |||
| 165 | } | |||
| 166 | ||||
| 167 | /// Return the nearest timer expiry without acquiring the mutex. | |||
| 168 | 629579x | inline time_point nearest_expiry() const noexcept | ||
| 169 | { | |||
| 170 | 629579x | auto ns = cached_nearest_ns_.load(std::memory_order_acquire); | ||
| 171 |
2/4✓ Branch 0 taken 629579 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 629579 times.
✗ Branch 3 not taken.
|
629579x | return time_point(time_point::duration(ns)); | |
| 172 | } | |||
| 173 | ||||
| 174 | /// Cancel all pending timers and free cached resources. | |||
| 175 | inline void shutdown() override; | |||
| 176 | ||||
| 177 | /// Construct a new timer implementation. | |||
| 178 | inline io_object::implementation* construct() override; | |||
| 179 | ||||
| 180 | /// Destroy a timer implementation, cancelling pending waiters. | |||
| 181 | inline void destroy(io_object::implementation* p) override; | |||
| 182 | ||||
| 183 | /// Cancel and recycle a timer implementation. | |||
| 184 | inline void destroy_impl(implementation& impl); | |||
| 185 | ||||
| 186 | /// Create or recycle a waiter node. | |||
| 187 | inline waiter_node* create_waiter(); | |||
| 188 | ||||
| 189 | /// Return a waiter node to the cache or free list. | |||
| 190 | inline void destroy_waiter(waiter_node* w); | |||
| 191 | ||||
| 192 | /// Update the timer expiry, cancelling existing waiters. | |||
| 193 | inline std::size_t update_timer(implementation& impl, time_point new_time); | |||
| 194 | ||||
| 195 | /// Insert a waiter into the timer's waiter list and the heap. | |||
| 196 | inline void insert_waiter(implementation& impl, waiter_node* w); | |||
| 197 | ||||
| 198 | /// Cancel all waiters on a timer. | |||
| 199 | inline std::size_t cancel_timer(implementation& impl); | |||
| 200 | ||||
| 201 | /// Cancel a single waiter ( stop_token callback path ). | |||
| 202 | inline void cancel_waiter(waiter_node* w); | |||
| 203 | ||||
| 204 | /// Cancel one waiter on a timer. | |||
| 205 | inline std::size_t cancel_one_waiter(implementation& impl); | |||
| 206 | ||||
| 207 | /// Complete all waiters whose timers have expired. | |||
| 208 | inline std::size_t process_expired(); | |||
| 209 | ||||
| 210 | private: | |||
| 211 | 698955x | inline void refresh_cached_nearest() noexcept | ||
| 212 | { | |||
| 213 |
2/2✓ Branch 0 taken 8331 times.
✓ Branch 1 taken 690624 times.
|
698955x | auto ns = heap_.empty() ? (std::numeric_limits<std::int64_t>::max)() | |
| 214 | 690624x | : heap_[0].time_.time_since_epoch().count(); | ||
| 215 | 698955x | cached_nearest_ns_.store(ns, std::memory_order_release); | ||
| 216 | 698955x | } | ||
| 217 | ||||
| 218 | inline void remove_timer_impl(implementation& impl); | |||
| 219 | inline void up_heap(std::size_t index); | |||
| 220 | inline void down_heap(std::size_t index); | |||
| 221 | inline void swap_heap(std::size_t i1, std::size_t i2); | |||
| 222 | }; | |||
| 223 | ||||
| 224 | struct BOOST_COROSIO_SYMBOL_VISIBLE waiter_node | |||
| 225 | : intrusive_list<waiter_node>::node | |||
| 226 | { | |||
| 227 | // Embedded completion op — avoids heap allocation per fire/cancel | |||
| 228 | struct completion_op final : scheduler_op | |||
| 229 | { | |||
| 230 | 247x | waiter_node* waiter_ = nullptr; | ||
| 231 | ||||
| 232 | static void do_complete( | |||
| 233 | void* owner, scheduler_op* base, std::uint32_t, std::uint32_t); | |||
| 234 | ||||
| 235 | 741x | completion_op() noexcept : scheduler_op(&do_complete) {} | ||
| 236 | ||||
| 237 | void operator()() override; | |||
| 238 | void destroy() override; | |||
| 239 | }; | |||
| 240 | ||||
| 241 | // Per-waiter stop_token cancellation | |||
| 242 | struct canceller | |||
| 243 | { | |||
| 244 | waiter_node* waiter_; | |||
| 245 | void operator()() const; | |||
| 246 | }; | |||
| 247 | ||||
| 248 | // nullptr once removed from timer's waiter list (concurrency marker) | |||
| 249 | 247x | timer_service::implementation* impl_ = nullptr; | ||
| 250 | 247x | timer_service* svc_ = nullptr; | ||
| 251 | std::coroutine_handle<> h_; | |||
| 252 | 247x | capy::continuation* cont_ = nullptr; | ||
| 253 | capy::executor_ref d_; | |||
| 254 | 247x | std::error_code* ec_out_ = nullptr; | ||
| 255 | std::stop_token token_; | |||
| 256 | std::optional<std::stop_callback<canceller>> stop_cb_; | |||
| 257 | completion_op op_; | |||
| 258 | std::error_code ec_value_; | |||
| 259 | 247x | waiter_node* next_free_ = nullptr; | ||
| 260 | ||||
| 261 | 1235x | waiter_node() noexcept | ||
| 262 | 247x | { | ||
| 263 | 247x | op_.waiter_ = this; | ||
| 264 | 494x | } | ||
| 265 | }; | |||
| 266 | ||||
| 267 | struct timer_service::implementation final : timer::implementation | |||
| 268 | { | |||
| 269 | using clock_type = std::chrono::steady_clock; | |||
| 270 | using time_point = clock_type::time_point; | |||
| 271 | using duration = clock_type::duration; | |||
| 272 | ||||
| 273 | timer_service* svc_ = nullptr; | |||
| 274 | intrusive_list<waiter_node> waiters_; | |||
| 275 | ||||
| 276 | // Free list linkage (reused when impl is on free_list) | |||
| 277 | 298x | implementation* next_free_ = nullptr; | ||
| 278 | ||||
| 279 | inline explicit implementation(timer_service& svc) noexcept; | |||
| 280 | ||||
| 281 | inline std::coroutine_handle<> wait( | |||
| 282 | std::coroutine_handle<>, | |||
| 283 | capy::executor_ref, | |||
| 284 | std::stop_token, | |||
| 285 | std::error_code*, | |||
| 286 | capy::continuation*) override; | |||
| 287 | }; | |||
| 288 | ||||
| 289 | // Thread-local caches avoid hot-path mutex acquisitions: | |||
| 290 | // 1. Impl cache — single-slot, validated by comparing svc_ | |||
| 291 | // 2. Waiter cache — single-slot, no service affinity | |||
| 292 | // All caches are cleared by timer_service_invalidate_cache() during shutdown. | |||
| 293 | ||||
| 294 | inline thread_local_ptr<timer_service::implementation> tl_cached_impl; | |||
| 295 | inline thread_local_ptr<waiter_node> tl_cached_waiter; | |||
| 296 | ||||
| 297 | inline timer_service::implementation* | |||
| 298 | 7240x | try_pop_tl_cache(timer_service* svc) noexcept | ||
| 299 | { | |||
| 300 | 7240x | auto* impl = tl_cached_impl.get(); | ||
| 301 |
2/2✓ Branch 0 taken 298 times.
✓ Branch 1 taken 6942 times.
|
7240x | if (impl) | |
| 302 | { | |||
| 303 | 6942x | tl_cached_impl.set(nullptr); | ||
| 304 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6942 times.
|
6942x | if (impl->svc_ == svc) | |
| 305 | 6942x | return impl; | ||
| 306 | // Stale impl from a destroyed service | |||
| 307 | ✗ | delete impl; | ||
| 308 | ✗ | } | ||
| 309 | 298x | return nullptr; | ||
| 310 | 7240x | } | ||
| 311 | ||||
| 312 | inline bool | |||
| 313 | 7232x | try_push_tl_cache(timer_service::implementation* impl) noexcept | ||
| 314 | { | |||
| 315 |
2/2✓ Branch 0 taken 90 times.
✓ Branch 1 taken 7142 times.
|
7232x | if (!tl_cached_impl.get()) | |
| 316 | { | |||
| 317 | 7142x | tl_cached_impl.set(impl); | ||
| 318 | 7142x | return true; | ||
| 319 | } | |||
| 320 | 90x | return false; | ||
| 321 | 7232x | } | ||
| 322 | ||||
| 323 | inline waiter_node* | |||
| 324 | 7199x | try_pop_waiter_tl_cache() noexcept | ||
| 325 | { | |||
| 326 | 7199x | auto* w = tl_cached_waiter.get(); | ||
| 327 |
2/2✓ Branch 0 taken 6950 times.
✓ Branch 1 taken 249 times.
|
7199x | if (w) | |
| 328 | { | |||
| 329 | 6950x | tl_cached_waiter.set(nullptr); | ||
| 330 | 6950x | return w; | ||
| 331 | } | |||
| 332 | 249x | return nullptr; | ||
| 333 | 7199x | } | ||
| 334 | ||||
| 335 | inline bool | |||
| 336 | 7183x | try_push_waiter_tl_cache(waiter_node* w) noexcept | ||
| 337 | { | |||
| 338 |
2/2✓ Branch 0 taken 86 times.
✓ Branch 1 taken 7097 times.
|
7183x | if (!tl_cached_waiter.get()) | |
| 339 | { | |||
| 340 | 7097x | tl_cached_waiter.set(w); | ||
| 341 | 7097x | return true; | ||
| 342 | } | |||
| 343 | 86x | return false; | ||
| 344 | 7183x | } | ||
| 345 | ||||
| 346 | inline void | |||
| 347 | 679x | timer_service_invalidate_cache() noexcept | ||
| 348 | { | |||
| 349 |
2/2✓ Branch 0 taken 479 times.
✓ Branch 1 taken 200 times.
|
679x | delete tl_cached_impl.get(); | |
| 350 | 679x | tl_cached_impl.set(nullptr); | ||
| 351 | ||||
| 352 |
2/2✓ Branch 0 taken 532 times.
✓ Branch 1 taken 147 times.
|
679x | delete tl_cached_waiter.get(); | |
| 353 | 679x | tl_cached_waiter.set(nullptr); | ||
| 354 | 679x | } | ||
| 355 | ||||
| 356 | // timer_service out-of-class member function definitions | |||
| 357 | ||||
| 358 |
1/2✓ Branch 0 taken 298 times.
✗ Branch 1 not taken.
|
596x | inline timer_service::implementation::implementation( | |
| 359 | timer_service& svc) noexcept | |||
| 360 | 298x | : svc_(&svc) | ||
| 361 | 596x | { | ||
| 362 | 596x | } | ||
| 363 | ||||
| 364 | inline void | |||
| 365 | 679x | timer_service::shutdown() | ||
| 366 | { | |||
| 367 | 679x | timer_service_invalidate_cache(); | ||
| 368 | 679x | shutting_down_ = true; | ||
| 369 | ||||
| 370 | // Snapshot impls and detach them from the heap so that | |||
| 371 | // coroutine-owned timer destructors (triggered by h.destroy() | |||
| 372 | // below) cannot re-enter remove_timer_impl() and mutate the | |||
| 373 | // vector during iteration. | |||
| 374 | 679x | std::vector<implementation*> impls; | ||
| 375 |
1/2✓ Branch 0 taken 679 times.
✗ Branch 1 not taken.
|
679x | impls.reserve(heap_.size()); | |
| 376 |
2/2✓ Branch 0 taken 679 times.
✓ Branch 1 taken 8 times.
|
687x | for (auto& entry : heap_) | |
| 377 | { | |||
| 378 | 8x | entry.timer_->heap_index_ = (std::numeric_limits<std::size_t>::max)(); | ||
| 379 |
1/2✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
|
8x | impls.push_back(entry.timer_); | |
| 380 | } | |||
| 381 | 679x | heap_.clear(); | ||
| 382 | 1358x | cached_nearest_ns_.store( | ||
| 383 | 679x | (std::numeric_limits<std::int64_t>::max)(), std::memory_order_release); | ||
| 384 | ||||
| 385 | // Cancel waiting timers. Each waiter called work_started() | |||
| 386 | // in implementation::wait(). On IOCP the scheduler shutdown | |||
| 387 | // loop exits when outstanding_work_ reaches zero, so we must | |||
| 388 | // call work_finished() here to balance it. On other backends | |||
| 389 | // this is harmless. | |||
| 390 |
2/2✓ Branch 0 taken 679 times.
✓ Branch 1 taken 8 times.
|
687x | for (auto* impl : impls) | |
| 391 | { | |||
| 392 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 8 times.
|
16x | while (auto* w = impl->waiters_.pop_front()) | |
| 393 | { | |||
| 394 | 8x | w->stop_cb_.reset(); | ||
| 395 | 8x | auto h = std::exchange(w->h_, {}); | ||
| 396 | 8x | sched_->work_finished(); | ||
| 397 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8x | if (h) | |
| 398 |
1/2✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
|
8x | h.destroy(); | |
| 399 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8x | delete w; | |
| 400 | } | |||
| 401 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8x | delete impl; | |
| 402 | } | |||
| 403 | ||||
| 404 | // Delete free-listed impls | |||
| 405 |
2/2✓ Branch 0 taken 90 times.
✓ Branch 1 taken 679 times.
|
769x | while (free_list_) | |
| 406 | { | |||
| 407 | 90x | auto* next = free_list_->next_free_; | ||
| 408 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90x | delete free_list_; | |
| 409 | 90x | free_list_ = next; | ||
| 410 | } | |||
| 411 | ||||
| 412 | // Delete free-listed waiters | |||
| 413 |
2/2✓ Branch 0 taken 84 times.
✓ Branch 1 taken 679 times.
|
763x | while (waiter_free_list_) | |
| 414 | { | |||
| 415 | 84x | auto* next = waiter_free_list_->next_free_; | ||
| 416 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 84 times.
|
84x | delete waiter_free_list_; | |
| 417 | 84x | waiter_free_list_ = next; | ||
| 418 | } | |||
| 419 | 679x | } | ||
| 420 | ||||
| 421 | inline io_object::implementation* | |||
| 422 | 7240x | timer_service::construct() | ||
| 423 | { | |||
| 424 | 7240x | implementation* impl = try_pop_tl_cache(this); | ||
| 425 |
2/2✓ Branch 0 taken 6942 times.
✓ Branch 1 taken 298 times.
|
7240x | if (impl) | |
| 426 | { | |||
| 427 | 6942x | impl->svc_ = this; | ||
| 428 | 6942x | impl->heap_index_ = (std::numeric_limits<std::size_t>::max)(); | ||
| 429 | 6942x | impl->might_have_pending_waits_ = false; | ||
| 430 | 6942x | return impl; | ||
| 431 | } | |||
| 432 | ||||
| 433 | 298x | std::lock_guard lock(mutex_); | ||
| 434 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 298 times.
|
298x | if (free_list_) | |
| 435 | { | |||
| 436 | ✗ | impl = free_list_; | ||
| 437 | ✗ | free_list_ = impl->next_free_; | ||
| 438 | ✗ | impl->next_free_ = nullptr; | ||
| 439 | ✗ | impl->svc_ = this; | ||
| 440 | ✗ | impl->heap_index_ = (std::numeric_limits<std::size_t>::max)(); | ||
| 441 | ✗ | impl->might_have_pending_waits_ = false; | ||
| 442 | ✗ | } | ||
| 443 | else | |||
| 444 | { | |||
| 445 |
1/2✓ Branch 0 taken 298 times.
✗ Branch 1 not taken.
|
298x | impl = new implementation(*this); | |
| 446 | } | |||
| 447 | 298x | return impl; | ||
| 448 | 7240x | } | ||
| 449 | ||||
| 450 | inline void | |||
| 451 | 7238x | timer_service::destroy(io_object::implementation* p) | ||
| 452 | { | |||
| 453 | 7238x | destroy_impl(static_cast<implementation&>(*p)); | ||
| 454 | 7238x | } | ||
| 455 | ||||
| 456 | inline void | |||
| 457 | 7238x | timer_service::destroy_impl(implementation& impl) | ||
| 458 | { | |||
| 459 | // During shutdown the impl is owned by the shutdown loop. | |||
| 460 | // Re-entering here (from a coroutine-owned timer destructor | |||
| 461 | // triggered by h.destroy()) must not modify the heap or | |||
| 462 | // recycle the impl — shutdown deletes it directly. | |||
| 463 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 7232 times.
|
7238x | if (shutting_down_) | |
| 464 | 6x | return; | ||
| 465 | ||||
| 466 | 7232x | cancel_timer(impl); | ||
| 467 | ||||
| 468 |
1/2✓ Branch 0 taken 7232 times.
✗ Branch 1 not taken.
|
7232x | if (impl.heap_index_ != (std::numeric_limits<std::size_t>::max)()) | |
| 469 | { | |||
| 470 | ✗ | std::lock_guard lock(mutex_); | ||
| 471 | ✗ | remove_timer_impl(impl); | ||
| 472 | ✗ | refresh_cached_nearest(); | ||
| 473 | ✗ | } | ||
| 474 | ||||
| 475 |
2/2✓ Branch 0 taken 7142 times.
✓ Branch 1 taken 90 times.
|
7232x | if (try_push_tl_cache(&impl)) | |
| 476 | 7142x | return; | ||
| 477 | ||||
| 478 | 90x | std::lock_guard lock(mutex_); | ||
| 479 | 90x | impl.next_free_ = free_list_; | ||
| 480 | 90x | free_list_ = &impl; | ||
| 481 | 7238x | } | ||
| 482 | ||||
| 483 | inline waiter_node* | |||
| 484 | 7199x | timer_service::create_waiter() | ||
| 485 | { | |||
| 486 |
2/2✓ Branch 0 taken 6950 times.
✓ Branch 1 taken 249 times.
|
7199x | if (auto* w = try_pop_waiter_tl_cache()) | |
| 487 | 6950x | return w; | ||
| 488 | ||||
| 489 | 249x | std::lock_guard lock(mutex_); | ||
| 490 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 247 times.
|
249x | if (waiter_free_list_) | |
| 491 | { | |||
| 492 | 2x | auto* w = waiter_free_list_; | ||
| 493 | 2x | waiter_free_list_ = w->next_free_; | ||
| 494 | 2x | w->next_free_ = nullptr; | ||
| 495 | 2x | return w; | ||
| 496 | } | |||
| 497 | ||||
| 498 |
1/2✓ Branch 0 taken 247 times.
✗ Branch 1 not taken.
|
247x | return new waiter_node(); | |
| 499 | 7199x | } | ||
| 500 | ||||
| 501 | inline void | |||
| 502 | 7183x | timer_service::destroy_waiter(waiter_node* w) | ||
| 503 | { | |||
| 504 |
2/2✓ Branch 0 taken 7097 times.
✓ Branch 1 taken 86 times.
|
7183x | if (try_push_waiter_tl_cache(w)) | |
| 505 | 7097x | return; | ||
| 506 | ||||
| 507 | 86x | std::lock_guard lock(mutex_); | ||
| 508 | 86x | w->next_free_ = waiter_free_list_; | ||
| 509 | 86x | waiter_free_list_ = w; | ||
| 510 | 7183x | } | ||
| 511 | ||||
| 512 | inline std::size_t | |||
| 513 | 6x | timer_service::update_timer(implementation& impl, time_point new_time) | ||
| 514 | { | |||
| 515 | 6x | bool in_heap = | ||
| 516 | 6x | (impl.heap_index_ != (std::numeric_limits<std::size_t>::max)()); | ||
| 517 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
6x | if (!in_heap && impl.waiters_.empty()) | |
| 518 | ✗ | return 0; | ||
| 519 | ||||
| 520 | 6x | bool notify = false; | ||
| 521 | 6x | intrusive_list<waiter_node> canceled; | ||
| 522 | ||||
| 523 | { | |||
| 524 | 6x | std::lock_guard lock(mutex_); | ||
| 525 | ||||
| 526 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 6 times.
|
16x | while (auto* w = impl.waiters_.pop_front()) | |
| 527 | { | |||
| 528 | 10x | w->impl_ = nullptr; | ||
| 529 | 10x | canceled.push_back(w); | ||
| 530 | } | |||
| 531 | ||||
| 532 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6x | if (impl.heap_index_ < heap_.size()) | |
| 533 | { | |||
| 534 | 6x | time_point old_time = heap_[impl.heap_index_].time_; | ||
| 535 | 6x | heap_[impl.heap_index_].time_ = new_time; | ||
| 536 | ||||
| 537 |
2/4✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
✗ Branch 3 not taken.
|
6x | if (new_time < old_time) | |
| 538 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6x | up_heap(impl.heap_index_); | |
| 539 | else | |||
| 540 | ✗ | down_heap(impl.heap_index_); | ||
| 541 | ||||
| 542 | 6x | notify = (impl.heap_index_ == 0); | ||
| 543 | 6x | } | ||
| 544 | ||||
| 545 | 6x | refresh_cached_nearest(); | ||
| 546 | 6x | } | ||
| 547 | ||||
| 548 | 6x | std::size_t count = 0; | ||
| 549 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 6 times.
|
16x | while (auto* w = canceled.pop_front()) | |
| 550 | { | |||
| 551 | 10x | w->ec_value_ = make_error_code(capy::error::canceled); | ||
| 552 | 10x | sched_->post(&w->op_); | ||
| 553 | 10x | ++count; | ||
| 554 | } | |||
| 555 | ||||
| 556 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6x | if (notify) | |
| 557 | 6x | on_earliest_changed_(); | ||
| 558 | ||||
| 559 | 6x | return count; | ||
| 560 | 6x | } | ||
| 561 | ||||
| 562 | inline void | |||
| 563 | 7199x | timer_service::insert_waiter(implementation& impl, waiter_node* w) | ||
| 564 | { | |||
| 565 | 7199x | bool notify = false; | ||
| 566 | { | |||
| 567 | 7199x | std::lock_guard lock(mutex_); | ||
| 568 |
2/2✓ Branch 0 taken 22 times.
✓ Branch 1 taken 7177 times.
|
7199x | if (impl.heap_index_ == (std::numeric_limits<std::size_t>::max)()) | |
| 569 | { | |||
| 570 | 7177x | impl.heap_index_ = heap_.size(); | ||
| 571 |
1/2✓ Branch 0 taken 7177 times.
✗ Branch 1 not taken.
|
7177x | heap_.push_back({impl.expiry_, &impl}); | |
| 572 |
1/2✓ Branch 0 taken 7177 times.
✗ Branch 1 not taken.
|
7177x | up_heap(heap_.size() - 1); | |
| 573 | 7177x | notify = (impl.heap_index_ == 0); | ||
| 574 | 7177x | refresh_cached_nearest(); | ||
| 575 | 7177x | } | ||
| 576 | 7199x | impl.waiters_.push_back(w); | ||
| 577 | 7199x | } | ||
| 578 |
2/2✓ Branch 0 taken 55 times.
✓ Branch 1 taken 7144 times.
|
7199x | if (notify) | |
| 579 | 7144x | on_earliest_changed_(); | ||
| 580 | 7199x | } | ||
| 581 | ||||
| 582 | inline std::size_t | |||
| 583 | 7826x | timer_service::cancel_timer(implementation& impl) | ||
| 584 | { | |||
| 585 |
2/2✓ Branch 0 taken 7216 times.
✓ Branch 1 taken 610 times.
|
7826x | if (!impl.might_have_pending_waits_) | |
| 586 | 7216x | return 0; | ||
| 587 | ||||
| 588 | // Not in heap and no waiters — just clear the flag | |||
| 589 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 610 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
610x | if (impl.heap_index_ == (std::numeric_limits<std::size_t>::max)() && | |
| 590 | ✗ | impl.waiters_.empty()) | ||
| 591 | { | |||
| 592 | ✗ | impl.might_have_pending_waits_ = false; | ||
| 593 | ✗ | return 0; | ||
| 594 | } | |||
| 595 | ||||
| 596 | 610x | intrusive_list<waiter_node> canceled; | ||
| 597 | ||||
| 598 | { | |||
| 599 | 610x | std::lock_guard lock(mutex_); | ||
| 600 |
1/2✓ Branch 0 taken 610 times.
✗ Branch 1 not taken.
|
610x | remove_timer_impl(impl); | |
| 601 |
2/2✓ Branch 0 taken 614 times.
✓ Branch 1 taken 610 times.
|
1224x | while (auto* w = impl.waiters_.pop_front()) | |
| 602 | { | |||
| 603 | 614x | w->impl_ = nullptr; | ||
| 604 | 614x | canceled.push_back(w); | ||
| 605 | } | |||
| 606 | 610x | refresh_cached_nearest(); | ||
| 607 | 610x | } | ||
| 608 | ||||
| 609 | 610x | impl.might_have_pending_waits_ = false; | ||
| 610 | ||||
| 611 | 610x | std::size_t count = 0; | ||
| 612 |
2/2✓ Branch 0 taken 614 times.
✓ Branch 1 taken 610 times.
|
1224x | while (auto* w = canceled.pop_front()) | |
| 613 | { | |||
| 614 | 614x | w->ec_value_ = make_error_code(capy::error::canceled); | ||
| 615 | 614x | sched_->post(&w->op_); | ||
| 616 | 614x | ++count; | ||
| 617 | } | |||
| 618 | ||||
| 619 | 610x | return count; | ||
| 620 | 7826x | } | ||
| 621 | ||||
| 622 | inline void | |||
| 623 | 36x | timer_service::cancel_waiter(waiter_node* w) | ||
| 624 | { | |||
| 625 | { | |||
| 626 | 36x | std::lock_guard lock(mutex_); | ||
| 627 | // Already removed by cancel_timer or process_expired | |||
| 628 |
1/2✓ Branch 0 taken 36 times.
✗ Branch 1 not taken.
|
36x | if (!w->impl_) | |
| 629 | ✗ | return; | ||
| 630 | 36x | auto* impl = w->impl_; | ||
| 631 | 36x | w->impl_ = nullptr; | ||
| 632 | 36x | impl->waiters_.remove(w); | ||
| 633 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 34 times.
|
36x | if (impl->waiters_.empty()) | |
| 634 | { | |||
| 635 |
1/2✓ Branch 0 taken 34 times.
✗ Branch 1 not taken.
|
34x | remove_timer_impl(*impl); | |
| 636 | 34x | impl->might_have_pending_waits_ = false; | ||
| 637 | 34x | } | ||
| 638 | 36x | refresh_cached_nearest(); | ||
| 639 | 36x | } | ||
| 640 | ||||
| 641 | 36x | w->ec_value_ = make_error_code(capy::error::canceled); | ||
| 642 | 36x | sched_->post(&w->op_); | ||
| 643 | 36x | } | ||
| 644 | ||||
| 645 | inline std::size_t | |||
| 646 | 2x | timer_service::cancel_one_waiter(implementation& impl) | ||
| 647 | { | |||
| 648 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2x | if (!impl.might_have_pending_waits_) | |
| 649 | ✗ | return 0; | ||
| 650 | ||||
| 651 | 2x | waiter_node* w = nullptr; | ||
| 652 | ||||
| 653 | { | |||
| 654 | 2x | std::lock_guard lock(mutex_); | ||
| 655 | 2x | w = impl.waiters_.pop_front(); | ||
| 656 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2x | if (!w) | |
| 657 | ✗ | return 0; | ||
| 658 | 2x | w->impl_ = nullptr; | ||
| 659 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2x | if (impl.waiters_.empty()) | |
| 660 | { | |||
| 661 | ✗ | remove_timer_impl(impl); | ||
| 662 | ✗ | impl.might_have_pending_waits_ = false; | ||
| 663 | ✗ | } | ||
| 664 | 2x | refresh_cached_nearest(); | ||
| 665 | 2x | } | ||
| 666 | ||||
| 667 | 2x | w->ec_value_ = make_error_code(capy::error::canceled); | ||
| 668 | 2x | sched_->post(&w->op_); | ||
| 669 | 2x | return 1; | ||
| 670 | 2x | } | ||
| 671 | ||||
| 672 | inline std::size_t | |||
| 673 | 691124x | timer_service::process_expired() | ||
| 674 | { | |||
| 675 | 691124x | intrusive_list<waiter_node> expired; | ||
| 676 | ||||
| 677 | { | |||
| 678 | 691124x | std::lock_guard lock(mutex_); | ||
| 679 | 691124x | auto now = clock_type::now(); | ||
| 680 | ||||
| 681 |
4/4✓ Branch 0 taken 7695 times.
✓ Branch 1 taken 689954 times.
✓ Branch 2 taken 691124 times.
✓ Branch 3 taken 6525 times.
|
697649x | while (!heap_.empty() && heap_[0].time_ <= now) | |
| 682 | { | |||
| 683 | 6525x | implementation* t = heap_[0].timer_; | ||
| 684 |
1/2✓ Branch 0 taken 6525 times.
✗ Branch 1 not taken.
|
6525x | remove_timer_impl(*t); | |
| 685 |
2/2✓ Branch 0 taken 6529 times.
✓ Branch 1 taken 6525 times.
|
13054x | while (auto* w = t->waiters_.pop_front()) | |
| 686 | { | |||
| 687 | 6529x | w->impl_ = nullptr; | ||
| 688 | 6529x | w->ec_value_ = {}; | ||
| 689 | 6529x | expired.push_back(w); | ||
| 690 | } | |||
| 691 | 6525x | t->might_have_pending_waits_ = false; | ||
| 692 | } | |||
| 693 | ||||
| 694 | 691124x | refresh_cached_nearest(); | ||
| 695 | 691124x | } | ||
| 696 | ||||
| 697 | 691124x | std::size_t count = 0; | ||
| 698 |
2/2✓ Branch 0 taken 6529 times.
✓ Branch 1 taken 691124 times.
|
697653x | while (auto* w = expired.pop_front()) | |
| 699 | { | |||
| 700 | 6529x | sched_->post(&w->op_); | ||
| 701 | 6529x | ++count; | ||
| 702 | } | |||
| 703 | ||||
| 704 | 691124x | return count; | ||
| 705 | ✗ | } | ||
| 706 | ||||
| 707 | inline void | |||
| 708 | 7169x | timer_service::remove_timer_impl(implementation& impl) | ||
| 709 | { | |||
| 710 | 7169x | std::size_t index = impl.heap_index_; | ||
| 711 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7169 times.
|
7169x | if (index >= heap_.size()) | |
| 712 | ✗ | return; // Not in heap | ||
| 713 | ||||
| 714 |
2/2✓ Branch 0 taken 757 times.
✓ Branch 1 taken 6412 times.
|
7169x | if (index == heap_.size() - 1) | |
| 715 | { | |||
| 716 | // Last element, just pop | |||
| 717 | 757x | impl.heap_index_ = (std::numeric_limits<std::size_t>::max)(); | ||
| 718 | 757x | heap_.pop_back(); | ||
| 719 | 757x | } | ||
| 720 | else | |||
| 721 | { | |||
| 722 | // Swap with last and reheapify | |||
| 723 | 6412x | swap_heap(index, heap_.size() - 1); | ||
| 724 | 6412x | impl.heap_index_ = (std::numeric_limits<std::size_t>::max)(); | ||
| 725 | 6412x | heap_.pop_back(); | ||
| 726 | ||||
| 727 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 6412 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
6412x | if (index > 0 && heap_[index].time_ < heap_[(index - 1) / 2].time_) | |
| 728 | ✗ | up_heap(index); | ||
| 729 | else | |||
| 730 | 6412x | down_heap(index); | ||
| 731 | } | |||
| 732 | 7169x | } | ||
| 733 | ||||
| 734 | inline void | |||
| 735 | 7183x | timer_service::up_heap(std::size_t index) | ||
| 736 | { | |||
| 737 |
2/2✓ Branch 0 taken 7150 times.
✓ Branch 1 taken 6422 times.
|
13572x | while (index > 0) | |
| 738 | { | |||
| 739 | 6422x | std::size_t parent = (index - 1) / 2; | ||
| 740 |
2/2✓ Branch 0 taken 33 times.
✓ Branch 1 taken 6389 times.
|
6422x | if (!(heap_[index].time_ < heap_[parent].time_)) | |
| 741 | 33x | break; | ||
| 742 | 6389x | swap_heap(index, parent); | ||
| 743 | 6389x | index = parent; | ||
| 744 | } | |||
| 745 | 7183x | } | ||
| 746 | ||||
| 747 | inline void | |||
| 748 | 6412x | timer_service::down_heap(std::size_t index) | ||
| 749 | { | |||
| 750 | 6412x | std::size_t child = index * 2 + 1; | ||
| 751 |
2/2✓ Branch 0 taken 6406 times.
✓ Branch 1 taken 6 times.
|
6412x | while (child < heap_.size()) | |
| 752 | { | |||
| 753 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
6x | std::size_t min_child = (child + 1 == heap_.size() || | |
| 754 | ✗ | heap_[child].time_ < heap_[child + 1].time_) | ||
| 755 | 6x | ? child | ||
| 756 | ✗ | : child + 1; | ||
| 757 | ||||
| 758 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6x | if (heap_[index].time_ < heap_[min_child].time_) | |
| 759 | 6x | break; | ||
| 760 | ||||
| 761 | ✗ | swap_heap(index, min_child); | ||
| 762 | ✗ | index = min_child; | ||
| 763 | ✗ | child = index * 2 + 1; | ||
| 764 | } | |||
| 765 | 6412x | } | ||
| 766 | ||||
| 767 | inline void | |||
| 768 | 12801x | timer_service::swap_heap(std::size_t i1, std::size_t i2) | ||
| 769 | { | |||
| 770 | 12801x | heap_entry tmp = heap_[i1]; | ||
| 771 | 12801x | heap_[i1] = heap_[i2]; | ||
| 772 | 12801x | heap_[i2] = tmp; | ||
| 773 | 12801x | heap_[i1].timer_->heap_index_ = i1; | ||
| 774 | 12801x | heap_[i2].timer_->heap_index_ = i2; | ||
| 775 | 12801x | } | ||
| 776 | ||||
| 777 | // waiter_node out-of-class member function definitions | |||
| 778 | ||||
| 779 | inline void | |||
| 780 | 36x | waiter_node::canceller::operator()() const | ||
| 781 | { | |||
| 782 | 36x | waiter_->svc_->cancel_waiter(waiter_); | ||
| 783 | 36x | } | ||
| 784 | ||||
| 785 | inline void | |||
| 786 | ✗ | waiter_node::completion_op::do_complete( | ||
| 787 | [[maybe_unused]] void* owner, | |||
| 788 | scheduler_op* base, | |||
| 789 | std::uint32_t, | |||
| 790 | std::uint32_t) | |||
| 791 | { | |||
| 792 | // owner is always non-null here. The destroy path (owner == nullptr) | |||
| 793 | // is unreachable because completion_op overrides destroy() directly, | |||
| 794 | // bypassing scheduler_op::destroy() which would call func_(nullptr, ...). | |||
| 795 | ✗ | BOOST_COROSIO_ASSERT(owner); | ||
| 796 | ✗ | static_cast<completion_op*>(base)->operator()(); | ||
| 797 | ✗ | } | ||
| 798 | ||||
| 799 | inline void | |||
| 800 | 7183x | waiter_node::completion_op::operator()() | ||
| 801 | { | |||
| 802 | 7183x | auto* w = waiter_; | ||
| 803 | 7183x | w->stop_cb_.reset(); | ||
| 804 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7183 times.
|
7183x | if (w->ec_out_) | |
| 805 | 7183x | *w->ec_out_ = w->ec_value_; | ||
| 806 | ||||
| 807 | 7183x | auto* cont = w->cont_; | ||
| 808 | 7183x | auto d = w->d_; | ||
| 809 | 7183x | auto* svc = w->svc_; | ||
| 810 | 7183x | auto& sched = svc->get_scheduler(); | ||
| 811 | ||||
| 812 | 7183x | svc->destroy_waiter(w); | ||
| 813 | ||||
| 814 | 7183x | d.post(*cont); | ||
| 815 | 7183x | sched.work_finished(); | ||
| 816 | 7183x | } | ||
| 817 | ||||
| 818 | // GCC 14 false-positive: inlining ~optional<stop_callback> through | |||
| 819 | // delete loses track that stop_cb_ was already .reset() above. | |||
| 820 | #if defined(__GNUC__) && !defined(__clang__) | |||
| 821 | #pragma GCC diagnostic push | |||
| 822 | #pragma GCC diagnostic ignored "-Wmaybe-uninitialized" | |||
| 823 | #endif | |||
| 824 | inline void | |||
| 825 | 8x | waiter_node::completion_op::destroy() | ||
| 826 | { | |||
| 827 | // Called during scheduler shutdown drain when this completion_op is | |||
| 828 | // in the scheduler's ready queue (posted by cancel_timer() or | |||
| 829 | // process_expired()). Balances the work_started() from | |||
| 830 | // implementation::wait(). The scheduler drain loop separately | |||
| 831 | // balances the work_started() from post(). On IOCP both decrements | |||
| 832 | // are required for outstanding_work_ to reach zero; on other | |||
| 833 | // backends this is harmless. | |||
| 834 | // | |||
| 835 | // This override also prevents scheduler_op::destroy() from calling | |||
| 836 | // do_complete(nullptr, ...). See also: timer_service::shutdown() | |||
| 837 | // which drains waiters still in the timer heap (the other path). | |||
| 838 | 8x | auto* w = waiter_; | ||
| 839 | 8x | w->stop_cb_.reset(); | ||
| 840 | 8x | auto h = std::exchange(w->h_, {}); | ||
| 841 | 8x | auto& sched = w->svc_->get_scheduler(); | ||
| 842 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8x | delete w; | |
| 843 | 8x | sched.work_finished(); | ||
| 844 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8x | if (h) | |
| 845 | 8x | h.destroy(); | ||
| 846 | 8x | } | ||
| 847 | #if defined(__GNUC__) && !defined(__clang__) | |||
| 848 | #pragma GCC diagnostic pop | |||
| 849 | #endif | |||
| 850 | ||||
| 851 | inline std::coroutine_handle<> | |||
| 852 | 7214x | timer_service::implementation::wait( | ||
| 853 | std::coroutine_handle<> h, | |||
| 854 | capy::executor_ref d, | |||
| 855 | std::stop_token token, | |||
| 856 | std::error_code* ec, | |||
| 857 | capy::continuation* cont) | |||
| 858 | { | |||
| 859 | // Already-expired fast path — no waiter_node, no mutex. | |||
| 860 | // Post instead of dispatch so the coroutine yields to the | |||
| 861 | // scheduler, allowing other queued work to run. | |||
| 862 |
2/2✓ Branch 0 taken 22 times.
✓ Branch 1 taken 7192 times.
|
7214x | if (heap_index_ == (std::numeric_limits<std::size_t>::max)()) | |
| 863 | { | |||
| 864 |
3/4✗ Branch 0 not taken.
✓ Branch 1 taken 7192 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 7177 times.
|
7192x | if (expiry_ == (time_point::min)() || expiry_ <= clock_type::now()) | |
| 865 | { | |||
| 866 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15x | if (ec) | |
| 867 | 15x | *ec = {}; | ||
| 868 | 15x | d.post(*cont); | ||
| 869 | 15x | return std::noop_coroutine(); | ||
| 870 | } | |||
| 871 | 7177x | } | ||
| 872 | ||||
| 873 | 7199x | auto* w = svc_->create_waiter(); | ||
| 874 | 7199x | w->impl_ = this; | ||
| 875 | 7199x | w->svc_ = svc_; | ||
| 876 | 7199x | w->h_ = h; | ||
| 877 | 7199x | w->cont_ = cont; | ||
| 878 | 7199x | w->d_ = d; | ||
| 879 | 7199x | w->token_ = std::move(token); | ||
| 880 | 7199x | w->ec_out_ = ec; | ||
| 881 | ||||
| 882 | 7199x | svc_->insert_waiter(*this, w); | ||
| 883 | 7199x | might_have_pending_waits_ = true; | ||
| 884 | 7199x | svc_->get_scheduler().work_started(); | ||
| 885 | ||||
| 886 |
2/2✓ Branch 0 taken 7139 times.
✓ Branch 1 taken 60 times.
|
7199x | if (w->token_.stop_possible()) | |
| 887 | 60x | w->stop_cb_.emplace(w->token_, waiter_node::canceller{w}); | ||
| 888 | ||||
| 889 | 7199x | return std::noop_coroutine(); | ||
| 890 | 7214x | } | ||
| 891 | ||||
| 892 | // Free functions | |||
| 893 | ||||
| 894 | struct timer_service_access | |||
| 895 | { | |||
| 896 | 7240x | static timer_service& get_timer(io_context& ctx) noexcept | ||
| 897 | { | |||
| 898 | 7240x | return *ctx.timer_svc_; | ||
| 899 | } | |||
| 900 | ||||
| 901 | 679x | static void set_timer(io_context& ctx, timer_service& svc) noexcept | ||
| 902 | { | |||
| 903 | 679x | ctx.timer_svc_ = &svc; | ||
| 904 | 679x | } | ||
| 905 | }; | |||
| 906 | ||||
| 907 | // Bypass find_service() mutex by reading io_context's cached pointer | |||
| 908 | inline io_object::io_service& | |||
| 909 | 7240x | timer_service_direct(capy::execution_context& ctx) noexcept | ||
| 910 | { | |||
| 911 | 7240x | return timer_service_access::get_timer(static_cast<io_context&>(ctx)); | ||
| 912 | } | |||
| 913 | ||||
| 914 | inline std::size_t | |||
| 915 | 6x | timer_service_update_expiry(timer::implementation& base) | ||
| 916 | { | |||
| 917 | 6x | auto& impl = static_cast<timer_service::implementation&>(base); | ||
| 918 | 6x | return impl.svc_->update_timer(impl, impl.expiry_); | ||
| 919 | } | |||
| 920 | ||||
| 921 | inline std::size_t | |||
| 922 | 594x | timer_service_cancel(timer::implementation& base) noexcept | ||
| 923 | { | |||
| 924 | 594x | auto& impl = static_cast<timer_service::implementation&>(base); | ||
| 925 |
1/2✓ Branch 0 taken 594 times.
✗ Branch 1 not taken.
|
594x | return impl.svc_->cancel_timer(impl); | |
| 926 | } | |||
| 927 | ||||
| 928 | inline std::size_t | |||
| 929 | 2x | timer_service_cancel_one(timer::implementation& base) noexcept | ||
| 930 | { | |||
| 931 | 2x | auto& impl = static_cast<timer_service::implementation&>(base); | ||
| 932 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2x | return impl.svc_->cancel_one_waiter(impl); | |
| 933 | } | |||
| 934 | ||||
| 935 | inline timer_service& | |||
| 936 | 679x | get_timer_service(capy::execution_context& ctx, scheduler& sched) | ||
| 937 | { | |||
| 938 | 679x | auto& svc = ctx.make_service<timer_service>(sched); | ||
| 939 | 679x | timer_service_access::set_timer(static_cast<io_context&>(ctx), svc); | ||
| 940 | 679x | return svc; | ||
| 941 | } | |||
| 942 | ||||
| 943 | } // namespace boost::corosio::detail | |||
| 944 | ||||
| 945 | #endif | |||
| 946 |