admin
e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
197 lines
7.8 KiB
Go
197 lines
7.8 KiB
Go
package models
|
||
|
||
import (
|
||
"database/sql/driver"
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/engigu/taskpool/internal/constant"
|
||
)
|
||
|
||
// TaskLanguages 自定义语言配置列表类型，处理 JSON 序列化
|
||
type TaskLanguages []map[string]string
|
||
|
||
func (t TaskLanguages) Value() (driver.Value, error) {
|
||
if t == nil {
|
||
return "[]", nil
|
||
}
|
||
b, err := json.Marshal(t)
|
||
return string(b), err
|
||
}
|
||
|
||
func (t *TaskLanguages) Scan(v interface{}) error {
|
||
if v == nil {
|
||
*t = nil
|
||
return nil
|
||
}
|
||
var data []byte
|
||
switch s := v.(type) {
|
||
case string:
|
||
data = []byte(s)
|
||
case []byte:
|
||
data = s
|
||
default:
|
||
return fmt.Errorf("invalid type for TaskLanguages: %T", v)
|
||
}
|
||
return json.Unmarshal(data, t)
|
||
}
|
||
|
||
// CleanConfig 清理配置结构
|
||
type CleanConfig struct {
|
||
Type string `json:"type"` // "day" 或 "count"
|
||
Keep int `json:"keep"` // 保留天数或条数
|
||
}
|
||
|
||
// RepoConfig 仓库同步配置
|
||
type RepoConfig struct {
|
||
SourceType string `json:"source_type"` // url 或 git
|
||
SourceURL string `json:"source_url"` // 源地址
|
||
TargetPath string `json:"target_path"` // 目标路径
|
||
Branch string `json:"branch"` // Git 分支
|
||
SparsePath string `json:"sparse_path"` // 稀疏检出路径（仅拉取指定目录或文件）
|
||
SingleFile bool `json:"single_file"` // 单文件模式（直接下载文件而非 sparse-checkout）
|
||
Proxy string `json:"proxy"` // 代理类型: none, ghproxy, mirror, custom
|
||
ProxyURL string `json:"proxy_url"` // 自定义代理地址
|
||
AuthToken string `json:"auth_token"` // 认证 Token
|
||
WhitelistPaths string `json:"whitelist_paths"` // 同步时保留的路径及脚本筛选白名单关键词，逗号或竖线分割
|
||
Blacklist string `json:"blacklist"` // 脚本筛选黑名单关键词，竖线分割
|
||
Dependence string `json:"dependence"` // 脚本依赖文件关键词，竖线分割
|
||
Extensions string `json:"extensions"` // 脚本文件后缀关键词，竖线分割
|
||
AutoAddCron bool `json:"auto_add_cron"` // 自动解析脚本注释添加定时任务
|
||
CommentToTask string `json:"commenttotask"` // 兼容 QL 格式任务脚本注释解析
|
||
RepoSource string `json:"repo_source"` // 仓库来源，如果是选择了这个 ql 导入的仓库，= ql
|
||
RepoDirName string `json:"repo_dir_name"` // 自定义仓库目录名
|
||
}
|
||
|
||
// TaskConfig 任务配置 RepoConfig+TaskConfig=task.config
|
||
type TaskConfig struct {
|
||
Concurrency int `json:"$task_concurrency"` // 0: disable concurrency, 1: enable concurrency
|
||
AllEnvs bool `json:"$task_all_envs"` // 开启则注入全部环境变量
|
||
}
|
||
|
||
// Task 代表一个计划任务
|
||
type Task struct {
|
||
ID string `json:"id" gorm:"primaryKey;size:20"`
|
||
Name string `json:"name" gorm:"size:255;not null"`
|
||
Remark string `json:"remark" gorm:"size:255;default:''"`
|
||
PinType string `json:"pin_type" gorm:"size:20;default:none;index"` // 置顶类型: constant.PinTypeNone, constant.PinTypeTop
|
||
Command BigText `json:"command"` // 普通任务的命令
|
||
PreCommand BigText `json:"pre_command"` // 执行前的命令
|
||
PostCommand BigText `json:"post_command"` // 执行后的命令
|
||
Tags string `json:"tags" gorm:"-"` // 标签，逗号分隔
|
||
Type string `json:"type" gorm:"size:20;default:'task'"` // 任务类型: constant.TaskTypeNormal, constant.TaskTypeRepo
|
||
TriggerType string `json:"trigger_type" gorm:"size:25;default:'cron'"` // 触发类型: constant.TriggerTypeCron, constant.TriggerTypeTaskPoolStartup
|
||
Config BigText `json:"config"` // 配置 JSON（仓库同步配置等）
|
||
Schedule string `json:"schedule" gorm:"size:100"` // cron 表达式
|
||
Timeout int `json:"timeout" gorm:"default:30"` // 超时时间（分钟），默认30分钟
|
||
WorkDir string `json:"work_dir" gorm:"size:255;default:''"` // 工作目录，为空则使用 scripts 目录
|
||
CleanConfig string `json:"clean_config" gorm:"size:255;default:''"` // 清理配置 JSON
|
||
Envs BigText `json:"envs" gorm:"-"` // 环境变量ID列表，逗号分隔
|
||
Languages TaskLanguages `json:"languages" gorm:"type:text"` // 针对本地任务的语言配置列表
|
||
AgentID *string `json:"agent_id" gorm:"size:20;index"` // Agent ID，为空表示本地执行
|
||
RetryCount int `json:"retry_count" gorm:"default:0"` // 失败重试次数
|
||
RetryInterval int `json:"retry_interval" gorm:"default:0"` // 失败重试间隔(秒)
|
||
RandomRange int `json:"random_range" gorm:"default:0"` // 随机延迟范围(秒)
|
||
Enabled *bool `json:"enabled" gorm:"default:true"`
|
||
RunningGo BigText `json:"running_go"` // 正在运行的 go routine id 数组 (JSON)
|
||
RuntimeEnvs []string `json:"-" gorm:"-"` // 运行时环境变量（非持久化）
|
||
RuntimeSecrets []string `json:"-" gorm:"-"` // 运行时安全机密（非持久化）
|
||
LastRun *LocalTime `json:"last_run"`
|
||
NextRun *LocalTime `json:"next_run"`
|
||
SourceID string `json:"source_id" gorm:"size:255;index"` // 脚本资源唯一标识（路径 sanitized）
|
||
RepoTaskID string `json:"repo_task_id" gorm:"size:20;index"` // 所属的仓库任务 ID
|
||
CreatedAt LocalTime `json:"created_at"`
|
||
UpdatedAt LocalTime `json:"updated_at"`
|
||
}
|
||
|
||
func (t *Task) IsRunning() bool {
|
||
if string(t.RunningGo) == "" || string(t.RunningGo) == "[]" {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
func (Task) TableName() string {
|
||
return constant.TablePrefix + "tasks"
|
||
}
|
||
|
||
func (t *Task) GetID() string {
|
||
return t.ID
|
||
}
|
||
|
||
func (t *Task) GetName() string {
|
||
return t.Name
|
||
}
|
||
|
||
func (t *Task) GetCommand() string {
|
||
return string(t.Command)
|
||
}
|
||
|
||
func (t *Task) GetPreCommand() string {
|
||
return string(t.PreCommand)
|
||
}
|
||
|
||
func (t *Task) GetPostCommand() string {
|
||
return string(t.PostCommand)
|
||
}
|
||
|
||
func (t *Task) GetTimeout() int {
|
||
return t.Timeout
|
||
}
|
||
|
||
func (t *Task) GetWorkDir() string {
|
||
return t.WorkDir
|
||
}
|
||
|
||
func (t *Task) GetEnvs() string {
|
||
return string(t.Envs)
|
||
}
|
||
|
||
func (t *Task) GetLanguages() []map[string]string {
|
||
return []map[string]string(t.Languages)
|
||
}
|
||
|
||
func (t *Task) GetEnvVars() []string {
|
||
return t.RuntimeEnvs
|
||
}
|
||
|
||
func (t *Task) GetSecrets() []string {
|
||
return t.RuntimeSecrets
|
||
}
|
||
|
||
func (t *Task) GetUseMise() bool {
|
||
return t.AgentID == nil || *t.AgentID == ""
|
||
}
|
||
|
||
func (t *Task) UseMise() bool {
|
||
return t.GetUseMise()
|
||
}
|
||
|
||
// CronTask 计划任务接口
|
||
func (t *Task) GetSchedule() string {
|
||
return t.Schedule
|
||
}
|
||
|
||
func (t *Task) GetRandomRange() int {
|
||
return t.RandomRange
|
||
}
|
||
|
||
// TaskLog 代表任务执行的日志记录
|
||
type TaskLog struct {
|
||
ID string `json:"id" gorm:"primaryKey;size:20"`
|
||
TaskID string `json:"task_id" gorm:"size:20;index"`
|
||
AgentID *string `json:"agent_id" gorm:"size:20;index"` // Agent ID，为空表示本地执行
|
||
Command BigText `json:"command"`
|
||
Output BigText `json:"-"` // gzip+base64 压缩后的日志
|
||
Error BigText `json:"error"` // 额外的系统错误信息
|
||
Status string `json:"status" gorm:"size:20;index"` // success, failed
|
||
Duration int64 `json:"duration"` // 执行耗时（毫秒）
|
||
ExitCode int `json:"exit_code"`
|
||
StartTime *LocalTime `json:"start_time"`
|
||
EndTime *LocalTime `json:"end_time"`
|
||
CreatedAt LocalTime `json:"created_at"`
|
||
}
|
||
|
||
func (TaskLog) TableName() string {
|
||
return constant.TablePrefix + "task_logs"
|
||
}
|