使用Rust编写MCP服务

2026-01-20

字数统计: 325字 | 阅读时长: 1分

本次实验是使用Rust编译一个时间日期的MCP服务。

项目依赖：

tokio = { version = "1.49", features = ["rt", "rt-multi-thread", "macros"] }
rmcp = { version = "0.13", features = ["server", "schemars", "transport-streamable-http-server", "transport-streamable-http-server-session", "transport-worker"] }
axum = "0.8.8"
chrono = "0.4"
serde_json = "1"

编写main.rs文件：

use chrono::Utc;
use rmcp::handler::server::{tool::ToolRouter, ServerHandler};
use rmcp::transport::streamable_http_server::session::local::LocalSessionManager;
use rmcp::transport::streamable_http_server::{StreamableHttpServerConfig, StreamableHttpService};
use rmcp::{tool, tool_handler, tool_router};
use std::sync::Arc;

#[derive(Clone)]
struct TimeService {
    tool_router: ToolRouter<Self>,
}

#[tool_router(router=tool_router)]
impl TimeService {
    fn new() -> Self {
        Self {
            tool_router: Self::tool_router(),
        }
    }

    #[tool(description = "获取当前时间，返回北京时间(UTC+8)")]
    async fn get_time(&self) -> Result<String, rmcp::ErrorData> {
        let now = Utc::now();

        let beijing_offset = chrono::FixedOffset::east_opt(8 * 3600).unwrap();
        let beijing_time = now.with_timezone(&beijing_offset);

        Ok(beijing_time.to_rfc3339())
    }
}

#[tool_handler(router=self.tool_router)]
impl ServerHandler for TimeService {}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let service = StreamableHttpService::new(
        || Ok(Arc::new(TimeService::new())),
        LocalSessionManager::default().into(),
        StreamableHttpServerConfig::default(),
    );

    let addr = std::env::var("HOST").unwrap_or("0.0.0.0:1234".to_string());
    let router = axum::Router::new().nest_service("/mcp", service);
    let tcp_listener = tokio::net::TcpListener::bind(addr).await?;

    let _ = axum::serve(tcp_listener, router).await;
    Ok(())
}

然后，将http://127.0.0.1:1234/mcp配置到支持MCP的AI工具中即可测试。

本文作者： killf
本文链接： https://www.killf.info/编程语言/Rust/使用Rust编写MCP服务/
版权声明： 本博客所有文章除特别声明外，均采用 Apache License 2.0 许可协议。转载请注明出处！