|
|
|
@@ -290,22 +290,22 @@ func (i *ImageInference) updateStatus(aiTaskList []*models.TaskAi, cs []*Filtere |
|
|
|
|
|
|
|
//change cluster status |
|
|
|
if len(i.clusters) != len(cs) { |
|
|
|
var acs []*strategy.AssignedCluster |
|
|
|
var rcs []*strategy.AssignedCluster |
|
|
|
var failedclusters []*strategy.AssignedCluster |
|
|
|
var runningclusters []*strategy.AssignedCluster |
|
|
|
for _, cluster := range i.clusters { |
|
|
|
if contains(cs, cluster.ClusterId) { |
|
|
|
var ac *strategy.AssignedCluster |
|
|
|
ac = cluster |
|
|
|
rcs = append(rcs, ac) |
|
|
|
runningclusters = append(runningclusters, ac) |
|
|
|
} else { |
|
|
|
var ac *strategy.AssignedCluster |
|
|
|
ac = cluster |
|
|
|
acs = append(acs, ac) |
|
|
|
failedclusters = append(failedclusters, ac) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// update failed cluster status |
|
|
|
for _, ac := range acs { |
|
|
|
for _, ac := range failedclusters { |
|
|
|
for _, t := range aiTaskList { |
|
|
|
if ac.ClusterId == strconv.Itoa(int(t.ClusterId)) { |
|
|
|
t.Status = constants.Failed |
|
|
|
@@ -322,7 +322,7 @@ func (i *ImageInference) updateStatus(aiTaskList []*models.TaskAi, cs []*Filtere |
|
|
|
} |
|
|
|
|
|
|
|
// update running cluster status |
|
|
|
for _, ac := range rcs { |
|
|
|
for _, ac := range runningclusters { |
|
|
|
for _, t := range aiTaskList { |
|
|
|
if ac.ClusterId == strconv.Itoa(int(t.ClusterId)) { |
|
|
|
t.Status = constants.Running |
|
|
|
|