feat(tool): 添加 Docker 日志实时跟踪功能

- 扩展 DockerTool 支持实时日志跟踪模式
- 新增 follow 参数用于启用持续日志输出
- 实现 SSE 流式推送日志到客户端
- 添加后台协程处理日志流和错误输出
- 集成 server 通知机制推送实时日志行
- 优化命令执行和上下文取消逻辑

internal/tool/docker.go

@@ -1,6 +1,7 @@
 package tool
 
 import (
+	"bufio"
 	"bytes"
 	"context"
 	"fmt"
@@ -12,7 +13,7 @@ import (
 	"github.com/mark3labs/mcp-go/server"
 )
 
-// DockerTool 提供 docker logs 查询。
+// DockerTool 提供 docker logs 查询与实时跟踪。
 type DockerTool struct{}
 
 func NewDockerTool() *DockerTool {
@@ -20,7 +21,7 @@ func NewDockerTool() *DockerTool {
 }
 
 func (d *DockerTool) Name() string        { return "docker" }
-func (d *DockerTool) Description() string { return "Docker 容器日志查询" }
+func (d *DockerTool) Description() string { return "Docker 容器日志查询与实时跟踪" }
 
 func (d *DockerTool) Initialize(_ context.Context) error {
 	if _, err := exec.LookPath("docker"); err != nil {
@@ -38,7 +39,7 @@ func (d *DockerTool) HealthCheck(ctx context.Context) error {
 
 func (d *DockerTool) Register(mcpServer *server.MCPServer) error {
 	mcpServer.AddTool(mcp.NewTool("docker_logs",
-		mcp.WithDescription("获取 Docker 容器日志，等价于 docker logs --tail N <container>"),
+		mcp.WithDescription("获取 Docker 容器日志，等价于 docker logs --tail N <container>；follow 模式通过 SSE 实时推送日志流"),
 		mcp.WithString("container",
 			mcp.Required(),
 			mcp.Description("容器名称或 ID"),
@@ -46,6 +47,9 @@ func (d *DockerTool) Register(mcpServer *server.MCPServer) error {
 		mcp.WithNumber("tail",
 			mcp.Description("返回最后 N 行日志，默认 100"),
 		),
+		mcp.WithBoolean("follow",
+			mcp.Description("是否持续跟踪日志输出（类似 docker logs -f），默认 false。SSE 模式下日志行通过 notifications/docker_logs/stream 推送"),
+		),
 	), d.handleLogs)
 
 	return nil
@@ -63,6 +67,18 @@ func (d *DockerTool) handleLogs(ctx context.Context, req mcp.CallToolRequest) (*
 		tail = int(t)
 	}
 
+	follow := false
+	if f, ok := args["follow"].(bool); ok {
+		follow = f
+	}
+
+	if !follow {
+		return d.oneShotLogs(ctx, container, tail)
+	}
+	return d.followLogs(ctx, container, tail)
+}
+
+func (d *DockerTool) oneShotLogs(ctx context.Context, container string, tail int) (*mcp.CallToolResult, error) {
 	cmd := exec.CommandContext(ctx, "docker", "logs",
 		"--tail", strconv.Itoa(tail),
 		"--timestamps",
@@ -86,3 +102,82 @@ func (d *DockerTool) handleLogs(ctx context.Context, req mcp.CallToolRequest) (*
 
 	return mcp.NewToolResultText(output), nil
 }
+
+func (d *DockerTool) followLogs(ctx context.Context, container string, tail int) (*mcp.CallToolResult, error) {
+	srv := server.ServerFromContext(ctx)
+
+	followCtx, cancelFollow := context.WithCancel(context.Background())
+
+	cmd := exec.CommandContext(followCtx, "docker", "logs",
+		"--tail", strconv.Itoa(tail),
+		"--timestamps",
+		"-f",
+		container,
+	)
+
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		cancelFollow()
+		return mcp.NewToolResultError(fmt.Sprintf("创建管道失败: %v", err)), nil
+	}
+	stderr, err := cmd.StderrPipe()
+	if err != nil {
+		cancelFollow()
+		return mcp.NewToolResultError(fmt.Sprintf("创建管道失败: %v", err)), nil
+	}
+
+	if err := cmd.Start(); err != nil {
+		cancelFollow()
+		return mcp.NewToolResultError(fmt.Sprintf("docker logs 启动失败: %v", err)), nil
+	}
+
+	// 后台读取 stderr
+	go func() {
+		scanner := bufio.NewScanner(stderr)
+		for scanner.Scan() {
+			slog.Warn("docker logs stderr", "container", container, "line", scanner.Text())
+		}
+		if err := scanner.Err(); err != nil {
+			slog.Warn("docker logs stderr scanner error", "container", container, "err", err)
+		}
+	}()
+
+	// 后台流式推送 stdout
+	go func() {
+		defer cancelFollow()
+		defer func() {
+			if err := cmd.Wait(); err != nil {
+				slog.Warn("docker logs -f exited with error", "container", container, "err", err)
+			}
+		}()
+
+		scanner := bufio.NewScanner(stdout)
+		for scanner.Scan() {
+			line := scanner.Text()
+			if srv == nil {
+				continue
+			}
+			if err := srv.SendNotificationToClient(ctx, "notifications/docker_logs/stream", map[string]any{
+				"container": container,
+				"line":      line,
+			}); err != nil {
+				slog.Debug("docker logs notification failed, stopping stream", "err", err)
+				return
+			}
+		}
+		if err := scanner.Err(); err != nil {
+			slog.Warn("docker logs stdout scanner error", "container", container, "err", err)
+		}
+
+		if srv != nil {
+			_ = srv.SendNotificationToClient(ctx, "notifications/docker_logs/stream", map[string]any{
+				"container": container,
+				"line":      "--- 日志流结束 ---",
+			})
+		}
+	}()
+
+	return mcp.NewToolResultText(
+		fmt.Sprintf("开始跟踪容器 %s 日志（tail=%d, follow=true）…", container, tail),
+	), nil
+}