address review feedback

Author: rwjblue-glean

mcp/example_middleware_test.go

@@ -20,32 +20,30 @@ func Example_loggingMiddleware() {
 	// Create a logger for demonstration purposes.
 	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
 		Level: slog.LevelInfo,
-		ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
-			// Simplify timestamp format for consistent output
+		ReplaceAttr: func(_ []string, a slog.Attr) slog.Attr {
+			// Simplify timestamp format for consistent output.
 			if a.Key == slog.TimeKey {
 				return slog.String("time", "2025-01-01T00:00:00Z")
 			}
 			return a
 		},
 	}))
 
-	// Create logging middleware
 	loggingMiddleware := func(next mcp.MethodHandler[*mcp.ServerSession]) mcp.MethodHandler[*mcp.ServerSession] {
 		return func(
 			ctx context.Context,
 			session *mcp.ServerSession,
 			method string,
 			params mcp.Params,
 		) (mcp.Result, error) {
-			start := time.Now()
-
 			logger.Info("MCP method started",
 				"method", method,
 				"session_id", session.ID(),
 				"has_params", params != nil,
 			)
 
-			// Call the actual handler
+			start := time.Now()
+
 			result, err := next(ctx, session, method, params)
 
 			duration := time.Since(start)