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: throw initialize error detail (#192)

Browse source
This commit is contained in:
4t145 2025-05-19 09:41:56 +08:00 • committed by GitHub
commit 817e62d895
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 115 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crates/rmcp/src/service.rs
View file

@ -77,6 +77,7 @@ pub trait ServiceRole: std::fmt::Debug + Send + Sync + 'static + Copy + Clone {
type PeerNot: TryInto<CancelledNotification, Error = Self::PeerNot>
+ From<CancelledNotification>
+ TransferObject;
type InitializeError<E>;
const IS_CLIENT: bool;
type Info: TransferObject;
type PeerInfo: TransferObject;
@ -113,7 +114,7 @@ pub trait ServiceExt<R: ServiceRole>: Service<R> + Sized {
fn serve<T, E, A>(
self,
transport: T,
) -> impl Future<Output = Result<RunningService<R, Self>, E>> + Send
) -> impl Future<Output = Result<RunningService<R, Self>, R::InitializeError<E>>> + Send
where
T: IntoTransport<R, E, A>,
E: std::error::Error + From<std::io::Error> + Send + Sync + 'static,
@ -125,7 +126,7 @@ pub trait ServiceExt<R: ServiceRole>: Service<R> + Sized {
self,
transport: T,
ct: CancellationToken,
) -> impl Future<Output = Result<RunningService<R, Self>, E>> + Send
) -> impl Future<Output = Result<RunningService<R, Self>, R::InitializeError<E>>> + Send
where
T: IntoTransport<R, E, A>,
E: std::error::Error + From<std::io::Error> + Send + Sync + 'static,
@ -469,7 +470,7 @@ pub async fn serve_directly<R, S, T, E, A>(
service: S,
transport: T,
peer_info: R::PeerInfo,
) -> Result<RunningService<R, S>, E>
) -> RunningService<R, S>
where
R: ServiceRole,
S: Service<R>,
@ -485,7 +486,7 @@ pub async fn serve_directly_with_ct<R, S, T, E, A>(
transport: T,
peer_info: R::PeerInfo,
ct: CancellationToken,
) -> Result<RunningService<R, S>, E>
) -> RunningService<R, S>
where
R: ServiceRole,
S: Service<R>,
@ -503,7 +504,7 @@ async fn serve_inner<R, S, T, E, A>(
peer: Peer<R>,
mut peer_rx: tokio::sync::mpsc::Receiver<PeerSinkMessage<R>>,
ct: CancellationToken,
) -> Result<RunningService<R, S>, E>
) -> RunningService<R, S>
where
R: ServiceRole,
S: Service<R>,
@ -788,10 +789,10 @@ where
tracing::info!(?quit_reason, "serve finished");
quit_reason
});
Ok(RunningService {
RunningService {
service,
peer: peer_return,
handle,
dg: ct.drop_guard(),
})
}
}