feat: refact scheduler

internal/models/agent.go

@@ -16,7 +16,7 @@ type Agent struct {
 	Status      string         `json:"status" gorm:"size:20;default:'pending'"` // 状态: pending(待审核), online, offline, blocked(拉黑)
 	LastSeen    *LocalTime     `json:"last_seen"`                               // 最后心跳时间
 	IP          string         `json:"ip" gorm:"size:45"`                       // Agent IP 地址
-	Version     string         `json:"version" gorm:"size:20"`                  // Agent 版本
+	Version     string         `json:"version" gorm:"size:50"`                  // Agent 版本
 	BuildTime   string         `json:"build_time" gorm:"size:30"`               // Agent 构建时间
 	Hostname    string         `json:"hostname" gorm:"size:100"`                // Agent 主机名
 	OS          string         `json:"os" gorm:"size:20"`                       // 操作系统
@@ -65,6 +65,7 @@ type AgentTask struct {
 // AgentTaskResult Agent 上报的任务执行结果
 type AgentTaskResult struct {
 	TaskID    uint   `json:"task_id"`
+	LogID     uint   `json:"log_id"`
 	AgentID   uint   `json:"agent_id"`
 	Command   string `json:"command"`
 	Output    string `json:"output"`

internal/models/task.go

@@ -1,6 +1,8 @@
 package models
 
 import (
+	"fmt"
+
 	"github.com/engigu/baihu-panel/internal/constant"
 
 	"gorm.io/gorm"
@@ -25,6 +27,11 @@ type RepoConfig struct {
 	AuthToken  string `json:"auth_token"`  // 认证 Token
 }
 
+// TaskConfig  任务配置  RepoConfig+TaskConfig=task.config
+type TaskConfig struct {
+	Concurrency int `json:"$task_concurrency"` // 0: disable concurrency, 1: enable concurrency
+}
+
 // Task represents a scheduled task
 type Task struct {
 	ID          uint           `json:"id" gorm:"primaryKey"`
@@ -39,6 +46,7 @@ type Task struct {
 	Envs        string         `json:"envs" gorm:"size:255;default:''"`         // 环境变量ID列表，逗号分隔
 	AgentID     *uint          `json:"agent_id" gorm:"index"`                   // Agent ID，为空表示本地执行
 	Enabled     bool           `json:"enabled" gorm:"default:true"`
+	RunningGo   string         `json:"running_go" gorm:"type:text"` // 正在运行的 go routine id 数组 (JSON)
 	LastRun     *LocalTime     `json:"last_run"`
 	NextRun     *LocalTime     `json:"next_run"`
 	CreatedAt   LocalTime      `json:"created_at"`
@@ -50,6 +58,26 @@ func (Task) TableName() string {
 	return constant.TablePrefix + "tasks"
 }
 
+func (t *Task) GetID() string {
+	return fmt.Sprintf("%d", t.ID)
+}
+
+func (t *Task) GetName() string {
+	return t.Name
+}
+
+func (t *Task) GetCommand() string {
+	return t.Command
+}
+
+func (t *Task) GetTimeout() int {
+	return t.Timeout
+}
+
+func (t *Task) GetSchedule() string {
+	return t.Schedule
+}
+
 // TaskLog represents a log entry for task execution
 type TaskLog struct {
 	ID        uint       `json:"id" gorm:"primaryKey"`