Struct SessionStore 

pub struct SessionStore { /* private fields */ }

In-memory session store with TTL-based cleanup.

Implementations

impl SessionStore

pub fn new() -> Self

Create a new empty session store.

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

Create a new task session and return it.

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

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

Returns the updated session on success, or an error if:

• Session not found
• Session expired (would return 408)
• Budget exceeded (would return 429)
• Tool not authorized (would return 403)

pub async fn use_session_batch( &self, session_id: SessionId, tool_names: &[&str], ) -> 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 (active, expired, and closed).

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.

When an agent is deactivated via cascade_deactivate, all its sessions must be immediately closed.

pub async fn cleanup_expired(&self) -> usize

Remove expired sessions from the store. Returns the number removed.

Trait Implementations

impl Clone for SessionStore

fn clone(&self) -> SessionStore

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Default for SessionStore

fn default() -> Self

Returns the “default value” for a type.