batpak 0.8.2

Event sourcing with causal graphs and caller-defined gates. Sync API, no async runtime.
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
122
123

use crate::store::config::duration_micros;
use crate::store::{
    AppendReceipt, HlcPoint, Open, Store, StoreError, StoreInvariant, WatermarkKind,
};
use std::time::{Duration, Instant};

struct GateWaitMeasurement {
    result: Result<(), StoreError>,
    waited_us: u64,
}

fn measure_gate_wait<ResolveTarget, WaitForTarget>(
    resolve_target: ResolveTarget,
    wait_for_target: WaitForTarget,
) -> Result<GateWaitMeasurement, StoreError>
where
    ResolveTarget: FnOnce() -> Result<HlcPoint, StoreError>,
    WaitForTarget: FnOnce(HlcPoint) -> Result<(), StoreError>,
{
    let target = resolve_target()?;
    let started = Instant::now();
    let result = wait_for_target(target);
    Ok(GateWaitMeasurement {
        waited_us: duration_micros(started.elapsed()),
        result,
    })
}

/// Inline append-time durability wait.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub struct DurabilityGate {
    /// Watermark that must cross the appended event's HLC.
    pub kind: WatermarkKind,
    /// Maximum time to wait for the watermark.
    ///
    /// `WaitTimeout` means the append committed but did not cross the
    /// requested watermark within `timeout`. The event is still in the log;
    /// query reflects it. Re-call `wait_for_<kind>` with a longer timeout if
    /// you need to re-acquire the guarantee.
    pub timeout: Duration,
}

impl Store<Open> {
    fn receipt_point(&self, receipt: &AppendReceipt) -> Result<HlcPoint, StoreError> {
        use crate::id::EntityIdType;
        let raw = receipt.event_id.as_u128();
        self.index
            .get_by_id(raw)
            .map(|entry| HlcPoint {
                wall_ms: entry.wall_ms,
                global_sequence: entry.global_sequence,
            })
            .ok_or_else(|| StoreError::InvariantViolation {
                kind: StoreInvariant::GateReceiptNotIndexed { event_id: raw },
            })
    }

    pub(crate) fn wait_for_gate(
        &self,
        receipt: &AppendReceipt,
        gate: DurabilityGate,
    ) -> Result<(), StoreError> {
        let measurement = measure_gate_wait(
            || self.receipt_point(receipt),
            |target| match gate.kind {
                WatermarkKind::Durable => self.wait_for_durable(target, gate.timeout),
                WatermarkKind::Applied => self.wait_for_applied(target, gate.timeout),
                WatermarkKind::Visible => self.wait_for_visible(target, gate.timeout),
            },
        )?;
        tracing::trace!(
            target: "batpak::durability_gate",
            kind = ?gate.kind,
            waited_us = measurement.waited_us,
            ok = measurement.result.is_ok(),
            "append durability gate wait completed",
        );
        measurement.result
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::sync::{
        atomic::{AtomicBool, Ordering},
        Arc,
    };

    #[test]
    fn gate_wait_measurement_excludes_receipt_lookup_work() {
        let lookup_complete = Arc::new(AtomicBool::new(false));
        let wait_saw_lookup_complete = Arc::clone(&lookup_complete);
        let before_lookup = Instant::now();

        let measurement = measure_gate_wait(
            || {
                std::thread::sleep(Duration::from_millis(40));
                lookup_complete.store(true, Ordering::SeqCst);
                Ok(HlcPoint {
                    wall_ms: 1,
                    global_sequence: 1,
                })
            },
            |_| {
                assert!(
                    wait_saw_lookup_complete.load(Ordering::SeqCst),
                    "gate wait must start only after receipt_point lookup completes"
                );
                Ok(())
            },
        )
        .expect("measurement succeeds");

        let lookup_inclusive_us = duration_micros(before_lookup.elapsed());
        assert!(
            measurement.waited_us < lookup_inclusive_us / 2,
            "waited_us must measure only the watermark wait window, not receipt lookup work; waited_us={} lookup_inclusive_us={lookup_inclusive_us}",
            measurement.waited_us
        );
        assert!(measurement.result.is_ok());
    }
}