Struct ExtensionState 

pub struct ExtensionState { /* private fields */ }

Shared state for extension management.

This manages all registered extensions, their event queues, and coordination for event delivery.

Implementations

impl ExtensionState

pub fn new() -> Self

Creates a new extension state.

pub async fn register( &self, name: String, events: Vec<EventType>, ) -> RegisteredExtension

Registers a new extension.

Arguments

• name - Name of the extension (from Lambda-Extension-Name header)
• events - List of events the extension wants to receive

Returns

The registered extension information including its ID.

pub async fn broadcast_event(&self, event: LifecycleEvent)

Broadcasts an event to all extensions subscribed to that event type.

Arguments

• event - The lifecycle event to broadcast

pub async fn next_event(&self, extension_id: &str) -> Option<LifecycleEvent>

Waits for and retrieves the next event for a specific extension.

This is a long-poll operation that blocks until an event is available.

Arguments

• extension_id - The ID of the extension requesting the next event

Returns

The next lifecycle event for this extension.

pub async fn get_extension( &self, extension_id: &str, ) -> Option<RegisteredExtension>

Gets information about a registered extension.

Arguments

• extension_id - The ID of the extension

Returns

The extension information if it exists.

pub async fn get_all_extensions(&self) -> Vec<RegisteredExtension>

Gets all registered extensions.

pub async fn extension_count(&self) -> usize

Returns the number of registered extensions.

pub async fn get_invoke_subscribers(&self) -> Vec<ExtensionId>

Returns the IDs of all extensions subscribed to INVOKE events.

This is used to determine which extensions need to signal readiness after each invocation.

pub async fn get_shutdown_subscribers(&self) -> Vec<ExtensionId>

Returns the IDs of all extensions subscribed to SHUTDOWN events.

This is used to determine which extensions need to receive the SHUTDOWN event during graceful shutdown.

pub async fn wake_all_extensions(&self)

Wakes all extensions waiting on /next.

This is used during shutdown to unblock extensions waiting for events.

pub async fn is_queue_empty(&self, extension_id: &str) -> bool

Checks if an extension’s event queue is empty.

This is used during shutdown to determine if an extension has consumed all events (including the SHUTDOWN event).

pub async fn mark_shutdown_acknowledged(&self, extension_id: &str)

Marks an extension as having acknowledged shutdown.

This is called when an extension polls /next after receiving the SHUTDOWN event, signaling it has completed its cleanup work.

pub async fn is_shutdown_acknowledged(&self, extension_id: &str) -> bool

Checks if an extension has acknowledged shutdown.

pub async fn wait_for_shutdown_acknowledged(&self, extension_ids: &[String])

Waits for all specified extensions to acknowledge shutdown.

Returns when all extensions have polled /next after receiving SHUTDOWN.

pub async fn clear_shutdown_acknowledged(&self)

Clears shutdown acknowledgment state.

Called when starting a new shutdown sequence.

Trait Implementations

impl Debug for ExtensionState

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ExtensionState

fn default() -> Self

Returns the “default value” for a type.