64 lines
2.1 KiB
Go
64 lines
2.1 KiB
Go
package server
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"agent/internal/agent"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// coopTaskView 是返回给前端的协作任务视图，附带容器的实时状态与协作轮次进度。
|
|
type coopTaskView struct {
|
|
ID string `json:"id"`
|
|
SessionID string `json:"session_id"`
|
|
Status string `json:"status"` // running | done
|
|
Error string `json:"error,omitempty"`
|
|
ExitCode int `json:"exit_code,omitempty"`
|
|
Blackboard string `json:"blackboard,omitempty"`
|
|
RoundMax int `json:"round_max"`
|
|
CurrentRound int `json:"current_round"`
|
|
ContainerState string `json:"container_state,omitempty"`
|
|
ContainerStatus string `json:"container_status,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
FinishedAt time.Time `json:"finished_at,omitempty"`
|
|
}
|
|
|
|
// coopTasks 返回协作任务列表，支持 ?session=<id> 过滤到某个会话。
|
|
func (s *Server) coopTasks(c *gin.Context) {
|
|
tasks := s.agent.CoopTasks(strings.TrimSpace(c.Query("session")))
|
|
views := make([]coopTaskView, 0, len(tasks))
|
|
for _, task := range tasks {
|
|
views = append(views, s.coopTaskView(task))
|
|
}
|
|
ok(c, gin.H{"tasks": views})
|
|
}
|
|
|
|
func (s *Server) coopTaskView(task *agent.CoopTask) coopTaskView {
|
|
view := coopTaskView{
|
|
ID: task.ID,
|
|
SessionID: task.SessionID,
|
|
Status: task.Status,
|
|
Error: task.Error,
|
|
ExitCode: task.ExitCode,
|
|
Blackboard: task.Blackboard,
|
|
RoundMax: task.RoundMax,
|
|
CreatedAt: task.CreatedAt,
|
|
FinishedAt: task.FinishedAt,
|
|
}
|
|
// 轮次进度以黑板 logs 目录实际出现的轮次为准（容器清理后日志仍在）
|
|
view.CurrentRound = agent.CurrentCoopRound(task.Blackboard)
|
|
if view.CurrentRound == 0 {
|
|
view.CurrentRound = view.RoundMax
|
|
}
|
|
// 运行中的任务实时查询容器状态；已完成任务容器已被清理，直接读记录即可
|
|
if task.Status == "running" && task.ContainerID != "" {
|
|
if state, status, err := agent.ContainerInspectStatus(s.dockerCfg.Socket(), task.ContainerID); err == nil {
|
|
view.ContainerState = state
|
|
view.ContainerStatus = status
|
|
}
|
|
}
|
|
return view
|
|
}
|