admin
e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
264 lines
9.8 KiB
Go
264 lines
9.8 KiB
Go
package vo
|
|
|
|
import (
|
|
"github.com/engigu/taskpool/internal/executor"
|
|
"github.com/engigu/taskpool/internal/models"
|
|
"github.com/engigu/taskpool/internal/utils"
|
|
)
|
|
|
|
// TaskCreateReq 任务创建请求
|
|
type TaskCreateReq struct {
|
|
Name string `json:"name" binding:"required" example:"测试任务"`
|
|
Remark string `json:"remark" example:"备注信息"`
|
|
Command string `json:"command" example:"echo 'Hello World'"`
|
|
PreCommand string `json:"pre_command" example:"echo 'pre'"`
|
|
PostCommand string `json:"post_command" example:"echo 'post'"`
|
|
Tags string `json:"tags" example:"test,dev"`
|
|
Type string `json:"type" example:"repo"` // 可以是 common, repo 等
|
|
Config string `json:"config" swaggertype:"string" example:"{\"source_url\":\"https://github.com/abc/repo\",\"branch\":\"main\"}"`
|
|
Schedule string `json:"schedule" example:"0 0 * * *"`
|
|
Timeout int `json:"timeout" example:"3600"`
|
|
WorkDir string `json:"work_dir" example:"/tmp"`
|
|
CleanConfig string `json:"clean_config" example:"true"`
|
|
Envs string `json:"envs" example:"{\"ENV_VAR\":\"value\"}"`
|
|
Languages models.TaskLanguages `json:"languages"`
|
|
AgentID *string `json:"agent_id" example:"agent-1"`
|
|
TriggerType string `json:"trigger_type" example:"cron"`
|
|
RetryCount int `json:"retry_count" example:"3"`
|
|
RetryInterval int `json:"retry_interval" example:"60"`
|
|
RandomRange int `json:"random_range" example:"10"`
|
|
PinType string `json:"pin_type" example:"time"`
|
|
}
|
|
|
|
// TaskUpdateReq 任务更新请求
|
|
type TaskUpdateReq struct {
|
|
Name string `json:"name" example:"测试任务"`
|
|
Remark string `json:"remark" example:"备注信息"`
|
|
Command string `json:"command" example:"echo 'Hello World'"`
|
|
PreCommand string `json:"pre_command" example:"echo 'pre'"`
|
|
PostCommand string `json:"post_command" example:"echo 'post'"`
|
|
Tags string `json:"tags" example:"test,dev"`
|
|
Type string `json:"type" example:"repo"`
|
|
Config string `json:"config" swaggertype:"string" example:"{\"source_url\":\"https://github.com/abc/repo\",\"branch\":\"main\"}"`
|
|
Schedule string `json:"schedule" example:"0 0 * * *"`
|
|
Timeout int `json:"timeout" example:"3600"`
|
|
WorkDir string `json:"work_dir" example:"/tmp"`
|
|
CleanConfig string `json:"clean_config" example:"true"`
|
|
Envs string `json:"envs" example:"{\"ENV_VAR\":\"value\"}"`
|
|
Enabled bool `json:"enabled" example:"true"`
|
|
Languages models.TaskLanguages `json:"languages"`
|
|
AgentID *string `json:"agent_id" example:"agent-1"`
|
|
TriggerType string `json:"trigger_type" example:"cron"`
|
|
RetryCount int `json:"retry_count" example:"3"`
|
|
RetryInterval int `json:"retry_interval" example:"60"`
|
|
RandomRange int `json:"random_range" example:"10"`
|
|
PinType string `json:"pin_type" example:"time"`
|
|
}
|
|
|
|
// TaskVO 任务视图对象
|
|
type TaskVO struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Remark string `json:"remark"`
|
|
Command string `json:"command"`
|
|
PreCommand string `json:"pre_command"`
|
|
PostCommand string `json:"post_command"`
|
|
Tags string `json:"tags"`
|
|
Type string `json:"type"`
|
|
TriggerType string `json:"trigger_type"`
|
|
Config string `json:"config"`
|
|
Schedule string `json:"schedule"`
|
|
Timeout int `json:"timeout"`
|
|
WorkDir string `json:"work_dir"`
|
|
CleanConfig string `json:"clean_config"`
|
|
Envs string `json:"envs"`
|
|
Languages models.TaskLanguages `json:"languages"`
|
|
AgentID *string `json:"agent_id"`
|
|
RepoTaskID string `json:"repo_task_id"`
|
|
Enabled bool `json:"enabled"`
|
|
RetryCount int `json:"retry_count"`
|
|
RetryInterval int `json:"retry_interval"`
|
|
RandomRange int `json:"random_range"`
|
|
PinType string `json:"pin_type"`
|
|
LastRun *models.LocalTime `json:"last_run"`
|
|
NextRun *models.LocalTime `json:"next_run"`
|
|
CreatedAt models.LocalTime `json:"created_at"`
|
|
UpdatedAt models.LocalTime `json:"updated_at"`
|
|
RunningStatus string `json:"running_status"`
|
|
}
|
|
|
|
// ToTaskVO 将 Task 模型转换为 TaskVO
|
|
func ToTaskVO(task *models.Task) *TaskVO {
|
|
if task == nil {
|
|
return nil
|
|
}
|
|
return &TaskVO{
|
|
ID: task.ID,
|
|
Name: task.Name,
|
|
Remark: task.Remark,
|
|
Command: string(task.Command),
|
|
PreCommand: string(task.PreCommand),
|
|
PostCommand: string(task.PostCommand),
|
|
Tags: task.Tags,
|
|
Type: task.Type,
|
|
TriggerType: task.TriggerType,
|
|
Config: string(task.Config),
|
|
Schedule: task.Schedule,
|
|
Timeout: task.Timeout,
|
|
WorkDir: task.WorkDir,
|
|
CleanConfig: task.CleanConfig,
|
|
Envs: string(task.Envs),
|
|
Languages: task.Languages,
|
|
AgentID: task.AgentID,
|
|
RepoTaskID: task.RepoTaskID,
|
|
Enabled: utils.DerefBool(task.Enabled, true),
|
|
RetryCount: task.RetryCount,
|
|
RetryInterval: task.RetryInterval,
|
|
RandomRange: task.RandomRange,
|
|
PinType: task.PinType,
|
|
LastRun: task.LastRun,
|
|
NextRun: task.NextRun,
|
|
CreatedAt: task.CreatedAt,
|
|
UpdatedAt: task.UpdatedAt,
|
|
RunningStatus: func() string {
|
|
if task.IsRunning() {
|
|
return "running"
|
|
}
|
|
return "idle"
|
|
}(),
|
|
}
|
|
}
|
|
|
|
// ToTaskVOList 将 Task 模型列表转换为 TaskVO 列表
|
|
func ToTaskVOList(tasks []*models.Task) []*TaskVO {
|
|
if tasks == nil {
|
|
return nil
|
|
}
|
|
vos := make([]*TaskVO, len(tasks))
|
|
for i, t := range tasks {
|
|
vos[i] = ToTaskVO(t)
|
|
}
|
|
return vos
|
|
}
|
|
|
|
// ToTaskVOListFromModels 将 Task 模型列表转换为 TaskVO 列表
|
|
func ToTaskVOListFromModels(tasks []models.Task) []*TaskVO {
|
|
vos := make([]*TaskVO, len(tasks))
|
|
for i := range tasks {
|
|
vos[i] = ToTaskVO(&tasks[i])
|
|
}
|
|
return vos
|
|
}
|
|
|
|
// TaskLogVO 任务历史视图对象
|
|
type TaskLogVO struct {
|
|
ID string `json:"id"`
|
|
TaskID string `json:"task_id"`
|
|
TaskName string `json:"task_name"`
|
|
TaskType string `json:"task_type"`
|
|
AgentID *string `json:"agent_id"`
|
|
Command string `json:"command"`
|
|
Error string `json:"error"`
|
|
Status string `json:"status"`
|
|
Duration int64 `json:"duration"`
|
|
ExitCode int `json:"exit_code"`
|
|
StartTime *models.LocalTime `json:"start_time"`
|
|
EndTime *models.LocalTime `json:"end_time"`
|
|
CreatedAt models.LocalTime `json:"created_at"`
|
|
Output string `json:"output,omitempty"`
|
|
}
|
|
|
|
// ToTaskLogVO 将 TaskLog 模型转换为 TaskLogVO
|
|
// Note: This function assumes the Task field within models.TaskLog is preloaded
|
|
// or that taskName and taskType are provided from an external source.
|
|
func ToTaskLogVO(log *models.TaskLog) *TaskLogVO {
|
|
if log == nil {
|
|
return nil
|
|
}
|
|
return &TaskLogVO{
|
|
ID: log.ID,
|
|
TaskID: log.TaskID,
|
|
AgentID: log.AgentID,
|
|
Command: string(log.Command),
|
|
Error: string(log.Error),
|
|
Status: log.Status,
|
|
Duration: log.Duration,
|
|
ExitCode: log.ExitCode,
|
|
StartTime: log.StartTime,
|
|
EndTime: log.EndTime,
|
|
CreatedAt: log.CreatedAt,
|
|
Output: string(log.Output),
|
|
}
|
|
}
|
|
|
|
// ToTaskLogVOList 将 TaskLog 模型列表转换为 TaskLogVO 列表
|
|
func ToTaskLogVOList(logs []*models.TaskLog) []*TaskLogVO {
|
|
if logs == nil {
|
|
return nil
|
|
}
|
|
vos := make([]*TaskLogVO, len(logs))
|
|
for i, l := range logs {
|
|
vos[i] = ToTaskLogVO(l)
|
|
}
|
|
return vos
|
|
}
|
|
|
|
// ToTaskLogVOListFromModels 将 TaskLog 模型列表转换为 TaskLogVO 列表
|
|
func ToTaskLogVOListFromModels(logs []models.TaskLog) []*TaskLogVO {
|
|
vos := make([]*TaskLogVO, len(logs))
|
|
for i := range logs {
|
|
vos[i] = ToTaskLogVO(&logs[i])
|
|
}
|
|
return vos
|
|
}
|
|
|
|
// ExecutionResultVO 任务执行结果视图对象
|
|
type ExecutionResultVO struct {
|
|
TaskID string `json:"task_id"`
|
|
LogID string `json:"log_id,omitempty"`
|
|
Success bool `json:"success"`
|
|
Status string `json:"status"`
|
|
Output string `json:"output,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
Duration int64 `json:"duration,omitempty"`
|
|
ExitCode int `json:"exit_code,omitempty"`
|
|
StartTime string `json:"start_time,omitempty"`
|
|
EndTime string `json:"end_time,omitempty"`
|
|
}
|
|
|
|
// ToExecutionResultVO 将 ExecutionResult 转换为 ExecutionResultVO
|
|
func ToExecutionResultVO(res *executor.ExecutionResult) *ExecutionResultVO {
|
|
if res == nil {
|
|
return nil
|
|
}
|
|
vo := &ExecutionResultVO{
|
|
TaskID: res.TaskID,
|
|
LogID: res.LogID,
|
|
Success: res.Success,
|
|
Status: res.Status,
|
|
Output: res.Output,
|
|
Error: res.Error,
|
|
Duration: res.Duration,
|
|
ExitCode: res.ExitCode,
|
|
}
|
|
if !res.StartTime.IsZero() {
|
|
vo.StartTime = res.StartTime.Format("2006-01-02 15:04:05")
|
|
}
|
|
if !res.EndTime.IsZero() {
|
|
vo.EndTime = res.EndTime.Format("2006-01-02 15:04:05")
|
|
}
|
|
return vo
|
|
}
|
|
|
|
// ToExecutionResultVOList 将 ExecutionResult 列表转换为 ExecutionResultVO 列表
|
|
func ToExecutionResultVOList(results []executor.ExecutionResult) []*ExecutionResultVO {
|
|
if results == nil {
|
|
return nil
|
|
}
|
|
vos := make([]*ExecutionResultVO, len(results))
|
|
for i := range results {
|
|
vos[i] = ToExecutionResultVO(&results[i])
|
|
}
|
|
return vos
|
|
}
|