From c8d0f54f5cb6fd73828d9953aa54517fbd81cc6d Mon Sep 17 00:00:00 2001 From: tzwang Date: Thu, 16 May 2024 11:20:26 +0800 Subject: [PATCH] fix aitasklist bugs Former-commit-id: b955cc404935f0e733b2eb5904ce4654d347d088 --- api/internal/logic/ai/getcentertasklistlogic.go | 2 +- api/internal/logic/core/pagelisttasklogic.go | 7 +++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/api/internal/logic/ai/getcentertasklistlogic.go b/api/internal/logic/ai/getcentertasklistlogic.go index d56a05a5..28393c71 100644 --- a/api/internal/logic/ai/getcentertasklistlogic.go +++ b/api/internal/logic/ai/getcentertasklistlogic.go @@ -101,7 +101,7 @@ func (l *GetCenterTaskListLogic) updateAiTaskStatus(mu *sync.RWMutex, ch chan<- } for _, task := range taskList { t := task - if t.Status == constants.Completed { + if t.Status == constants.Completed || task.Status == constants.Failed { continue } wg.Add(1) diff --git a/api/internal/logic/core/pagelisttasklogic.go b/api/internal/logic/core/pagelisttasklogic.go index c0be0b29..ec1b8945 100644 --- a/api/internal/logic/core/pagelisttasklogic.go +++ b/api/internal/logic/core/pagelisttasklogic.go @@ -80,8 +80,7 @@ func (l *PageListTaskLogic) PageListTask(req *types.PageTaskReq) (resp *types.Pa for _, ch := range chs { select { case <-ch: - return - case <-time.After(1 * time.Second): + case <-time.After(2 * time.Second): return } } @@ -93,7 +92,7 @@ func (l *PageListTaskLogic) updateTaskStatus(tasks []*types.TaskModel, ch chan<- if task.AdapterTypeDict != 1 { continue } - if task.Status == constants.Succeeded { + if task.Status == constants.Succeeded || task.Status == constants.Failed { continue } @@ -171,7 +170,7 @@ func (l *PageListTaskLogic) updateAiTaskStatus(tasks []*types.TaskModel, ch chan if task.AdapterTypeDict != 1 { continue } - if task.Status == constants.Succeeded { + if task.Status == constants.Succeeded || task.Status == constants.Failed { continue }