Struct ContextStore 

pub struct ContextStore { /* private fields */ }

Store for active contexts and traces

Implementations

impl ContextStore

pub fn new( max_traces: usize, max_spans_per_trace: usize, trace_ttl: Duration, ) -> Self

Create a new store with the given capacity limits and TTL

pub fn with_sampler(self, sampler: Sampler) -> Self

Replace the default sampler. Useful when an operator wants deterministic capture (e.g. AlwaysOn for diagnostics) or a custom ratio different from the default.

impl ContextStore

pub fn with_sampling(self, strategy: SamplingStrategy) -> Self

Override the default sampling strategy for this store

pub fn create_context( &self, origin_node: NodeId, ) -> Result<Context, ContextError>

Create a new context and register it

Capacity admission goes through the atomic try_reserve_slot CAS rather than a contexts.len() >= max probe. Two threads inserting at exactly capacity could otherwise both observe len < max after a cleanup_expired and both insert, growing the map past max_traces. The slot is reserved atomically before the insert; if the reserve fails after a cleanup_expired retry, we surface CapacityExceeded.

pub fn continue_context(&self, ctx: Context) -> Result<Context, ContextError>

Continue a context from a remote node

pub fn add_span(&self, span: Span) -> Result<(), ContextError>

Add a span to a trace

pub fn get_context(&self, trace_id: &TraceId) -> Option<Context>

Get a context by trace ID

pub fn get_spans(&self, trace_id: &TraceId) -> Vec<Span>

Get all spans for a trace

pub fn complete_trace(&self, trace_id: &TraceId) -> Option<(Context, Vec<Span>)>

Complete a trace and return all spans

Also releases the active_count slot so the try_reserve_slot admission gate can re-admit.

pub fn cleanup_expired(&self)

Cleanup expired traces

Every successful removal also releases an active_count slot so the try_reserve_slot admission gate can re-admit work as soon as expired entries are reclaimed.

pub fn stats(&self) -> ContextStoreStats

Get statistics