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: sse client optionally skip the endpoint event (#187)

Browse source
This commit is contained in:
4t145 2025-05-18 18:55:02 +08:00 • committed by GitHub
commit 35282ee34c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 17 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

27
crates/rmcp/src/transport/sse_client.rs
View file

@ -113,16 +113,20 @@ impl<C: SseClient> SseClientTransport<C> {
config: SseClientConfig,
) -> Result<Self, SseTransportError<C::Error>> {
let mut sse_stream = client.get_stream(config.uri.clone(), None, None).await?;
// wait the endpoint event
let endpoint = loop {
let sse = sse_stream
.next()
.await
.ok_or(SseTransportError::UnexpectedEndOfStream)??;
let Some("endpoint") = sse.event.as_deref() else {
continue;
};
break sse.data.unwrap_or_default();
let endpoint = if let Some(endpoint) = config.use_endpoint.clone() {
endpoint
} else {
// wait the endpoint event
loop {
let sse = sse_stream
.next()
.await
.ok_or(SseTransportError::UnexpectedEndOfStream)??;
let Some("endpoint") = sse.event.as_deref() else {
continue;
};
break sse.data.unwrap_or_default();
}
};
let post_uri: Arc<str> = format!(
"{}/{}",
@ -151,6 +155,8 @@ impl<C: SseClient> SseClientTransport<C> {
pub struct SseClientConfig {
pub uri: Arc<str>,
pub retry_policy: Arc<dyn SseRetryPolicy>,
/// if this is settled, the client will use this endpoint to send message and skip get the endpoint event
pub use_endpoint: Option<String>,
}
impl Default for SseClientConfig {
@ -158,6 +164,7 @@ impl Default for SseClientConfig {
Self {
uri: "".into(),
retry_policy: Arc::new(super::common::client_side_sse::FixedInterval::default()),
use_endpoint: None,
}
}
}