moirai-executor 0.9.0

Hybrid executor implementation for Moirai concurrency library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

#[cfg(feature = "registry-diagnostics")]
use std::time::Duration;

#[cfg(feature = "registry-diagnostics")]
use super::registry::TaskRegistry;
#[cfg(feature = "registry-diagnostics")]
use super::state::{TIMESTAMP_NOT_RECORDED, TaskState};

#[cfg(feature = "registry-diagnostics")]
impl TaskRegistry {
    /// Diagnostic-only block lookup path for benchmark attribution.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_block_lookup(&self) -> u64 {
        let id = self.issue_id();
        let (block_index, slot_index) = id.location();
        let block = self.ensure_block(block_index);
        let slot_occupied = block.get(slot_index).is_some();
        std::hint::black_box(slot_occupied);
        id.get()
    }

    /// Diagnostic-only synchronization cost on the registration path.
    ///
    /// Replaces the former `registry_mutex_lock_only` row: the executor no
    /// longer wraps the registry in a mutex, so the cost this attributes is
    /// the shared acquire on the block directory that registration takes
    /// instead. Keeping the row lets the tradeoff recorded in ADR 0005's
    /// 2026-08-29 revision be re-measured rather than argued.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_directory_shared_acquire(&self) -> usize {
        let blocks = self
            .blocks
            .read()
            .expect("task registry block directory is never poisoned");
        let len = blocks.end();
        drop(blocks);
        len
    }

    /// Diagnostic-only slot initialization path for benchmark attribution.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_slot_initialize(&self) -> u64 {
        let id = self.issue_id();
        let task_id = id.get();
        let (block_index, slot_index) = id.location();
        // SAFETY: `id` was issued once above and this is its only registration.
        unsafe { self.ensure_block(block_index).insert(slot_index) };
        task_id
    }

    /// Diagnostic-only lifecycle timestamp publication path for benchmark attribution.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_lifecycle_timestamp_publication() -> Duration {
        let state = TaskState::new();
        let started_after_ns = state.mark_started(0);
        state.mark_completed_since(started_after_ns)
    }

    /// Diagnostic-only task-state construction path for benchmark attribution.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_task_state_construct() -> usize {
        let state = TaskState::new();
        std::hint::black_box(state);
        core::mem::size_of::<TaskState>()
    }

    /// Diagnostic-only start timestamp publication on an existing slot.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_mark_started(&self, task_id: u64, worker_id: usize) -> u64 {
        self.with_state(task_id, |state| state.mark_started(worker_id))
            .unwrap_or(TIMESTAMP_NOT_RECORDED)
    }

    /// Diagnostic-only completion timestamp publication on an existing slot.
    #[doc(hidden)]
    #[cold]
    #[inline(never)]
    pub fn diagnostic_mark_completed_since(&self, task_id: u64, started_after_ns: u64) -> Duration {
        self.with_state(task_id, |state| {
            state.mark_completed_since(started_after_ns)
        })
        .unwrap_or(Duration::ZERO)
    }

    /// Diagnostic-only production token lifecycle path with registry-local ID allocation.
    #[doc(hidden)]
    pub fn diagnostic_register_next_and_complete_with_token(&self) -> Duration {
        // SAFETY: this method completes and drops the token before returning,
        // so `self` and its block storage outlive the scheduled lease.
        let (_id, lifecycle) = unsafe { self.register_next_scheduled_task() };
        lifecycle.start(0).complete()
    }

    /// Diagnostic-only block-retaining token lifecycle for ownership-cost attribution.
    #[doc(hidden)]
    pub fn diagnostic_register_next_and_complete_with_retained_token(&self) -> Duration {
        let (_id, lifecycle) = self.register_next_task();
        lifecycle.start(0).complete()
    }

    /// Diagnostic-only production token lifecycle path with registry-local ID output.
    #[doc(hidden)]
    pub fn diagnostic_register_next_and_complete_with_token_id(&self) -> (u64, Duration) {
        // SAFETY: this method completes and drops the token before returning,
        // so `self` and its block storage outlive the scheduled lease.
        let (id, lifecycle) = unsafe { self.register_next_scheduled_task() };
        (id, lifecycle.start(0).complete())
    }
}