Struct InvocationContextManager 

pub struct InvocationContextManager { /* private fields */ }

Central coordinator for correlating platform spans with function spans.

This manager maintains state for each active invocation and provides methods for registering parent spans from function telemetry and waiting for correlation with platform events.

Implementations

impl InvocationContextManager

pub fn new(config: CorrelationConfig) -> Self

Creates a new context manager with the given configuration.

pub fn with_defaults() -> Self

Creates a new context manager with default configuration.

pub async fn register_invocation(&self, request_id: RequestId)

Registers a new invocation context.

Call this when receiving an INVOKE event from the Extensions API.

pub async fn set_parent_span(&self, request_id: &str, context: SpanContext)

Sets the parent span context for an invocation.

Call this when a span with faas.parent_span = true is received from the function’s OTLP telemetry.

pub async fn wait_for_parent_span( &self, request_id: &str, ) -> Option<SpanContext>

Waits for the parent span context to become available.

This method waits up to max_correlation_delay for the parent span to be registered. Returns None if the timeout expires.

pub async fn get_parent_span(&self, request_id: &str) -> Option<SpanContext>

Gets the current parent span context if available (non-blocking).

pub async fn add_platform_event(&self, event: PlatformEvent)

Adds a platform event to the invocation context.

Events are buffered until the parent span is available for correlation.

pub async fn take_platform_events(&self, request_id: &str) -> Vec<PlatformEvent>

Takes all pending platform events for an invocation.

Returns the events and clears the buffer.

pub async fn remove_invocation(&self, request_id: &str)

Removes an invocation context.

Call this after the invocation is complete and all data has been flushed.

pub async fn active_count(&self) -> usize

Returns the number of active invocation contexts.

pub async fn cleanup_stale_contexts(&self)

Cleans up stale invocation contexts.

Removes contexts that have exceeded max_invocation_lifetime.

pub fn emit_orphaned_spans(&self) -> bool

Returns whether to emit orphaned spans (spans without parent context).