Add docs about using custom hooks

Author: jboolean

contrib/mark3labs/mcp-go/README.md

@@ -18,9 +18,10 @@ func main() {
 	defer tracer.Stop()
 
     // Do not use with `server.WithHooks(...)`, as this overwrites the tracing hooks. 
-    // Pass custom hooks via TracingConfig.Hooks instead which in turn is passed to server.WithHooks(...).
+    // To add custom hooks alongside tracing, pass them via TracingConfig.Hooks, e.g.:
+    // mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{Hooks: customHooks})
     srv := server.NewMCPServer("my-server", "1.0.0",
-		mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{}))
+		mcpgotrace.WithTracing(nil))
 }
 ```
 

contrib/mark3labs/mcp-go/example_test.go

@@ -6,17 +6,33 @@
 package mcpgo_test
 
 import (
+	"context"
+
 	mcpgotrace "github.com/DataDog/dd-trace-go/contrib/mark3labs/mcp-go/v2"
 	"github.com/DataDog/dd-trace-go/v2/ddtrace/tracer"
+	"github.com/mark3labs/mcp-go/mcp"
 	"github.com/mark3labs/mcp-go/server"
 )
 
 func Example() {
 	tracer.Start()
 	defer tracer.Stop()
 
-	// Create server hooks and add Datadog tracing
 	srv := server.NewMCPServer("my-server", "1.0.0",
-		mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{}))
+		mcpgotrace.WithTracing(nil))
+	_ = srv
+}
+
+func Example_withCustomHooks() {
+	tracer.Start()
+	defer tracer.Stop()
+
+	customHooks := &server.Hooks{}
+	customHooks.AddBeforeInitialize(func(ctx context.Context, id any, request *mcp.InitializeRequest) {
+		// Your custom logic here
+	})
+
+	srv := server.NewMCPServer("my-server", "1.0.0",
+		mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{Hooks: customHooks}))
 	_ = srv
 }

contrib/mark3labs/mcp-go/option.go

@@ -11,13 +11,33 @@ import (
 
 // The file contains methods for easily adding tracing to a MCP server.
 
+// TracingConfig holds configuration for adding tracing to an MCP server.
 type TracingConfig struct {
+	// Hooks allows you to provide custom hooks that will be merged with Datadog tracing hooks.
+	// If nil, only Datadog tracing hooks will be added and any custom hooks provided via server.WithHooks(...) will be removed.
+	// If provided, your custom hooks will be executed alongside Datadog tracing hooks.
 	Hooks *server.Hooks
 }
 
-// Pass to server.NewMCPServer to add tracing to the server.
+// WithTracing adds Datadog tracing to an MCP server.
+// Pass this option to server.NewMCPServer to enable tracing.
+//
 // Do not use with `server.WithHooks(...)`, as this overwrites the hooks.
-// Pass custom hooks in the TracingConfig instead, which in turn is passed to server.WithHooks(...).
+// Instead, pass custom hooks in the TracingConfig, which will be merged with tracing hooks.
+//
+// Usage:
+//
+//	// Simple usage with only tracing hooks
+//	srv := server.NewMCPServer("my-server", "1.0.0",
+//	    WithTracing(nil))
+//
+//	// With custom hooks
+//	customHooks := &server.Hooks{}
+//	customHooks.AddBeforeInitialize(func(ctx context.Context, id any, request *mcp.InitializeRequest) {
+//	    // Your custom logic here
+//	})
+//	srv := server.NewMCPServer("my-server", "1.0.0",
+//	    WithTracing(&TracingConfig{Hooks: customHooks}))
 func WithTracing(options *TracingConfig) server.ServerOption {
 	return func(s *server.MCPServer) {
 		if options == nil {