From f8a4656d83b34d895cbde7644173789dfd2dffd7 Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Thu, 16 May 2024 18:17:57 +0800
Subject: [PATCH] fix aischeduler bugs

Former-commit-id: 4df953f2a614800510376210d69f82a646cae02f
---
 api/internal/scheduler/schedulers/aiScheduler.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/api/internal/scheduler/schedulers/aiScheduler.go b/api/internal/scheduler/schedulers/aiScheduler.go
index 84b2fe35..39b7e427 100644
--- a/api/internal/scheduler/schedulers/aiScheduler.go
+++ b/api/internal/scheduler/schedulers/aiScheduler.go
@@ -129,6 +129,7 @@ func (as *AiScheduler) AssignTask(clusters []*strategy.AssignedCluster) (interfa
 
 	var wg sync.WaitGroup
 	var results []*AiResult
+	var mu sync.Mutex
 	var errs []interface{}
 	var taskNum int32
 	for _, cluster := range clusters {
@@ -158,7 +159,10 @@ func (as *AiScheduler) AssignTask(clusters []*strategy.AssignedCluster) (interfa
 					return
 				}
 
+				mu.Lock()
 				result, _ := convertType(resp)
+				mu.Unlock()
+
 				result.Replica = c.Replicas
 				result.ClusterId = c.ClusterId
 				result.Strategy = as.option.StrategyName