fix tasklist bugs

Former-commit-id: 64b3cd1678e9cf7435e8308ba4096409c9de1812
2024-05-20 20:30:15 +08:00 · 2024-05-20 20:30:15 +08:00 · 00d56e62dc
parent 2fe3f2f976
commit 00d56e62dc
2 changed files with 8 additions and 0 deletions
--- 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
--- 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