engigu
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package tasks
|
|
|
|
import (
|
|
"github.com/engigu/baihu-panel/internal/constant"
|
|
"github.com/engigu/baihu-panel/internal/database"
|
|
"github.com/engigu/baihu-panel/internal/models"
|
|
)
|
|
|
|
type TaskService struct{}
|
|
|
|
func NewTaskService() *TaskService {
|
|
return &TaskService{}
|
|
}
|
|
|
|
func (ts *TaskService) CreateTask(name, command, schedule string, timeout int, workDir, cleanConfig, envs, taskType, config string, agentID *uint, languages []map[string]string, triggerType string) *models.Task {
|
|
if taskType == "" {
|
|
taskType = "task"
|
|
}
|
|
if triggerType == "" {
|
|
triggerType = constant.TriggerTypeCron
|
|
}
|
|
task := &models.Task{
|
|
Name: name,
|
|
Command: command,
|
|
Type: taskType,
|
|
TriggerType: triggerType,
|
|
Config: config,
|
|
Schedule: schedule,
|
|
Timeout: timeout,
|
|
WorkDir: workDir,
|
|
CleanConfig: cleanConfig,
|
|
Envs: envs,
|
|
Languages: languages,
|
|
AgentID: agentID,
|
|
Enabled: true,
|
|
}
|
|
if triggerType != constant.TriggerTypeCron {
|
|
task.NextRun = nil
|
|
}
|
|
database.DB.Create(task)
|
|
return task
|
|
}
|
|
|
|
func (ts *TaskService) GetTasks() []models.Task {
|
|
var tasks []models.Task
|
|
database.DB.Find(&tasks)
|
|
return tasks
|
|
}
|
|
|
|
// GetTasksWithPagination 分页获取任务列表
|
|
func (ts *TaskService) GetTasksWithPagination(page, pageSize int, name string, agentID *uint) ([]models.Task, int64) {
|
|
var tasks []models.Task
|
|
var total int64
|
|
|
|
query := database.DB.Model(&models.Task{})
|
|
if name != "" {
|
|
query = query.Where("name LIKE ?", "%"+name+"%")
|
|
}
|
|
if agentID != nil {
|
|
query = query.Where("agent_id = ?", *agentID)
|
|
}
|
|
|
|
query.Count(&total)
|
|
query.Order("id DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&tasks)
|
|
|
|
return tasks, total
|
|
}
|
|
|
|
func (ts *TaskService) GetTaskByID(id int) *models.Task {
|
|
var task models.Task
|
|
if err := database.DB.First(&task, id).Error; err != nil {
|
|
return nil
|
|
}
|
|
return &task
|
|
}
|
|
|
|
func (ts *TaskService) UpdateTask(id int, name, command, schedule string, timeout int, workDir, cleanConfig, envs string, enabled bool, taskType, config string, agentID *uint, languages []map[string]string, triggerType string) *models.Task {
|
|
var task models.Task
|
|
if err := database.DB.First(&task, id).Error; err != nil {
|
|
return nil
|
|
}
|
|
task.Name = name
|
|
task.Command = command
|
|
task.Schedule = schedule
|
|
task.Timeout = timeout
|
|
task.WorkDir = workDir
|
|
task.CleanConfig = cleanConfig
|
|
task.Envs = envs
|
|
task.Enabled = enabled
|
|
task.AgentID = agentID
|
|
task.Languages = languages
|
|
if taskType != "" {
|
|
task.Type = taskType
|
|
}
|
|
if triggerType != "" {
|
|
task.TriggerType = triggerType
|
|
}
|
|
if task.TriggerType != constant.TriggerTypeCron {
|
|
task.NextRun = nil
|
|
}
|
|
task.Config = config
|
|
database.DB.Save(&task)
|
|
return &task
|
|
}
|
|
|
|
func (ts *TaskService) DeleteTask(id int) bool {
|
|
result := database.DB.Delete(&models.Task{}, id)
|
|
return result.RowsAffected > 0
|
|
}
|