fix: adviced comments

Author: greenhat616

crates/rmcp/src/transport/streamable_http_server/tower.rs

@@ -34,6 +34,10 @@ pub struct StreamableHttpServerConfig {
     pub sse_keep_alive: Option<Duration>,
     /// If true, the server will create a session for each request and keep it alive.
     pub stateful_mode: bool,
+    /// Cancellation token for the Streamable HTTP server.
+    ///
+    /// When this token is cancelled, all active sessions are terminated and
+    /// the server stops accepting new requests.
     pub cancellation_token: CancellationToken,
 }
 

crates/rmcp/tests/test_with_js.rs

@@ -94,18 +94,20 @@ async fn test_with_js_streamable_http_client() -> anyhow::Result<()> {
         .wait()
         .await?;
 
+    let ct = CancellationToken::new();
     let service: StreamableHttpService<Calculator, LocalSessionManager> =
         StreamableHttpService::new(
             || Ok(Calculator::new()),
             Default::default(),
             StreamableHttpServerConfig {
                 stateful_mode: true,
                 sse_keep_alive: None,
+                cancellation_token: ct.child_token(),
             },
         );
     let router = axum::Router::new().nest_service("/mcp", service);
     let tcp_listener = tokio::net::TcpListener::bind(STREAMABLE_HTTP_BIND_ADDRESS).await?;
-    let ct = CancellationToken::new();
+
     let handle = tokio::spawn({
         let ct = ct.clone();
         async move {

examples/servers/src/counter_streamhttp.rs

@@ -1,6 +1,5 @@
-use rmcp::transport::{
-    StreamableHttpServerConfig,
-    streamable_http_server::{StreamableHttpService, session::local::LocalSessionManager},
+use rmcp::transport::streamable_http_server::{
+    StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
 };
 use tracing_subscriber::{
     layer::SubscriberExt,