optimize hooks

Author: dugenkui03

examples/everything/main.go

@@ -28,31 +28,93 @@ const (
 	COMPLEX PromptName = "complex_prompt"
 )
 
+type exampleHook struct{}
+
+func (*exampleHook) RegisterSession(ctx context.Context, session server.ClientSession) {
+}
+
+func (*exampleHook) CreateHookContext() server.HookContext {
+	return nil
+}
+
+func (*exampleHook) GetHookContext() server.HookContext {
+	return nil
+}
+
+func (*exampleHook) BeforeAny(ctx context.Context, hookContext server.HookContext, id any, method mcp.MCPMethod, message any) {
+	fmt.Printf("beforeAny: %s, %v, %v\n", method, id, message)
+}
+
+func (*exampleHook) OnSuccess(ctx context.Context, hookContext server.HookContext, id any, method mcp.MCPMethod, message any, result any) {
+	fmt.Printf("onSuccess: %s, %v, %v, %v\n", method, id, message, result)
+}
+
+func (*exampleHook) OnError(ctx context.Context, hookContext server.HookContext, id any, method mcp.MCPMethod, message any, err error) {
+	fmt.Printf("onError: %s, %v, %v, %v\n", method, id, message, err)
+}
+
+func (*exampleHook) BeforeInitialize(ctx context.Context, hookContext server.HookContext, id any, message *mcp.InitializeRequest) {
+	fmt.Printf("beforeInitialize: %v, %v\n", id, message)
+}
+
+func (*exampleHook) AfterInitialize(ctx context.Context, hookContext server.HookContext, id any, message *mcp.InitializeRequest, result *mcp.InitializeResult) {
+	fmt.Printf("afterInitialize: %v, %v, %v\n", id, message, result)
+}
+
+func (*exampleHook) BeforePing(ctx context.Context, hookContext server.HookContext, id any, message *mcp.PingRequest) {
+}
+
+func (*exampleHook) AfterPing(ctx context.Context, hookContext server.HookContext, id any, message *mcp.PingRequest, result *mcp.EmptyResult) {
+}
+
+func (*exampleHook) BeforeListResources(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListResourcesRequest) {
+}
+
+func (*exampleHook) AfterListResources(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListResourcesRequest, result *mcp.ListResourcesResult) {
+}
+
+func (*exampleHook) BeforeListResourceTemplates(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListResourceTemplatesRequest) {
+}
+
+func (*exampleHook) AfterListResourceTemplates(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListResourceTemplatesRequest, result *mcp.ListResourceTemplatesResult) {
+}
+
+func (*exampleHook) BeforeReadResource(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ReadResourceRequest) {
+}
+
+func (*exampleHook) AfterReadResource(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ReadResourceRequest, result *mcp.ReadResourceResult) {
+}
+
+func (*exampleHook) BeforeListPrompts(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListPromptsRequest) {
+}
+
+func (*exampleHook) AfterListPrompts(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListPromptsRequest, result *mcp.ListPromptsResult) {
+}
+
+func (*exampleHook) BeforeGetPrompt(ctx context.Context, hookContext server.HookContext, id any, message *mcp.GetPromptRequest) {
+}
+
+func (*exampleHook) AfterGetPrompt(ctx context.Context, hookContext server.HookContext, id any, message *mcp.GetPromptRequest, result *mcp.GetPromptResult) {
+}
+
+func (*exampleHook) BeforeListTools(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListToolsRequest) {
+	fmt.Printf("beforeCallTool: %v, %v\n", id, message)
+}
+
+func (*exampleHook) AfterListTools(ctx context.Context, hookContext server.HookContext, id any, message *mcp.ListToolsRequest, result *mcp.ListToolsResult) {
+	fmt.Printf("afterCallTool: %v, %v, %v\n", id, message, result)
+}
+
+func (*exampleHook) BeforeCallTool(ctx context.Context, hookContext server.HookContext, id any, message *mcp.CallToolRequest) {
+}
+
+func (*exampleHook) AfterCallTool(ctx context.Context, hookContext server.HookContext, id any, message *mcp.CallToolRequest, result *mcp.CallToolResult) {
+}
+
 func NewMCPServer() *server.MCPServer {
 
-	hooks := &server.Hooks{}
-
-	hooks.AddBeforeAny(func(ctx context.Context, id any, method mcp.MCPMethod, message any) {
-		fmt.Printf("beforeAny: %s, %v, %v\n", method, id, message)
-	})
-	hooks.AddOnSuccess(func(ctx context.Context, id any, method mcp.MCPMethod, message any, result any) {
-		fmt.Printf("onSuccess: %s, %v, %v, %v\n", method, id, message, result)
-	})
-	hooks.AddOnError(func(ctx context.Context, id any, method mcp.MCPMethod, message any, err error) {
-		fmt.Printf("onError: %s, %v, %v, %v\n", method, id, message, err)
-	})
-	hooks.AddBeforeInitialize(func(ctx context.Context, id any, message *mcp.InitializeRequest) {
-		fmt.Printf("beforeInitialize: %v, %v\n", id, message)
-	})
-	hooks.AddAfterInitialize(func(ctx context.Context, id any, message *mcp.InitializeRequest, result *mcp.InitializeResult) {
-		fmt.Printf("afterInitialize: %v, %v, %v\n", id, message, result)
-	})
-	hooks.AddAfterCallTool(func(ctx context.Context, id any, message *mcp.CallToolRequest, result *mcp.CallToolResult) {
-		fmt.Printf("afterCallTool: %v, %v, %v\n", id, message, result)
-	})
-	hooks.AddBeforeCallTool(func(ctx context.Context, id any, message *mcp.CallToolRequest) {
-		fmt.Printf("beforeCallTool: %v, %v\n", id, message)
-	})
+	hooks := &server.ChainedHook{}
+	hooks.Add(&exampleHook{})
 
 	mcpServer := server.NewMCPServer(
 		"example-servers/everything",