portugalfuturista/rust-sdk
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pin transport once, use project in write_message (#28)

Browse source
This commit is contained in:
Marc-Andre Giroux 2025-03-18 00:52:10 -04:00 committed by GitHub
parent b1ed4e7204
commit 042299f251
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
crates/mcp-server/src/lib.rs
View file

@ -105,13 +105,17 @@ where
R: AsyncRead + Unpin,
W: AsyncWrite + Unpin,
{
pub async fn write_message(&mut self, msg: JsonRpcMessage) -> Result<(), std::io::Error> {
pub async fn write_message(
self: &mut Pin<&mut Self>,
msg: JsonRpcMessage,
) -> Result<(), std::io::Error> {
let json = serde_json::to_string(&msg)?;
Pin::new(&mut self.writer)
.write_all(json.as_bytes())
.await?;
Pin::new(&mut self.writer).write_all(b"\n").await?;
Pin::new(&mut self.writer).flush().await?;
let mut this = self.as_mut().project();
this.writer.write_all(json.as_bytes()).await?;
this.writer.write_all(b"\n").await?;
this.writer.flush().await?;
Ok(())
}
}
@ -139,6 +143,7 @@ where
{
use futures::StreamExt;
let mut service = self.service;
let mut transport = Pin::new(&mut transport);
tracing::info!("Server started");
while let Some(msg_result) = transport.next().await {