refactor: migrate real-time log viewing from WebSocket to SSE with JSON wrapping

internal/controllers/log_sse_controller.go

@@ -0,0 +1,99 @@
+package controllers
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/engigu/baihu-panel/internal/database"
+	"github.com/engigu/baihu-panel/internal/models"
+	"github.com/engigu/baihu-panel/internal/services/tasks"
+	"github.com/engigu/baihu-panel/internal/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type LogSSEController struct{}
+
+func NewLogSSEController() *LogSSEController {
+	return &LogSSEController{}
+}
+
+func (lc *LogSSEController) StreamLog(c *gin.Context) {
+	logIDStr := c.Query("log_id")
+	if logIDStr == "" {
+		c.JSON(400, gin.H{"error": "log_id is required"})
+		return
+	}
+
+	logID := logIDStr
+
+	c.Header("Content-Type", "text/event-stream")
+	c.Header("Cache-Control", "no-cache")
+	c.Header("Connection", "keep-alive")
+	c.Header("Transfer-Encoding", "chunked")
+	// c.Header("Access-Control-Allow-Origin", "*")
+
+	// 1. 检查数据库中是否已结束
+	var taskLog models.TaskLog
+	res := database.DB.Where("id = ?", logID).Limit(1).Find(&taskLog)
+	if res.Error == nil && res.RowsAffected > 0 {
+		if taskLog.Status != "running" {
+			// 已结束，读取库内日志
+			content, err := utils.DecompressFromBase64(string(taskLog.Output))
+			if err != nil {
+				c.SSEvent("message", gin.H{"text": "解压日志失败: " + err.Error()})
+				c.Writer.Flush()
+				return
+			}
+			c.SSEvent("message", gin.H{"text": content})
+			c.Writer.Flush()
+			return
+		}
+	}
+
+	// 2. 未结束或未找到记录，尝试从 TinyLogManager 获取
+	tl := tasks.GetActiveLog(logID)
+	if tl == nil {
+		c.SSEvent("message", gin.H{"text": "未找到正在运行的任务日志"})
+		c.Writer.Flush()
+		return
+	}
+
+	// 发送系统提示
+	c.SSEvent("message", gin.H{"text": fmt.Sprintf("[System] 连接成功，正在监听日志... (LogID: %s)\n", logID)})
+	c.Writer.Flush()
+
+	// 发送最后 100 行
+	lastLines, err := tl.ReadLastLines(100)
+	if err == nil && len(lastLines) > 0 {
+		c.SSEvent("message", gin.H{"text": string(lastLines)})
+		c.Writer.Flush()
+	}
+
+	// 订阅实时更新
+	sub := tl.Subscribe()
+	defer tl.Unsubscribe(sub)
+
+	// 推送更新
+	c.Stream(func(w io.Writer) bool {
+		select {
+		case data, ok := <-sub:
+			if !ok {
+				// 任务结束，尝试刷新最后一次库内完整内容
+				var finalLog models.TaskLog
+				res := database.DB.Where("id = ?", logID).Limit(1).Find(&finalLog)
+				if res.Error == nil && res.RowsAffected > 0 {
+					content, _ := utils.DecompressFromBase64(string(finalLog.Output))
+					if content != "" {
+						c.SSEvent("message", gin.H{"text": "\n--- 任务已结束 ---\n"})
+					}
+				}
+				return false
+			}
+			c.SSEvent("message", gin.H{"text": string(data)})
+			return true
+		case <-c.Request.Context().Done():
+			return false
+		}
+	})
+}

internal/controllers/log_ws_controller.go

