yggdrasil/mcp/
server.rs

1//! MCP 服务器：rmcp `ServerHandler` 实现，组合所有工具组。
2//!
3//! 装配模式（rmcp `tool_router` 宏支持的「命名路由 + 组合」）：
4//! - 每个工具组在各自文件（`tools/<x>.rs`）里用
5//!   `#[tool_router(router = <x>_router, vis = "pub")] impl YggMcpServer { ... }`
6//!   生成一个返回 `ToolRouter<YggMcpServer>` 的公开函数。
7//! - 本文件定义 `YggMcpServer` 结构体，并在 `#[tool_handler(router = ...)]`
8//!   里把所有路由用 `+` 合并成单一 `ServerHandler`。
9//!
10//! 所有工具都在 `YggMcpServer` 上 `impl`，故 `ToolRouter` 的类型参数一致，
11//! 可用 `+`（`Add` for `ToolRouter<S>`）合并。鉴权与作用域校验在各工具内部
12//! 经 `Extension<http::request::Parts>` 读取 `McpPrincipal` 完成。
13
14use rmcp::{tool_handler, ServerHandler};
15
16/// MCP 服务器实例（无状态：每个请求由 service_factory 新建一份）。
17///
18/// 共享状态（DB 连接等）通过 `get_conn()` 全局池获取，无需在实例里持有。
19/// 工具方法分散在各 `tools/*.rs` 的 `impl YggMcpServer` 块里。
20#[derive(Clone, Default)]
21pub struct YggMcpServer;
22
23impl YggMcpServer {
24    /// 组合所有工具组的路由表。新增工具组时在此追加一行。
25    fn combined_router() -> rmcp::handler::server::router::tool::ToolRouter<Self> {
26        YggMcpServer::read_router()
27            + YggMcpServer::posts_router()
28            + YggMcpServer::comments_router()
29            + YggMcpServer::tags_router()
30            + YggMcpServer::media_router()
31            + YggMcpServer::settings_router()
32            + YggMcpServer::runner_router()
33    }
34}
35
36/// 单一 `ServerHandler`：工具调度委托给合并后的路由表。
37/// `router = Self::combined_router()` 让 `tool_handler` 宏生成的
38/// `call_tool`/`list_tools`/`get_tool` 全部走组合路由。
39#[tool_handler(router = Self::combined_router())]
40impl ServerHandler for YggMcpServer {
41    // get_info 由宏自动生成（name/version 来自宏的默认或属性）。
42    // 这里不手写 get_info，让宏用默认 ServerInfo。
43}
yggdrasil/mcp/server.rs

yggdrasil/mcp/
server.rs
