admin/TaskPool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: adjust grom find 1 sql

Browse Source
This commit is contained in:
engigu
2026-03-23 12:04:29 +08:00
parent ec2cd7d116
commit d396e1629f
20 changed files with 87 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/services/tasks/task_service.go
+6 -3
View File
@@ -15,7 +15,8 @@ func NewTaskService() *TaskService {
func (ts *TaskService) GetTaskBySourceID(sourceID string) *models.Task {
var task models.Task
if err := database.DB.Where("source_id = ?", sourceID).First(&task).Error; err != nil {
res := database.DB.Where("source_id = ?", sourceID).Limit(1).Find(&task)
if res.Error != nil || res.RowsAffected == 0 {
return nil
}
return &task
@@ -91,7 +92,8 @@ func (ts *TaskService) GetTasksWithPagination(page, pageSize int, name string, a
func (ts *TaskService) GetTaskByID(id string) *models.Task {
var task models.Task
if err := database.DB.Where("id = ?", id).First(&task).Error; err != nil {
res := database.DB.Where("id = ?", id).Limit(1).Find(&task)
if res.Error != nil || res.RowsAffected == 0 {
return nil
}
return &task
@@ -99,7 +101,8 @@ func (ts *TaskService) GetTaskByID(id string) *models.Task {
func (ts *TaskService) UpdateTask(id string, name, command, schedule string, timeout int, workDir, cleanConfig, envs string, enabled bool, taskType, config string, agentID *string, languages []map[string]string, triggerType string, tags string, retryCount int, retryInterval int, randomRange int, sourceID string) *models.Task {
var task models.Task
if err := database.DB.Where("id = ?", id).First(&task).Error; err != nil {
res := database.DB.Where("id = ?", id).Limit(1).Find(&task)
if res.Error != nil || res.RowsAffected == 0 {
return nil
}
task.Name = name