@@ -1,122 +0,0 @@
-package controllers
-
-import (
-	"fmt"
-
-	"github.com/engigu/baihu-panel/internal/constant"
-	"github.com/engigu/baihu-panel/internal/database"
-	"github.com/engigu/baihu-panel/internal/models"
-	"github.com/engigu/baihu-panel/internal/services/tasks"
-	"github.com/engigu/baihu-panel/internal/utils"
-
-	"github.com/gin-gonic/gin"
-	"github.com/gorilla/websocket"
-	"time"
-)
-
-type LogWSController struct{}
-
-func NewLogWSController() *LogWSController {
-	return &LogWSController{}
-}
-
-func (lc *LogWSController) StreamLog(c *gin.Context) {
-	logIDStr := c.Query("log_id")
-	if logIDStr == "" {
-		return
-	}
-
-	logID := logIDStr
-
-	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
-	if err != nil {
-		return
-	}
-	defer conn.Close()
-
-	// DoS 保护与心跳设置
-	conn.SetReadLimit(constant.MaxMessageSize) // 使用与终端一致的限制
-	conn.SetReadDeadline(time.Now().Add(constant.PongWait))
-	conn.SetPongHandler(func(string) error {
-		conn.SetReadDeadline(time.Now().Add(constant.PongWait))
-		return nil
-	})
-
-	// 启动一个读取循环，用于处理 pong 和检测断开
-	go func() {
-		for {
-			if _, _, err := conn.ReadMessage(); err != nil {
-				break
-			}
-		}
-	}()
-
-	// 1. 检查数据库中是否已结束
-	var taskLog models.TaskLog
-	res := database.DB.Where("id = ?", logID).Limit(1).Find(&taskLog)
-	if res.Error == nil && res.RowsAffected > 0 {
-		if taskLog.Status != "running" {
-			// 已结束，读取库内日志
-			content, err := utils.DecompressFromBase64(string(taskLog.Output))
-			if err != nil {
-				conn.WriteMessage(websocket.TextMessage, []byte("解压日志失败: "+err.Error()))
-				return
-			}
-			conn.WriteMessage(websocket.TextMessage, []byte(content))
-			return
-		}
-	}
-
-	// 2. 未结束或未找到记录，尝试从 TinyLogManager 获取
-	tl := tasks.GetActiveLog(logID)
-	if tl == nil {
-		conn.WriteMessage(websocket.TextMessage, []byte("未找到正在运行的任务日志"))
-		return
-	}
-
-	// 发送系统提示
-	conn.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("[System] 连接成功，正在监听日志... (LogID: %s)\n", logID)))
-
-	// 发送最后 100 行
-	lastLines, err := tl.ReadLastLines(100)
-	if err == nil && len(lastLines) > 0 {
-		conn.WriteMessage(websocket.TextMessage, lastLines)
-	}
-
-	// 订阅实时更新
-	sub := tl.Subscribe()
-	defer tl.Unsubscribe(sub)
-
-	ticker := time.NewTicker(constant.PingPeriod)
-	defer ticker.Stop()
-
-	// 推送更新
-	for {
-		select {
-		case <-ticker.C:
-			conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
-			if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
-				return
-			}
-		case data, ok := <-sub:
-			if !ok {
-				// 任务结束，尝试刷新最后一次库内完整内容
-				var finalLog models.TaskLog
-				res := database.DB.Where("id = ?", logID).Limit(1).Find(&finalLog)
-				if res.Error == nil && res.RowsAffected > 0 {
-					content, _ := utils.DecompressFromBase64(string(finalLog.Output))
-					if content != "" {
-						conn.WriteMessage(websocket.TextMessage, []byte("\n--- 任务已结束 ---\n"))
-						// 这里可以选择性再推一次完整版，或直接退出
-					}
-				}
-				return
-			}
-			if err := conn.WriteMessage(websocket.TextMessage, data); err != nil {
-				return
-			}
-		case <-c.Request.Context().Done():
-			return
-		}
-	}
-}

internal/router/api_routes.go

@@ -161,7 +161,7 @@ func registerLogRoutes(g *gin.RouterGroup, c *Controllers) {
 	{
 		logs.GET("", c.Log.GetLogs)
 		logs.POST("/clear", c.Log.ClearLogs)
-		logs.GET("/ws", c.LogWS.StreamLog)
+		logs.GET("/sse", c.LogSSE.StreamLog)
 		logs.GET("/:id", c.Log.GetLogDetail)
 		logs.DELETE("/:id", c.Log.DeleteLog)
 	}

internal/router/register.go

@@ -58,7 +58,7 @@ func RegisterControllers() *Controllers {
 		File:         controllers.NewFileController(constant.ScriptsWorkDir),
 		Dashboard:    controllers.NewDashboardController(executorService),
 		Log:          controllers.NewLogController(),
-		LogWS:        controllers.NewLogWSController(),
+		LogSSE:       controllers.NewLogSSEController(),
 		Terminal:     controllers.NewTerminalController(envService),
 		Settings:     controllers.NewSettingsController(userService, loginLogService, executorService),
 		Dependency:   controllers.NewDependencyController(),

internal/router/router.go

@@ -21,7 +21,7 @@ type Controllers struct {
 	File         *controllers.FileController
 	Dashboard    *controllers.DashboardController
 	Log          *controllers.LogController
-	LogWS        *controllers.LogWSController
+	LogSSE       *controllers.LogSSEController
 	Terminal     *controllers.TerminalController
 	Settings     *controllers.SettingsController
 	Dependency   *controllers.DependencyController