Merge pull request 'fix tasklist bugs' (#191) from tzwang/pcm-coordinator:master into master

Former-commit-id: 150403b0dee834be0191679fbb5c058da3e13a3c
2024-05-20 20:39:09 +08:00 · 2024-05-20 20:39:09 +08:00 · 68ffa06ece
parent af7b932962 bc8d65af40
commit 68ffa06ece
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