update: update

Author: xmapst

docs/docs.go

@@ -493,7 +493,7 @@ const docTemplate = `{
                 "tags": [
                     "工作池"
                 ],
-                "summary": "详情",
+                "summary": "详情, 支持SSE订阅",
                 "responses": {
                     "200": {
                         "description": "OK",

docs/swagger.json

@@ -485,7 +485,7 @@
                 "tags": [
                     "工作池"
                 ],
-                "summary": "详情",
+                "summary": "详情, 支持SSE订阅",
                 "responses": {
                     "200": {
                         "description": "OK",

docs/swagger.yaml

@@ -866,7 +866,7 @@ paths:
           description: Internal Server Error
           schema:
             $ref: '#/definitions/types.SBase-any'
-      summary: 详情
+      summary: 详情, 支持SSE订阅
       tags:
       - 工作池
     post:

internal/server/api/v1/pool/get.go

@@ -1,15 +1,20 @@
 package pool
 
 import (
+	"io"
+	"reflect"
+	"time"
+
 	"github.com/gin-gonic/gin"
 
 	"github.com/xmapst/AutoExecFlow/internal/server/api/base"
 	"github.com/xmapst/AutoExecFlow/internal/service"
+	"github.com/xmapst/AutoExecFlow/types"
 	_ "github.com/xmapst/AutoExecFlow/types"
 )
 
 // Detail
-// @Summary		详情
+// @Summary		详情, 支持SSE订阅
 // @Description	获取工作池信息
 // @Tags		工作池
 // @Accept		application/json
@@ -18,5 +23,32 @@ import (
 // @Failure		500 {object} types.SBase[any]
 // @Router		/api/v1/pool [get]
 func Detail(c *gin.Context) {
-	base.Send(c, base.WithData(service.Pool().Get()))
+	if c.GetHeader("Accept") != base.EventStreamMimeType {
+		base.Send(c, base.WithData(service.Pool().Get()))
+		return
+	}
+
+	ticker := time.NewTicker(30 * time.Second) // 每30秒发送心跳
+	defer ticker.Stop()
+
+	var last *types.SPoolRes
+	c.Stream(func(w io.Writer) bool {
+		select {
+		case <-ticker.C:
+			c.SSEvent("heartbeat", "keepalive")
+			return true
+		case <-c.Done():
+			return false
+		default:
+			current := service.Pool().Get()
+			if reflect.DeepEqual(last, current) {
+				time.Sleep(1 * time.Second)
+				return true
+			}
+			c.SSEvent("message", base.WithData(current))
+			last = current
+			time.Sleep(1 * time.Second)
+			return true
+		}
+	})
 }