1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
mod in_memory_session_store;
#[cfg(feature = "server")]
use crate::mcp_server::ServerRuntime;
use async_trait::async_trait;
pub use in_memory_session_store::*;
use rust_mcp_transport::SessionId;
use std::sync::Arc;
/// Trait defining the interface for session storage operations
///
/// This trait provides asynchronous methods for managing session data,
/// Implementors must be Send and Sync to support concurrent access.
#[async_trait]
pub trait SessionStore: Send + Sync {
/// Retrieves a session by its identifier
///
/// # Arguments
/// * `key` - The session identifier to look up
///
/// # Returns
/// * `Option<Arc<ServerRuntime>>` - The session stream if found, None otherwise
async fn get(&self, key: &SessionId) -> Option<Arc<ServerRuntime>>;
/// Stores a new session with the given identifier
///
/// # Arguments
/// * `key` - The session identifier
/// * `value` - The duplex stream to store
async fn set(&self, key: SessionId, value: Arc<ServerRuntime>);
/// Deletes a session by its identifier
///
/// # Arguments
/// * `key` - The session identifier to delete
async fn delete(&self, key: &SessionId);
async fn has(&self, session: &SessionId) -> bool;
async fn keys(&self) -> Vec<SessionId>;
async fn values(&self) -> Vec<Arc<ServerRuntime>>;
/// Clears all sessions from the store
async fn clear(&self);
/// Returns `true` when the store cannot accept a new session.
///
/// Callers should reject new-session creation (e.g. `initialize`) with
/// `503 Service Unavailable` when this returns `true`. The default
/// implementation reports unlimited capacity.
async fn is_full(&self) -> bool {
false
}
}