2.1 KiB
2.1 KiB
MCP Streamable HTTP Proxy
MCP Streamable HTTP Proxy
Streamable HTTP proxy implementation for MCP using rmcp 0.12 with stateful session management.
Overview
This module provides a proxy implementation for MCP (Model Context Protocol) using Streamable HTTP transport with stateful session management.
Features
- Streamable HTTP Support: Uses rmcp 0.12 with enhanced Streamable HTTP transport
- Stateful Sessions: Custom SessionManager with backend version tracking
- Hot Swap: Supports backend connection replacement without downtime
- Version Control: Automatically invalidates sessions when backend reconnects
- High-level Client API: Simple connection interface hiding transport details
Architecture
Client → Streamable HTTP → ProxyAwareSessionManager → ProxyHandler → Backend MCP Service
↓
Version Tracking
(DashMap<SessionId, BackendVersion>)
Installation
Add to Cargo.toml:
[dependencies]
mcp-streamable-proxy = { version = "0.1.5", path = "../mcp-streamable-proxy" }
Usage
Server
use mcp_streamable_proxy::{McpServiceConfig, run_stream_server};
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let config = McpServiceConfig::new("my-service".to_string());
run_stream_server(config).await?;
Ok(())
}
Client
use mcp_streamable_proxy::{StreamClientConnection, McpClientConfig};
// Connect to an MCP server
let config = McpClientConfig::new("http://localhost:8080/mcp");
let conn = StreamClientConnection::connect(config).await?;
// List available tools
let tools = conn.list_tools().await?;
Session Management
The ProxyAwareSessionManager provides:
- Backend version tracking using DashMap
- Automatic session invalidation on backend reconnect
- Concurrent-safe session operations
Development
# Build
cargo build -p mcp-streamable-proxy
# Test
cargo test -p mcp-streamable-proxy
License
MIT OR Apache-2.0