zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-07 15:10:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor routes and harden AddTool (#5375)

Browse Source
This commit is contained in:
mk0walsk
2026-01-24 14:13:35 +02:00
committed by GitHub
parent 173f76acf9
commit 94e2f5bd12
3 changed files with 30 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
mcp/server.go
View File

@@ -85,20 +85,7 @@ func (s *mcpServerImpl) setupSSETransport() {
return s.mcpServer
}, nil)
// Register the SSE endpoint
s.httpServer.AddRoute(rest.Route{
Method: http.MethodGet,
Path: s.conf.Mcp.SseEndpoint,
Handler: handler.ServeHTTP,
}, rest.WithSSE(), rest.WithTimeout(s.conf.Mcp.SseTimeout))
// The SSE handler also handles POST requests to message endpoints
// We need to route those as well
s.httpServer.AddRoute(rest.Route{
Method: http.MethodPost,
Path: s.conf.Mcp.SseEndpoint,
Handler: handler.ServeHTTP,
}, rest.WithTimeout(s.conf.Mcp.MessageTimeout))
s.registerRoutes(handler, s.conf.Mcp.SseEndpoint)
}
// setupStreamableTransport configures the server to use Streamable HTTP transport (2025-03-26 spec)
@@ -109,16 +96,20 @@ func (s *mcpServerImpl) setupStreamableTransport() {
return s.mcpServer
}, nil)
// Register the message endpoint (handles both GET for SSE and POST for messages)
s.registerRoutes(handler, s.conf.Mcp.MessageEndpoint)
}
func (s *mcpServerImpl) registerRoutes(handler http.Handler, endpoint string) {
// Register the endpoint (handles both GET for SSE and POST for messages)
s.httpServer.AddRoute(rest.Route{
Method: http.MethodGet,
Path: s.conf.Mcp.MessageEndpoint,
Path: endpoint,
Handler: handler.ServeHTTP,
}, rest.WithSSE(), rest.WithTimeout(s.conf.Mcp.SseTimeout))
s.httpServer.AddRoute(rest.Route{
Method: http.MethodPost,
Path: s.conf.Mcp.MessageEndpoint,
Path: endpoint,
Handler: handler.ServeHTTP,
}, rest.WithTimeout(s.conf.Mcp.MessageTimeout))
}