53ea832773
- 添加官方 SDK 依赖 github.com/modelcontextprotocol/go-sdk v0.7.0 - 新增 mcp_server.go 使用官方 SDK 注册 8 个 MCP 工具 - 删除自实现的 streamable_http.go（约 400 行） - 更新 routes.go 使用 mcp.NewStreamableHTTPHandler - 优化服务器优雅关闭逻辑（5秒超时 + 警告日志） - 清理 types.go 中的 JSON-RPC 相关类型 🤖 Generated with [Claude Code](https://claude.ai/code) Co-authored-by: Claude <noreply@anthropic.com>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
)
|
|
|
|
// setupRoutes 设置路由配置
|
|
func setupRoutes(appServer *AppServer) *gin.Engine {
|
|
// 设置 Gin 模式
|
|
gin.SetMode(gin.ReleaseMode)
|
|
|
|
router := gin.New()
|
|
router.Use(gin.Logger())
|
|
router.Use(gin.Recovery())
|
|
|
|
// 添加中间件
|
|
router.Use(errorHandlingMiddleware())
|
|
router.Use(corsMiddleware())
|
|
|
|
// 健康检查
|
|
router.GET("/health", healthHandler)
|
|
|
|
// MCP 端点 - 使用官方 SDK 的 Streamable HTTP Handler
|
|
mcpHandler := mcp.NewStreamableHTTPHandler(
|
|
func(r *http.Request) *mcp.Server {
|
|
return appServer.mcpServer
|
|
},
|
|
&mcp.StreamableHTTPOptions{
|
|
JSONResponse: true, // 支持 JSON 响应
|
|
},
|
|
)
|
|
router.Any("/mcp", gin.WrapH(mcpHandler))
|
|
router.Any("/mcp/*path", gin.WrapH(mcpHandler))
|
|
|
|
// API 路由组
|
|
api := router.Group("/api/v1")
|
|
{
|
|
api.GET("/login/status", appServer.checkLoginStatusHandler)
|
|
api.GET("/login/qrcode", appServer.getLoginQrcodeHandler)
|
|
api.POST("/publish", appServer.publishHandler)
|
|
api.GET("/feeds/list", appServer.listFeedsHandler)
|
|
api.GET("/feeds/search", appServer.searchFeedsHandler)
|
|
api.POST("/feeds/detail", appServer.getFeedDetailHandler)
|
|
api.POST("/user/profile", appServer.userProfileHandler)
|
|
api.POST("/feeds/comment", appServer.postCommentHandler)
|
|
}
|
|
|
|
return router
|
|
}
|