modelcontextprotocol/rust-sdk
Watch
1
0
Fork
You've already forked rust-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

feat(transport): support streamable http server (#152)

Browse source 
* feat(transport): support streamable http server

* refactor(transport): add common module for shared transport utilities

* perf(tracing): add more log for streamable http session

* fix(test): fix port conflict in test with js

* fix(transport): fix id generating and session management

1. cancel when deleting
2. use `wrapping` and `saturating` correctly

* chore(naming): rename sse server to streamable http server
This commit is contained in:
4t145 2025-05-08 13:57:51 +08:00 • committed by GitHub
commit 57e14c56e8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1247 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

37
crates/rmcp/tests/test_with_js.rs
View file

@ -1,12 +1,13 @@
use rmcp::{
ServiceExt,
transport::{SseServer, TokioChildProcess},
transport::{SseServer, TokioChildProcess, streamable_http_server::axum::StreamableHttpServer},
};
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
mod common;
use common::calculator::Calculator;
const BIND_ADDRESS: &str = "127.0.0.1:8000";
const SSE_BIND_ADDRESS: &str = "127.0.0.1:8000";
const STREAMABLE_HTTP_BIND_ADDRESS: &str = "127.0.0.1:8001";
#[tokio::test]
async fn test_with_js_client() -> anyhow::Result<()> {
@ -24,7 +25,7 @@ async fn test_with_js_client() -> anyhow::Result<()> {
.wait()
.await?;
let ct = SseServer::serve(BIND_ADDRESS.parse()?)
let ct = SseServer::serve(SSE_BIND_ADDRESS.parse()?)
.await?
.with_service(Calculator::default);
@ -66,3 +67,33 @@ async fn test_with_js_server() -> anyhow::Result<()> {
client.cancel().await?;
Ok(())
}
#[tokio::test]
async fn test_with_js_streamable_http_client() -> anyhow::Result<()> {
let _ = tracing_subscriber::registry()
.with(
tracing_subscriber::EnvFilter::try_from_default_env()
.unwrap_or_else(|_| "debug".to_string().into()),
)
.with(tracing_subscriber::fmt::layer())
.try_init();
tokio::process::Command::new("npm")
.arg("install")
.current_dir("tests/test_with_js")
.spawn()?
.wait()
.await?;
let ct = StreamableHttpServer::serve(STREAMABLE_HTTP_BIND_ADDRESS.parse()?)
.await?
.with_service(Calculator::default);
let exit_status = tokio::process::Command::new("node")
.arg("tests/test_with_js/streamable_client.js")
.spawn()?
.wait()
.await?;
assert!(exit_status.success());
ct.cancel();
Ok(())
}