Struct MemoryExecutionStore 

pub struct MemoryExecutionStore { /* private fields */ }

In-memory execution store for testing, development, and prototyping.

Tracks tool executions and token usage records in memory-backed HashMaps protected by RwLock. Data is lost when the process exits. Implements ExecutionStore.

Implementations

impl MemoryExecutionStore

pub fn new() -> Self

Creates an empty in-memory execution store.

pub async fn set_message_count(&self, session_id: Uuid, count: u64)

Sets the message count for a session, used by session_stats.

In a real backend this would be computed from a messages table; the in-memory store requires the caller to explicitly provide it since messages are maintained in the session store, not here.

Trait Implementations

impl Default for MemoryExecutionStore

fn default() -> MemoryExecutionStore

Returns the “default value” for a type.

impl ExecutionStore for MemoryExecutionStore

fn record_execution<'life0, 'async_trait>( &'life0 self, execution: ToolExecution, ) -> Pin<Box<dyn Future<Output = StoreResult<ToolExecution>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Records a tool execution and returns it with server-assigned fields.

fn list_executions<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 Uuid, ) -> Pin<Box<dyn Future<Output = StoreResult<Vec<ToolExecution>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Returns all tool executions for a session ordered by creation time (ascending).

fn list_executions_by_message<'life0, 'life1, 'async_trait>( &'life0 self, message_id: &'life1 Uuid, ) -> Pin<Box<dyn Future<Output = StoreResult<Vec<ToolExecution>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Returns all tool executions for a specific message ordered by creation time.

fn record_usage<'life0, 'async_trait>( &'life0 self, record: UsageRecord, ) -> Pin<Box<dyn Future<Output = StoreResult<UsageRecord>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Records token usage for a single provider interaction.

fn list_usage<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 Uuid, ) -> Pin<Box<dyn Future<Output = StoreResult<Vec<UsageRecord>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Returns all usage records for a session ordered by creation time (ascending).

fn session_stats<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 Uuid, ) -> Pin<Box<dyn Future<Output = StoreResult<SessionStats>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Computes aggregate statistics for a session.

fn delete_by_session<'life0, 'life1, 'async_trait>( &'life0 self, session_id: &'life1 Uuid, ) -> Pin<Box<dyn Future<Output = StoreResult<u64>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Deletes all executions and usage records for a session.