Struct StorageBackedSessionStore 

pub struct StorageBackedSessionStore { /* private fields */ }

A session store backed by persistent storage with an in-memory cache.

All reads hit the cache first. All writes go to both the cache and the underlying storage backend. On construction, the cache is warmed from storage to handle process restarts.

Implementations

impl StorageBackedSessionStore

pub async fn new( storage: Arc<dyn StorageSessionStore>, ) -> Result<Self, StorageError>

Create a new storage-backed session store.

Loads all existing sessions from storage into the in-memory cache.

pub async fn create(&self, req: CreateSessionRequest) -> TaskSession

Create a new task session and return it.

pub async fn create_if_under_cap( &self, req: CreateSessionRequest, max_sessions: u64, ) -> Result<TaskSession, SessionError>

Atomically check per-agent session cap and create if under the limit.

pub async fn use_session( &self, session_id: SessionId, tool_name: &str, requesting_agent_id: Option<Uuid>, ) -> Result<TaskSession, SessionError>

Record a tool call against the session, checking all constraints.

pub async fn use_session_batch( &self, session_id: SessionId, tool_names: &[&str], requesting_agent_id: Option<Uuid>, ) -> Result<TaskSession, SessionError>

Atomically validate and record a batch of tool calls against the session.

Acquires the write lock once, validates ALL tools against the whitelist and budget, and only increments calls_made by the full batch count if every tool passes. If any tool fails validation, no budget is consumed for any of them.

pub async fn close( &self, session_id: SessionId, ) -> Result<TaskSession, SessionError>

Close a session, preventing further use.

pub async fn get( &self, session_id: SessionId, ) -> Result<TaskSession, SessionError>

Get a session by ID without modifying it.

pub async fn list_all(&self) -> Vec<TaskSession>

List all sessions currently in the store.

pub async fn count_active_for_agent(&self, agent_id: Uuid) -> u64

Count the number of active sessions for a given agent.

P0: Used to enforce per-agent concurrent session caps.

pub async fn close_sessions_for_agent(&self, agent_id: Uuid) -> usize

Close all active sessions belonging to a specific agent.

Called during agent deactivation.

pub async fn cleanup_expired(&self) -> usize

Remove expired sessions from cache and storage. Returns the number removed.

Trait Implementations

impl Clone for StorageBackedSessionStore

fn clone(&self) -> StorageBackedSessionStore

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.