Revert "New: Global middleware"

Author: jesse-engineer

README.md

@@ -116,13 +116,6 @@ func main() {
 	if err != nil {
 		log.Fatalf("Failed to create MCP server: %v", err)
 	}
-	
-	// Optional: Global middleware (variadic - multiple middleware supported)
-	// mcpServer.Use(
-	// 	LoggingMiddleware,
-	// 	AuthMiddleware,
-	// 	MetricsMiddleware,
-	// )
 
 	// Register time query tool
 	tool, err := protocol.NewTool("current_time", "Get current time for specified timezone", TimeRequest{})

README_CN.md

@@ -118,13 +118,6 @@ func main() {
 		log.Fatalf("创建 MCP 服务器失败: %v", err)
 	}
 
-	// 可选：全局中间件（支持多个中间件参数）
-	// mcpServer.Use(
-	// 	LoggingMiddleware,
-	// 	AuthMiddleware,
-	// 	MetricsMiddleware,
-	// )
-
 	// 注册时间查询工具
 	tool, err := protocol.NewTool("current_time", "获取指定时区的当前时间", TimeRequest{})
 	if err != nil {

README_TW.md

@@ -111,13 +111,6 @@ func main() {
     log.Fatalf("建立 MCP 伺服器失敗: %v", err)
   }
 
-  // 可選：全域中介軟體（支援多個中介軟體參數）
-  // mcpServer.Use(
-  // 	LoggingMiddleware,
-  // 	AuthMiddleware,
-  // 	MetricsMiddleware,
-  // )
-
   // 註冊時間查詢工具
   tool, err := protocol.NewTool("current_time", "取得指定時區的目前時間", TimeRequest{})
   if err != nil {

README_VI.md

@@ -112,13 +112,6 @@ func main() {
 		log.Fatalf("Failed to create MCP server: %v", err)
 	}
 
-	// Optional: Global middleware (variadic - multiple middleware supported)
-	// mcpServer.Use(
-	// 	LoggingMiddleware,
-	// 	AuthMiddleware,
-	// 	MetricsMiddleware,
-	// )
-
 	// Register time query tool
 	tool, err := protocol.NewTool("current_time", "Get current time for specified timezone", TimeRequest{})
 	if err != nil {

examples/middleware_example/main.go

@@ -96,8 +96,6 @@ type Server struct {
 	logger pkg.Logger
 
 	genSessionID func(ctx context.Context) string
-
-	globalMiddlewares []ToolMiddleware
 }
 
 func NewServer(t transport.ServerTransport, opts ...Option) (*Server, error) {
@@ -153,10 +151,7 @@ func (server *Server) RegisterTool(tool *protocol.Tool, toolHandler ToolHandlerF
 	for i := len(middlewares) - 1; i >= 0; i-- {
 		toolHandler = middlewares[i](toolHandler)
 	}
-
-	finalHandler := server.buildMiddlewareChain(toolHandler)
-
-	server.tools.Store(tool.Name, &toolEntry{tool: tool, handler: finalHandler})
+	server.tools.Store(tool.Name, &toolEntry{tool: tool, handler: toolHandler})
 	if !server.sessionManager.IsEmpty() {
 		if err := server.sendNotification4ToolListChanges(context.Background()); err != nil {
 			server.logger.Warnf("send notification toll list changes fail: %v", err)
@@ -258,25 +253,6 @@ func (server *Server) UnregisterResourceTemplate(uriTemplate string) {
 	}
 }
 
-func (server *Server) Use(middlewares ...ToolMiddleware) {
-	server.globalMiddlewares = append(server.globalMiddlewares, middlewares...)
-}
-
-func (server *Server) buildMiddlewareChain(finalHandler ToolHandlerFunc) ToolHandlerFunc {
-	if len(server.globalMiddlewares) == 0 {
-		return finalHandler
-	}
-
-	handler := finalHandler
-	for i := len(server.globalMiddlewares) - 1; i >= 0; i-- {
-		middleware := server.globalMiddlewares[i]
-		currentHandler := handler
-		handler = middleware(currentHandler)
-	}
-
-	return handler
-}
-
 func (server *Server) Shutdown(userCtx context.Context) error {
 	server.inShutdown.Store(true)
 

server/server.go

@@ -2,7 +2,6 @@ package server
 
 import (
 	"bufio"
-	"bytes"
 	"context"
 	"encoding/json"
 	"fmt"
@@ -929,78 +928,3 @@ func testServerRateLimiter(t *testing.T, tt testLimiter) {
 		// No error, continue
 	}
 }
-
-// Middleware tests
-func TestMiddleware(t *testing.T) {
-	mockTransport := transport.NewMockServerTransport(io.NopCloser(bytes.NewReader(nil)), io.Discard)
-
-	s, err := NewServer(mockTransport)
-	if err != nil {
-		t.Fatalf("Failed to create server: %v", err)
-	}
-
-	if s == nil {
-		t.Fatal("Server is nil")
-	}
-
-	testMiddleware := func(next ToolHandlerFunc) ToolHandlerFunc {
-		return func(ctx context.Context, req *protocol.CallToolRequest) (*protocol.CallToolResult, error) {
-			return next(ctx, req)
-		}
-	}
-
-	s.Use(testMiddleware)
-}
-
-// TestMiddlewareEarlyReturn tests middleware that can return early
-func TestMiddlewareEarlyReturn(t *testing.T) {
-	mockTransport := transport.NewMockServerTransport(io.NopCloser(bytes.NewReader(nil)), io.Discard)
-
-	s, err := NewServer(mockTransport)
-	if err != nil {
-		t.Fatalf("Failed to create server: %v", err)
-	}
-
-	if s == nil {
-		t.Fatal("Server is nil")
-	}
-
-	authMiddleware := func(next ToolHandlerFunc) ToolHandlerFunc {
-		return func(ctx context.Context, req *protocol.CallToolRequest) (*protocol.CallToolResult, error) {
-			return nil, context.Canceled
-		}
-	}
-
-	s.Use(authMiddleware)
-}
-
-// TestNoMiddleware tests that tools work without middleware
-func TestNoMiddleware(t *testing.T) {
-	mockTransport := transport.NewMockServerTransport(io.NopCloser(bytes.NewReader(nil)), io.Discard)
-
-	s, err := NewServer(mockTransport)
-	if err != nil {
-		t.Fatalf("Failed to create server: %v", err)
-	}
-
-	if s == nil {
-		t.Fatal("Server is nil")
-	}
-
-	testHandler := func(ctx context.Context, req *protocol.CallToolRequest) (*protocol.CallToolResult, error) {
-		return protocol.NewCallToolResult([]protocol.Content{
-			&protocol.TextContent{Type: "text", Text: "success"},
-		}, false), nil
-	}
-
-	testTool := &protocol.Tool{
-		Name:        "test",
-		Description: "Test tool",
-		InputSchema: protocol.InputSchema{
-			Type:       "object",
-			Properties: map[string]*protocol.Property{},
-		},
-	}
-	s.RegisterTool(testTool, testHandler)
-
-}