Struct StateManager 

pub struct StateManager { /* private fields */ }

State manager for pending generation sessions.

Uses an in-memory HashMap protected by RwLock for thread-safe access. Sessions expire after 30 minutes and are cleaned up lazily.

Examples

use mcp_execution_server::state::StateManager;
use mcp_execution_server::types::PendingGeneration;
use mcp_execution_core::{ServerId, ServerConfig};
use mcp_execution_introspector::ServerInfo;
use std::path::PathBuf;

let state = StateManager::new();

let pending = PendingGeneration::new(
    ServerId::new("github"),
    server_info,
    ServerConfig::builder().command("npx".to_string()).build(),
    PathBuf::from("/tmp/output"),
);

// Store and get session ID
let session_id = state.store(pending).await;

// Retrieve session data
let retrieved = state.take(session_id).await;
assert!(retrieved.is_some());

Implementations

impl StateManager

pub fn new() -> Self

Creates a new state manager.

pub async fn store(&self, generation: PendingGeneration) -> Uuid

Stores a pending generation and returns a session ID.

This operation also performs lazy cleanup of expired sessions.

Examples

use mcp_execution_server::state::StateManager;

let state = StateManager::new();
let session_id = state.store(pending).await;

pub async fn take(&self, session_id: Uuid) -> Option<PendingGeneration>

Retrieves and removes a pending generation.

Returns None if the session is not found or has expired. This operation also performs lazy cleanup of expired sessions.

Examples

use mcp_execution_server::state::StateManager;

let state = StateManager::new();
let session_id = state.store(pending).await;

let retrieved = state.take(session_id).await;
assert!(retrieved.is_some());

// Second take returns None (already removed)
let second = state.take(session_id).await;
assert!(second.is_none());

pub async fn get(&self, session_id: Uuid) -> Option<PendingGeneration>

Gets a pending generation without removing it.

Returns None if the session is not found or has expired.

Examples

use mcp_execution_server::state::StateManager;

let state = StateManager::new();
let session_id = state.store(pending).await;

// Get without removing
let peeked = state.get(session_id).await;
assert!(peeked.is_some());

// Still available
let peeked_again = state.get(session_id).await;
assert!(peeked_again.is_some());

pub async fn pending_count(&self) -> usize

Returns the current pending session count (excluding expired).

Examples

use mcp_execution_server::state::StateManager;

let state = StateManager::new();
assert_eq!(state.pending_count().await, 0);

pub async fn cleanup_expired(&self) -> usize

Cleans up all expired sessions.

Returns the number of sessions that were removed.

Examples

use mcp_execution_server::state::StateManager;

let state = StateManager::new();
let removed = state.cleanup_expired().await;
assert_eq!(removed, 0);

Trait Implementations

impl Debug for StateManager

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

Formats the value using the given formatter.

impl Default for StateManager

fn default() -> StateManager

Returns the “default value” for a type.