include/boost/corosio/detail/timer_service.hpp

92.6% Lines (375/405) 97.9% List of functions (47/49) 68.9% Branches (124/180)
timer_service.hpp
f(x) Functions (49)
Function Calls Lines Branches Blocks
<unknown function 91> :91 boost::corosio::detail::timer_service::callback::callback() :96 2184x 100.0% 100.0% boost::corosio::detail::timer_service::callback::callback(void*, void (*)(void*)) :99 2184x 100.0% 100.0% boost::corosio::detail::timer_service::callback::operator()() const :108 5975x 100.0% 50.0% 100.0% boost::corosio::detail::timer_service::timer_service(boost::capy::execution_context&, boost::corosio::detail::scheduler&) :137 2184x 100.0% 100.0% boost::corosio::detail::timer_service::get_scheduler() :143 12048x 100.0% 100.0% boost::corosio::detail::timer_service::~timer_service() :149 3276x 100.0% 100.0% boost::corosio::detail::timer_service::set_on_earliest_changed(boost::corosio::detail::timer_service::callback) :155 1092x 100.0% 100.0% boost::corosio::detail::timer_service::nearest_expiry() const :168 1120214x 100.0% 50.0% 75.0% boost::corosio::detail::timer_service::refresh_cached_nearest() :211 1192024x 100.0% 100.0% 100.0% boost::corosio::detail::waiter_node::~waiter_node() :224 608x 100.0% 100.0% boost::corosio::detail::waiter_node::completion_op::~completion_op() :228 608x 100.0% 100.0% boost::corosio::detail::waiter_node::completion_op::completion_op() :235 608x 100.0% 100.0% boost::corosio::detail::waiter_node::waiter_node() :261 608x 100.0% 100.0% boost::corosio::detail::timer_service::implementation::~implementation() :267 744x 100.0% 100.0% boost::corosio::detail::try_pop_tl_cache(boost::corosio::detail::timer_service*) :298 6130x 80.0% 50.0% 62.0% boost::corosio::detail::try_push_tl_cache(boost::corosio::detail::timer_service::implementation*) :313 6122x 100.0% 100.0% 100.0% boost::corosio::detail::try_pop_waiter_tl_cache() :324 6028x 100.0% 100.0% 100.0% boost::corosio::detail::try_push_waiter_tl_cache(boost::corosio::detail::waiter_node*) :336 6012x 100.0% 100.0% 100.0% boost::corosio::detail::timer_service_invalidate_cache() :347 1092x 100.0% 100.0% 100.0% boost::corosio::detail::timer_service::implementation::implementation(boost::corosio::detail::timer_service&) :358 744x 100.0% 50.0% 100.0% boost::corosio::detail::timer_service::shutdown() :365 1092x 100.0% 69.2% 93.0% boost::corosio::detail::timer_service::construct() :422 6130x 63.2% 66.7% 70.0% boost::corosio::detail::timer_service::destroy(boost::corosio::io_object::implementation*) :451 6128x 100.0% 100.0% boost::corosio::detail::timer_service::destroy_impl(boost::corosio::detail::timer_service::implementation&) :457 6128x 73.3% 62.5% 63.0% boost::corosio::detail::timer_service::create_waiter() :484 6028x 100.0% 83.3% 80.0% boost::corosio::detail::timer_service::destroy_waiter(boost::corosio::detail::waiter_node*) :502 6012x 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 8x 96.7% 65.0% 82.0% boost::corosio::detail::timer_service::insert_waiter(boost::corosio::detail::timer_service::implementation&, boost::corosio::detail::waiter_node*) :563 6028x 100.0% 75.0% 77.0% boost::corosio::detail::timer_service::cancel_timer(boost::corosio::detail::timer_service::implementation&) :583 6653x 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 1185423x 95.0% 90.0% 86.0% boost::corosio::detail::timer_service::remove_timer_impl(boost::corosio::detail::timer_service::implementation&) :708 5998x 86.7% 50.0% 70.0% boost::corosio::detail::timer_service::up_heap(unsigned long) :735 6012x 100.0% 100.0% 100.0% boost::corosio::detail::timer_service::down_heap(unsigned long) :748 5239x 100.0% 87.5% 100.0% boost::corosio::detail::timer_service::swap_heap(unsigned long, unsigned long) :768 10453x 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 6012x 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 6074x 100.0% 80.0% 100.0% boost::corosio::detail::timer_service_access::get_timer(boost::corosio::io_context&) :896 6130x 100.0% 100.0% boost::corosio::detail::timer_service_access::set_timer(boost::corosio::io_context&, boost::corosio::detail::timer_service&) :901 1092x 100.0% 100.0% boost::corosio::detail::timer_service_direct(boost::capy::execution_context&) :909 6130x 100.0% 100.0% boost::corosio::detail::timer_service_update_expiry(boost::corosio::io_timer::implementation&) :915 8x 100.0% 100.0% boost::corosio::detail::timer_service_cancel(boost::corosio::io_timer::implementation&) :922 531x 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 1092x 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 1092x void* ctx_ = nullptr;
92 1092x void (*fn_)(void*) = nullptr;
93
94 public:
95 /// Construct an empty callback.
96 3276x callback() = default;
97
98 /// Construct a callback with the given context and function.
99 2184x 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 5975x void operator()() const
109 {
110
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5975 times.
5975x if (fn_)
111 5975x fn_(ctx_);
112 5975x }
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 1092x implementation* free_list_ = nullptr;
128 1092x waiter_node* waiter_free_list_ = nullptr;
129 callback on_earliest_changed_;
130 1092x bool shutting_down_ = false;
131 // Avoids mutex in nearest_expiry() and empty()
132 2184x mutable std::atomic<std::int64_t> cached_nearest_ns_{
133 1092x (std::numeric_limits<std::int64_t>::max)()};
134
135 public:
136 /// Construct the timer service bound to a scheduler.
137 5460x inline timer_service(capy::execution_context&, scheduler& sched)
138 1092x : sched_(&sched)
139 3276x {
140 2184x }
141
142 /// Return the associated scheduler.
143 12048x inline scheduler& get_scheduler() noexcept
144 {
145 12048x return *sched_;
146 }
147
148 /// Destroy the timer service.
149 3276x ~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 1092x inline void set_on_earliest_changed(callback cb)
156 {
157 1092x on_earliest_changed_ = cb;
158 1092x }
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 1120214x inline time_point nearest_expiry() const noexcept
169 {
170 1120214x auto ns = cached_nearest_ns_.load(std::memory_order_acquire);
171
2/4
✓ Branch 0 taken 1120214 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1120214 times.
✗ Branch 3 not taken.
1120214x 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 1192024x inline void refresh_cached_nearest() noexcept
212 {
213
2/2
✓ Branch 0 taken 9063 times.
✓ Branch 1 taken 1182961 times.
1192024x auto ns = heap_.empty() ? (std::numeric_limits<std::int64_t>::max)()
214 1182961x : heap_[0].time_.time_since_epoch().count();
215 1192024x cached_nearest_ns_.store(ns, std::memory_order_release);
216 1192024x }
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 304x waiter_node* waiter_ = nullptr;
231
232 static void do_complete(
233 void* owner, scheduler_op* base, std::uint32_t, std::uint32_t);
234
235 912x 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 304x timer_service::implementation* impl_ = nullptr;
250 304x timer_service* svc_ = nullptr;
251 std::coroutine_handle<> h_;
252 304x capy::continuation* cont_ = nullptr;
253 capy::executor_ref d_;
254 304x 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 304x waiter_node* next_free_ = nullptr;
260
261 1520x waiter_node() noexcept
262 304x {
263 304x op_.waiter_ = this;
264 608x }
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 372x 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 6130x try_pop_tl_cache(timer_service* svc) noexcept
299 {
300 6130x auto* impl = tl_cached_impl.get();
301
2/2
✓ Branch 0 taken 372 times.
✓ Branch 1 taken 5758 times.
6130x if (impl)
302 {
303 5758x tl_cached_impl.set(nullptr);
304
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5758 times.
5758x if (impl->svc_ == svc)
305 5758x return impl;
306 // Stale impl from a destroyed service
307 delete impl;
308 }
309 372x return nullptr;
310 6130x }
311
312 inline bool
313 6122x try_push_tl_cache(timer_service::implementation* impl) noexcept
314 {
315
2/2
✓ Branch 0 taken 109 times.
✓ Branch 1 taken 6013 times.
6122x if (!tl_cached_impl.get())
316 {
317 6013x tl_cached_impl.set(impl);
318 6013x return true;
319 }
320 109x return false;
321 6122x }
322
323 inline waiter_node*
324 6028x try_pop_waiter_tl_cache() noexcept
325 {
326 6028x auto* w = tl_cached_waiter.get();
327
2/2
✓ Branch 0 taken 5722 times.
✓ Branch 1 taken 306 times.
6028x if (w)
328 {
329 5722x tl_cached_waiter.set(nullptr);
330 5722x return w;
331 }
332 306x return nullptr;
333 6028x }
334
335 inline bool
336 6012x try_push_waiter_tl_cache(waiter_node* w) noexcept
337 {
338
2/2
✓ Branch 0 taken 88 times.
✓ Branch 1 taken 5924 times.
6012x if (!tl_cached_waiter.get())
339 {
340 5924x tl_cached_waiter.set(w);
341 5924x return true;
342 }
343 88x return false;
344 6012x }
345
346 inline void
347 1092x timer_service_invalidate_cache() noexcept
348 {
349
2/2
✓ Branch 0 taken 837 times.
✓ Branch 1 taken 255 times.
1092x delete tl_cached_impl.get();
350 1092x tl_cached_impl.set(nullptr);
351
352
2/2
✓ Branch 0 taken 890 times.
✓ Branch 1 taken 202 times.
1092x delete tl_cached_waiter.get();
353 1092x tl_cached_waiter.set(nullptr);
354 1092x }
355
356 // timer_service out-of-class member function definitions
357
358
1/2
✓ Branch 0 taken 372 times.
✗ Branch 1 not taken.
744x inline timer_service::implementation::implementation(
359 timer_service& svc) noexcept
360 372x : svc_(&svc)
361 744x {
362 744x }
363
364 inline void
365 1092x timer_service::shutdown()
366 {
367 1092x timer_service_invalidate_cache();
368 1092x 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 1092x std::vector<implementation*> impls;
375
1/2
✓ Branch 0 taken 1092 times.
✗ Branch 1 not taken.
1092x impls.reserve(heap_.size());
376
2/2
✓ Branch 0 taken 1092 times.
✓ Branch 1 taken 8 times.
1100x 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 1092x heap_.clear();
382 2184x cached_nearest_ns_.store(
383 1092x (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 1092 times.
✓ Branch 1 taken 8 times.
1100x 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 109 times.
✓ Branch 1 taken 1092 times.
1201x while (free_list_)
406 {
407 109x auto* next = free_list_->next_free_;
408
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 109 times.
109x delete free_list_;
409 109x free_list_ = next;
410 }
411
412 // Delete free-listed waiters
413
2/2
✓ Branch 0 taken 86 times.
✓ Branch 1 taken 1092 times.
1178x while (waiter_free_list_)
414 {
415 86x auto* next = waiter_free_list_->next_free_;
416
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 86 times.
86x delete waiter_free_list_;
417 86x waiter_free_list_ = next;
418 }
419 1092x }
420
421 inline io_object::implementation*
422 6130x timer_service::construct()
423 {
424 6130x implementation* impl = try_pop_tl_cache(this);
425
2/2
✓ Branch 0 taken 5758 times.
✓ Branch 1 taken 372 times.
6130x if (impl)
426 {
427 5758x impl->svc_ = this;
428 5758x impl->heap_index_ = (std::numeric_limits<std::size_t>::max)();
429 5758x impl->might_have_pending_waits_ = false;
430 5758x return impl;
431 }
432
433 372x std::lock_guard lock(mutex_);
434
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 372 times.
372x 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 372 times.
✗ Branch 1 not taken.
372x impl = new implementation(*this);
446 }
447 372x return impl;
448 6130x }
449
450 inline void
451 6128x timer_service::destroy(io_object::implementation* p)
452 {
453 6128x destroy_impl(static_cast<implementation&>(*p));
454 6128x }
455
456 inline void
457 6128x 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 6122 times.
6128x if (shutting_down_)
464 6x return;
465
466 6122x cancel_timer(impl);
467
468
1/2
✓ Branch 0 taken 6122 times.
✗ Branch 1 not taken.
6122x 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 6013 times.
✓ Branch 1 taken 109 times.
6122x if (try_push_tl_cache(&impl))
476 6013x return;
477
478 109x std::lock_guard lock(mutex_);
479 109x impl.next_free_ = free_list_;
480 109x free_list_ = &impl;
481 6128x }
482
483 inline waiter_node*
484 6028x timer_service::create_waiter()
485 {
486
2/2
✓ Branch 0 taken 5722 times.
✓ Branch 1 taken 306 times.
6028x if (auto* w = try_pop_waiter_tl_cache())
487 5722x return w;
488
489 306x std::lock_guard lock(mutex_);
490
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 304 times.
306x 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 304 times.
✗ Branch 1 not taken.
304x return new waiter_node();
499 6028x }
500
501 inline void
502 6012x timer_service::destroy_waiter(waiter_node* w)
503 {
504
2/2
✓ Branch 0 taken 5924 times.
✓ Branch 1 taken 88 times.
6012x if (try_push_waiter_tl_cache(w))
505 5924x return;
506
507 88x std::lock_guard lock(mutex_);
508 88x w->next_free_ = waiter_free_list_;
509 88x waiter_free_list_ = w;
510 6012x }
511
512 inline std::size_t
513 8x timer_service::update_timer(implementation& impl, time_point new_time)
514 {
515 8x bool in_heap =
516 8x (impl.heap_index_ != (std::numeric_limits<std::size_t>::max)());
517
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
8x if (!in_heap && impl.waiters_.empty())
518 return 0;
519
520 8x bool notify = false;
521 8x intrusive_list<waiter_node> canceled;
522
523 {
524 8x std::lock_guard lock(mutex_);
525
526
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 8 times.
20x while (auto* w = impl.waiters_.pop_front())
527 {
528 12x w->impl_ = nullptr;
529 12x canceled.push_back(w);
530 }
531
532
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
8x if (impl.heap_index_ < heap_.size())
533 {
534 8x time_point old_time = heap_[impl.heap_index_].time_;
535 8x heap_[impl.heap_index_].time_ = new_time;
536
537
3/4
✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 2 times.
8x 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
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
2x down_heap(impl.heap_index_);
541
542 8x notify = (impl.heap_index_ == 0);
543 8x }
544
545 8x refresh_cached_nearest();
546 8x }
547
548 8x std::size_t count = 0;
549
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 8 times.
20x while (auto* w = canceled.pop_front())
550 {
551 12x w->ec_value_ = make_error_code(capy::error::canceled);
552 12x sched_->post(&w->op_);
553 12x ++count;
554 }
555
556
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 6 times.
8x if (notify)
557 6x on_earliest_changed_();
558
559 8x return count;
560 8x }
561
562 inline void
563 6028x timer_service::insert_waiter(implementation& impl, waiter_node* w)
564 {
565 6028x bool notify = false;
566 {
567 6028x std::lock_guard lock(mutex_);
568
2/2
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 6006 times.
6028x if (impl.heap_index_ == (std::numeric_limits<std::size_t>::max)())
569 {
570 6006x impl.heap_index_ = heap_.size();
571
1/2
✓ Branch 0 taken 6006 times.
✗ Branch 1 not taken.
6006x heap_.push_back({impl.expiry_, &impl});
572
1/2
✓ Branch 0 taken 6006 times.
✗ Branch 1 not taken.
6006x up_heap(heap_.size() - 1);
573 6006x notify = (impl.heap_index_ == 0);
574 6006x refresh_cached_nearest();
575 6006x }
576 6028x impl.waiters_.push_back(w);
577 6028x }
578
2/2
✓ Branch 0 taken 59 times.
✓ Branch 1 taken 5969 times.
6028x if (notify)
579 5969x on_earliest_changed_();
580 6028x }
581
582 inline std::size_t
583 6653x timer_service::cancel_timer(implementation& impl)
584 {
585
2/2
✓ Branch 0 taken 6104 times.
✓ Branch 1 taken 549 times.
6653x if (!impl.might_have_pending_waits_)
586 6104x return 0;
587
588 // Not in heap and no waiters — just clear the flag
589
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 549 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
549x 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 549x intrusive_list<waiter_node> canceled;
597
598 {
599 549x std::lock_guard lock(mutex_);
600
1/2
✓ Branch 0 taken 549 times.
✗ Branch 1 not taken.
549x remove_timer_impl(impl);
601
2/2
✓ Branch 0 taken 551 times.
✓ Branch 1 taken 549 times.
1100x while (auto* w = impl.waiters_.pop_front())
602 {
603 551x w->impl_ = nullptr;
604 551x canceled.push_back(w);
605 }
606 549x refresh_cached_nearest();
607 549x }
608
609 549x impl.might_have_pending_waits_ = false;
610
611 549x std::size_t count = 0;
612
2/2
✓ Branch 0 taken 551 times.
✓ Branch 1 taken 549 times.
1100x while (auto* w = canceled.pop_front())
613 {
614 551x w->ec_value_ = make_error_code(capy::error::canceled);
615 551x sched_->post(&w->op_);
616 551x ++count;
617 }
618
619 549x return count;
620 6653x }
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 1185423x timer_service::process_expired()
674 {
675 1185423x intrusive_list<waiter_node> expired;
676
677 {
678 1185423x std::lock_guard lock(mutex_);
679 1185423x auto now = clock_type::now();
680
681
4/4
✓ Branch 0 taken 8488 times.
✓ Branch 1 taken 1182350 times.
✓ Branch 2 taken 1185423 times.
✓ Branch 3 taken 5415 times.
1190838x while (!heap_.empty() && heap_[0].time_ <= now)
682 {
683 5415x implementation* t = heap_[0].timer_;
684
1/2
✓ Branch 0 taken 5415 times.
✗ Branch 1 not taken.
5415x remove_timer_impl(*t);
685
2/2
✓ Branch 0 taken 5419 times.
✓ Branch 1 taken 5415 times.
10834x while (auto* w = t->waiters_.pop_front())
686 {
687 5419x w->impl_ = nullptr;
688 5419x w->ec_value_ = {};
689 5419x expired.push_back(w);
690 }
691 5415x t->might_have_pending_waits_ = false;
692 }
693
694 1185423x refresh_cached_nearest();
695 1185423x }
696
697 1185423x std::size_t count = 0;
698
2/2
✓ Branch 0 taken 5419 times.
✓ Branch 1 taken 1185423 times.
1190842x while (auto* w = expired.pop_front())
699 {
700 5419x sched_->post(&w->op_);
701 5419x ++count;
702 }
703
704 1185423x return count;
705 }
706
707 inline void
708 5998x timer_service::remove_timer_impl(implementation& impl)
709 {
710 5998x std::size_t index = impl.heap_index_;
711
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5998 times.
5998x if (index >= heap_.size())
712 return; // Not in heap
713
714
2/2
✓ Branch 0 taken 761 times.
✓ Branch 1 taken 5237 times.
5998x if (index == heap_.size() - 1)
715 {
716 // Last element, just pop
717 761x impl.heap_index_ = (std::numeric_limits<std::size_t>::max)();
718 761x heap_.pop_back();
719 761x }
720 else
721 {
722 // Swap with last and reheapify
723 5237x swap_heap(index, heap_.size() - 1);
724 5237x impl.heap_index_ = (std::numeric_limits<std::size_t>::max)();
725 5237x heap_.pop_back();
726
727
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5237 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
5237x if (index > 0 && heap_[index].time_ < heap_[(index - 1) / 2].time_)
728 up_heap(index);
729 else
730 5237x down_heap(index);
731 }
732 5998x }
733
734 inline void
735 6012x timer_service::up_heap(std::size_t index)
736 {
737
2/2
✓ Branch 0 taken 5975 times.
✓ Branch 1 taken 5249 times.
11224x while (index > 0)
738 {
739 5249x std::size_t parent = (index - 1) / 2;
740
2/2
✓ Branch 0 taken 37 times.
✓ Branch 1 taken 5212 times.
5249x if (!(heap_[index].time_ < heap_[parent].time_))
741 37x break;
742 5212x swap_heap(index, parent);
743 5212x index = parent;
744 }
745 6012x }
746
747 inline void
748 5239x timer_service::down_heap(std::size_t index)
749 {
750 5239x std::size_t child = index * 2 + 1;
751
2/2
✓ Branch 0 taken 5233 times.
✓ Branch 1 taken 10 times.
5243x while (child < heap_.size())
752 {
753
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
10x std::size_t min_child = (child + 1 == heap_.size() ||
754 2x heap_[child].time_ < heap_[child + 1].time_)
755 8x ? child
756 2x : child + 1;
757
758
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 4 times.
10x if (heap_[index].time_ < heap_[min_child].time_)
759 6x break;
760
761 4x swap_heap(index, min_child);
762 4x index = min_child;
763 4x child = index * 2 + 1;
764 }
765 5239x }
766
767 inline void
768 10453x timer_service::swap_heap(std::size_t i1, std::size_t i2)
769 {
770 10453x heap_entry tmp = heap_[i1];
771 10453x heap_[i1] = heap_[i2];
772 10453x heap_[i2] = tmp;
773 10453x heap_[i1].timer_->heap_index_ = i1;
774 10453x heap_[i2].timer_->heap_index_ = i2;
775 10453x }
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 6012x waiter_node::completion_op::operator()()
801 {
802 6012x auto* w = waiter_;
803 6012x w->stop_cb_.reset();
804
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6012 times.
6012x if (w->ec_out_)
805 6012x *w->ec_out_ = w->ec_value_;
806
807 6012x auto* cont = w->cont_;
808 6012x auto d = w->d_;
809 6012x auto* svc = w->svc_;
810 6012x auto& sched = svc->get_scheduler();
811
812 6012x svc->destroy_waiter(w);
813
814 6012x d.post(*cont);
815 6012x sched.work_finished();
816 6012x }
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 6074x 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 6052 times.
6074x if (heap_index_ == (std::numeric_limits<std::size_t>::max)())
863 {
864
3/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6052 times.
✓ Branch 2 taken 46 times.
✓ Branch 3 taken 6006 times.
6052x if (expiry_ == (time_point::min)() || expiry_ <= clock_type::now())
865 {
866
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46x if (ec)
867 46x *ec = {};
868 46x d.post(*cont);
869 46x return std::noop_coroutine();
870 }
871 6006x }
872
873 6028x auto* w = svc_->create_waiter();
874 6028x w->impl_ = this;
875 6028x w->svc_ = svc_;
876 6028x w->h_ = h;
877 6028x w->cont_ = cont;
878 6028x w->d_ = d;
879 6028x w->token_ = std::move(token);
880 6028x w->ec_out_ = ec;
881
882 6028x svc_->insert_waiter(*this, w);
883 6028x might_have_pending_waits_ = true;
884 6028x svc_->get_scheduler().work_started();
885
886
2/2
✓ Branch 0 taken 5968 times.
✓ Branch 1 taken 60 times.
6028x if (w->token_.stop_possible())
887 60x w->stop_cb_.emplace(w->token_, waiter_node::canceller{w});
888
889 6028x return std::noop_coroutine();
890 6074x }
891
892 // Free functions
893
894 struct timer_service_access
895 {
896 6130x static timer_service& get_timer(io_context& ctx) noexcept
897 {
898 6130x return *ctx.timer_svc_;
899 }
900
901 1092x static void set_timer(io_context& ctx, timer_service& svc) noexcept
902 {
903 1092x ctx.timer_svc_ = &svc;
904 1092x }
905 };
906
907 // Bypass find_service() mutex by reading io_context's cached pointer
908 inline io_object::io_service&
909 6130x timer_service_direct(capy::execution_context& ctx) noexcept
910 {
911 6130x return timer_service_access::get_timer(static_cast<io_context&>(ctx));
912 }
913
914 inline std::size_t
915 8x timer_service_update_expiry(timer::implementation& base)
916 {
917 8x auto& impl = static_cast<timer_service::implementation&>(base);
918 8x return impl.svc_->update_timer(impl, impl.expiry_);
919 }
920
921 inline std::size_t
922 531x timer_service_cancel(timer::implementation& base) noexcept
923 {
924 531x auto& impl = static_cast<timer_service::implementation&>(base);
925
1/2
✓ Branch 0 taken 531 times.
✗ Branch 1 not taken.
531x 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 1092x get_timer_service(capy::execution_context& ctx, scheduler& sched)
937 {
938 1092x auto& svc = ctx.make_service<timer_service>(sched);
939 1092x timer_service_access::set_timer(static_cast<io_context&>(ctx), svc);
940 1092x return svc;
941 }
942
943 } // namespace boost::corosio::detail
944
945 #endif
946