low-latency-data-structures 0.0.7

Experimental lock-free SPSC, SPMC broadcast, and SeqLock primitives tuned for ultra-low-latency systems
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

//! Adapter for the `rtrb` SPSC ring. Gated behind `_bench_rtrb`.

use std::marker::PhantomData;

use rtrb::{Consumer, Producer, PushError, RingBuffer};

use crate::bench::harness::spsc::{SpscBench, SpscCons, SpscProd};

/// Marker type that constructs an rtrb SPSC pair with capacity `C`.
///
/// rtrb takes capacity at runtime; the const generic just gets forwarded
/// to `RingBuffer::new` so the same bench-core call site works uniformly
/// across compile-time-capacity (ours, heapless) and runtime-capacity
/// (rtrb, ringbuf, crossbeam, ...) adapters.
pub struct RtrbSpsc<T, const C: usize>(PhantomData<fn() -> T>);

impl<T: Send + 'static, const C: usize> SpscBench<T, C> for RtrbSpsc<T, C> {
    type Prod = Producer<T>;
    type Cons = Consumer<T>;
    const NAME: &'static str = "rtrb";
    fn new() -> (Self::Prod, Self::Cons) {
        RingBuffer::new(C)
    }
}

impl<T: Send> SpscProd<T> for Producer<T> {
    #[inline]
    fn try_push(&mut self, v: T) -> Result<(), T> {
        self.push(v).map_err(|PushError::Full(v)| v)
    }
}

impl<T: Send> SpscCons<T> for Consumer<T> {
    #[inline]
    fn try_pop(&mut self) -> Option<T> {
        self.pop().ok()
    }
}