🚀 MCP-RS

MCP-RS是一个用Rust实现的MCP（模型上下文协议）服务器。它是原始TypeScript实现此处的Rust端口，为MCP协议提供了Rust版本的高效实现方案。

✨ 主要特性

此Rust实现提供了所有MCP核心功能：

• 具备所有核心MCP服务器功能，保障基础服务稳定运行。
• 内置身份验证处理程序，增强系统安全性。
• 支持自定义分层中间件，可灵活扩展功能。
• 提供灵活的传输选项，包括标准输入输出和SSE，满足不同场景需求。

🚀 快速开始

📦 安装指南

在Cargo.toml中添加以下内容：

[dependencies]
mcp-rs = "0.1.0"

💻 使用示例

基础用法

以下是使用该库的基本示例：

use mcp_rs::{Emcp, SimpleToolBuilder};
use serde_json::json;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建一个新的服务器实例
    let server = Emcp::new("my-mcp-server", "1.0.0", None);
    
    // 添加一个工具
    let add_tool = SimpleToolBuilder::new("add", |args| async move {
        let a = args["a"].as_f64().unwrap_or(0.0);
        let b = args["b"].as_f64().unwrap_or(0.0);
        Ok(json!(a + b))
    })
    .description("Add two numbers")
    .input_schema(json!({
        "type": "object",
        "properties": {
            "a": { "type": "number" },
            "b": { "type": "number" }
        },
        "required": ["a", "b"]
    }))
    .build();
    
    server.add_tool(add_tool);
    
    // 启动服务器
    server.start(None).await?;
    
    Ok(())
}

高级用法 - 身份验证示例

use mcp_rs::{Emcp, EmcpOptions, SimpleToolBuilder};
use serde_json::json;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建身份验证处理程序
    let auth_handler = Box::new(|request| {
        // 检查认证头、令牌等。
        Box::pin(async { 
            // 返回true表示请求被授权
            true 
        })
    });
    
    // 使用身份验证处理程序创建一个新的服务器
    let server = Emcp::new(
        "mcp-server-with-auth",
        "1.0.0",
        Some(EmcpOptions {
            authentication_handler: Some(auth_handler),
        }),
    );
    
    // 添加工具、资源或提示...
    
    // 启动服务器
    server.start(None).await?;
    
    Ok(())
}

高级用法 - 中间件示例

use mcp_rs::{Emcp, SimpleToolBuilder};
use std::time::Instant;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建一个新的服务器
    let mut server = Emcp::new("mcp-server-with-middleware", "1.0.0", None);
    
    // 添加计时中间件
    server.use_middleware(|request, next| {
        let start_time = Instant::now();
        let method = request.method.clone();
        
        Box::pin(async move {
            // 处理逻辑...
            
            Ok(())
        })
    });
    
    // 启动服务器
    server.start(None).await?;
    
    Ok(())
}

运行示例

仓库包含几个演示库功能的示例：

# 运行基本示例
cargo run --example basic

# 运行身份验证示例
cargo run --example auth

# 运行中间件示例
cargo run --example middleware

# 运行标准IO传输示例
cargo run --example stdio_transport

📄 许可证

本项目采用MIT许可证。