From 00d56e62dc8dc532f84b83b1fcb9bccb6e8972b0 Mon Sep 17 00:00:00 2001 From: tzwang Date: Mon, 20 May 2024 20:30:15 +0800 Subject: [PATCH] fix tasklist bugs Former-commit-id: 64b3cd1678e9cf7435e8308ba4096409c9de1812 --- api/internal/logic/ai/getcentertasklistlogic.go | 4 ++++ api/internal/logic/core/pagelisttasklogic.go | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/api/internal/logic/ai/getcentertasklistlogic.go b/api/internal/logic/ai/getcentertasklistlogic.go index bbae384b..b7fa034f 100644 --- a/api/internal/logic/ai/getcentertasklistlogic.go +++ b/api/internal/logic/ai/getcentertasklistlogic.go @@ -113,6 +113,10 @@ func (l *GetCenterTaskListLogic) updateAiTaskStatus(mu *sync.RWMutex, ch chan<- wg.Done() return } + if trainingTask == nil { + wg.Done() + return + } t.Status = trainingTask.Status t.StartTime = trainingTask.Start t.EndTime = trainingTask.End diff --git a/api/internal/logic/core/pagelisttasklogic.go b/api/internal/logic/core/pagelisttasklogic.go index f1858a4a..db5ea094 100644 --- a/api/internal/logic/core/pagelisttasklogic.go +++ b/api/internal/logic/core/pagelisttasklogic.go @@ -255,6 +255,10 @@ func (l *PageListTaskLogic) updateAiTaskStatus(tasks []*types.TaskModel, ch chan wg.Done() return } + if trainingTask == nil { + wg.Done() + return + } t.Status = trainingTask.Status t.StartTime = trainingTask.Start t.EndTime = trainingTask.End