examples: Fix SSE streaming issue by moving logging from HTTP to MCP middlewarre (#614)
In `examples/http`, previous HTTP-level logging middleware blocked SSE streaming because `handler.ServeHTTP()` does not return until the connection closes for GET requests (SSE streams). This prevented response headers from being sent to the client. Changes: - Replace HTTP-level logging handler with MCP middleware pattern - Implement logging at the MCP protocol layer using mcp.Middleware - Add middleware to server using AddReceivingMiddleware() - Simplify main.go by removing HTTP handler wrapping This allows SSE streams to work properly while still providing detailed logging for all MCP protocol-level method calls. Fixes #632
This commit is contained in:
parent
d297272705
commit
35dfcf21c9
1 changed files with 46 additions and 43 deletions
|
|
@ -117,6 +117,9 @@ func runServer(url string) {
|
|||
Version: "1.0.0",
|
||||
}, nil)
|
||||
|
||||
// Add MCP-level logging middleware.
|
||||
server.AddReceivingMiddleware(createLoggingMiddleware())
|
||||
|
||||
// Add the cityTime tool.
|
||||
mcp.AddTool(server, &mcp.Tool{
|
||||
Name: "cityTime",
|
||||
|
|
@ -128,13 +131,11 @@ func runServer(url string) {
|
|||
return server
|
||||
}, nil)
|
||||
|
||||
handlerWithLogging := loggingHandler(handler)
|
||||
|
||||
log.Printf("MCP server listening on %s", url)
|
||||
log.Printf("Available tool: cityTime (cities: nyc, sf, boston)")
|
||||
|
||||
// Start the HTTP server with logging handler.
|
||||
if err := http.ListenAndServe(url, handlerWithLogging); err != nil {
|
||||
// Start the HTTP server.
|
||||
if err := http.ListenAndServe(url, handler); err != nil {
|
||||
log.Fatalf("Server failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue