优化 调度方案

This commit is contained in:
songjc 2023-09-19 15:25:03 +08:00
parent 997a6f374c
commit 894796ec87
1 changed files with 11 additions and 9 deletions

View File

@ -362,18 +362,20 @@ func (s *DefaultSchedule) ComputeAllSlwNodeScore(files jobmod.JobFiles) (map[int
for id, image := range imageStgScores {
// TODO 此处id错误根据算力中心id做判断待修改
var isLoaded bool
if id == res.SlwNodeImageID {
for _, info := range res.ImportingInfos {
if id == info.SlwNodeID {
imageResp, err := colCli.GetImageList(collector.NewGetImageList(id))
if err != nil {
return nil, err
}
for _, imageID := range imageResp.ImageIDs {
if imageID == res.SlwNodeImageID {
if imageID == info.SlwNodeImageID {
isLoaded = true
break
}
}
}
}
if exist, ok := slwNodeScores[id]; ok {
exist.ImageScore = stgScore{
cachedScore: image.cachedScore,