From 6b6aa42d017491002938740355ccfd0e7ac09f5f Mon Sep 17 00:00:00 2001 From: tzwang Date: Mon, 20 May 2024 19:10:43 +0800 Subject: [PATCH] fix aicenter bugs Former-commit-id: d9185bf8675a7e4e3cfb470d1dd7b54dd15a2cf9 --- api/internal/logic/ai/getcenteroverviewlogic.go | 9 +++++++++ api/internal/logic/ai/getcenterqueueinglogic.go | 3 +++ 2 files changed, 12 insertions(+) diff --git a/api/internal/logic/ai/getcenteroverviewlogic.go b/api/internal/logic/ai/getcenteroverviewlogic.go index a4ecf111..e6a45106 100644 --- a/api/internal/logic/ai/getcenteroverviewlogic.go +++ b/api/internal/logic/ai/getcenteroverviewlogic.go @@ -99,11 +99,20 @@ func (l *GetCenterOverviewLogic) updateClusterResource(mu *sync.RWMutex, ch chan } wg.Add(1) go func() { + _, ok := l.svcCtx.Scheduler.AiService.AiCollectorAdapterMap[adapter.Id][c.Id] + if !ok { + wg.Done() + return + } stat, err := l.svcCtx.Scheduler.AiService.AiCollectorAdapterMap[adapter.Id][c.Id].GetResourceStats(l.ctx) if err != nil { wg.Done() return } + if stat == nil { + wg.Done() + return + } clusterType, err := strconv.ParseInt(adapter.Type, 10, 64) if err != nil { wg.Done() diff --git a/api/internal/logic/ai/getcenterqueueinglogic.go b/api/internal/logic/ai/getcenterqueueinglogic.go index bd5e5e2b..e7405298 100644 --- a/api/internal/logic/ai/getcenterqueueinglogic.go +++ b/api/internal/logic/ai/getcenterqueueinglogic.go @@ -42,6 +42,9 @@ func (l *GetCenterQueueingLogic) GetCenterQueueing() (resp *types.CenterQueueing if err != nil { continue } + if queues == nil { + continue + } //todo sync current task queues current := &types.CenterQueue{ Name: cluster.Name,