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

chore: disable unix_socket example on windows (#146)

Browse source
This commit is contained in:
Eucalypt 2025-04-26 14:27:15 +08:00 • committed by GitHub
commit 6c2c02454d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 62 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

119
examples/transport/src/unix_socket.rs
View file

@ -1,15 +1,66 @@
use std::fs;
use common::calculator::Calculator;
use rmcp::{serve_client, serve_server};
use tokio::net::{UnixListener, UnixStream};
mod common;
const SOCKET_PATH: &str = "/tmp/rmcp_example.sock";
#[cfg(target_family = "unix")]
#[tokio::main]
async fn main() -> anyhow::Result<()> {
use std::fs;
use common::calculator::Calculator;
use rmcp::{serve_client, serve_server};
use tokio::net::{UnixListener, UnixStream};
const SOCKET_PATH: &str = "/tmp/rmcp_example.sock";
async fn server(unix_listener: UnixListener) -> anyhow::Result<()> {
while let Ok((stream, addr)) = unix_listener.accept().await {
println!("Client connected: {:?}", addr);
tokio::spawn(async move {
match serve_server(Calculator, stream).await {
Ok(server) => {
println!("Server initialized successfully");
if let Err(e) = server.waiting().await {
println!("Error while server waiting: {}", e);
}
}
Err(e) => println!("Server initialization failed: {}", e),
}
anyhow::Ok(())
});
}
Ok(())
}
async fn client() -> anyhow::Result<()> {
println!("Client connecting to {}", SOCKET_PATH);
let stream = UnixStream::connect(SOCKET_PATH).await?;
let client = serve_client((), stream).await?;
println!("Client connected and initialized successfully");
// List available tools
let tools = client.peer().list_tools(Default::default()).await?;
println!("Available tools: {:?}", tools);
// Call the sum tool
if let Some(sum_tool) = tools.tools.iter().find(|t| t.name.contains("sum")) {
println!("Calling sum tool: {}", sum_tool.name);
let result = client
.peer()
.call_tool(rmcp::model::CallToolRequestParam {
name: sum_tool.name.clone(),
arguments: Some(rmcp::object!({
"a": 10,
"b": 20
})),
})
.await?;
println!("Result: {:?}", result);
}
Ok(())
}
// Remove any existing socket file
let _ = fs::remove_file(SOCKET_PATH);
match UnixListener::bind(SOCKET_PATH) {
@ -30,53 +81,7 @@ async fn main() -> anyhow::Result<()> {
Ok(())
}
async fn server(unix_listener: UnixListener) -> anyhow::Result<()> {
while let Ok((stream, addr)) = unix_listener.accept().await {
println!("Client connected: {:?}", addr);
tokio::spawn(async move {
match serve_server(Calculator, stream).await {
Ok(server) => {
println!("Server initialized successfully");
if let Err(e) = server.waiting().await {
println!("Error while server waiting: {}", e);
}
}
Err(e) => println!("Server initialization failed: {}", e),
}
anyhow::Ok(())
});
}
Ok(())
}
async fn client() -> anyhow::Result<()> {
println!("Client connecting to {}", SOCKET_PATH);
let stream = UnixStream::connect(SOCKET_PATH).await?;
let client = serve_client((), stream).await?;
println!("Client connected and initialized successfully");
// List available tools
let tools = client.peer().list_tools(Default::default()).await?;
println!("Available tools: {:?}", tools);
// Call the sum tool
if let Some(sum_tool) = tools.tools.iter().find(|t| t.name.contains("sum")) {
println!("Calling sum tool: {}", sum_tool.name);
let result = client
.peer()
.call_tool(rmcp::model::CallToolRequestParam {
name: sum_tool.name.clone(),
arguments: Some(rmcp::object!({
"a": 10,
"b": 20
})),
})
.await?;
println!("Result: {:?}", result);
}
Ok(())
#[cfg(not(target_family = "unix"))]
fn main() {
println!("Unix socket example is not supported on this platform.");
}