Enum AnySessionStore 

pub enum AnySessionStore {
    InMemory(SessionStore),
    StorageBacked(StorageBackedSessionStore),
}

A session store that dispatches to either in-memory or storage-backed.

Variants

InMemory(SessionStore)

StorageBacked(StorageBackedSessionStore)

Implementations

impl AnySessionStore

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.

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.

Validates ALL tools and budget atomically under a single lock acquisition. No budget is consumed unless every tool in the batch passes validation.

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

Close a session.

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

Get a session by ID.

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

List all sessions.

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. Returns the number removed.

Trait Implementations

impl Clone for AnySessionStore

fn clone(&self) -> AnySessionStore

Returns a duplicate of the value.

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

Performs copy-assignment from source.