fix instancelist bugs

This commit is contained in:
tzwang 2024-10-15 17:27:03 +08:00
parent 24c8de7251
commit 27d76bbaab
4 changed files with 21 additions and 15 deletions

View File

@ -9,7 +9,6 @@ import (
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models"
"time"
)
type DeployInstanceListLogic struct {
@ -72,18 +71,6 @@ func (l *DeployInstanceListLogic) DeployInstanceList(req *types.DeployInstanceLi
if len(list) != 0 {
go status.UpdateDeployInstanceStatusBatch(l.svcCtx, list, true)
ins := list[0]
for i := range list {
uTime, _ := time.Parse(time.RFC3339, ins.UpdateTime)
latest, _ := time.Parse(time.RFC3339, list[i].UpdateTime)
if latest.After(uTime) {
ins = list[i]
}
}
go status.UpdateDeployInstanceStatus(l.svcCtx, ins, true, nil)
go status.UpdateDeployTaskStatus(l.svcCtx)
}
resp.List = &deployTasks

View File

@ -45,7 +45,11 @@ func (l *StartDeployInstanceListLogic) StartDeployInstanceList(req *types.StartD
}
}
go status.UpdateDeployInstanceStatus(l.svcCtx, ins, true, nil)
ins.Status = "Updating"
err = l.svcCtx.Scheduler.AiStorages.UpdateInferDeployInstance(ins, true)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -45,7 +45,11 @@ func (l *StopDeployInstanceLogic) StopDeployInstance(req *types.StopDeployInstan
}
}
go status.UpdateDeployInstanceStatus(l.svcCtx, ins, true, nil)
ins.Status = "Updating"
err = l.svcCtx.Scheduler.AiStorages.UpdateInferDeployInstance(ins, true)
if err != nil {
return nil, err
}
return resp, nil
}

View File

@ -483,6 +483,17 @@ func (s *AiStorage) UpdateDeployTaskById(id int64) error {
if err != nil {
return err
}
list, err := s.GetInstanceListByDeployTaskId(id)
if err != nil {
return err
}
for _, instance := range list {
instance.Status = "Updating"
err := s.UpdateInferDeployInstance(instance, true)
if err != nil {
return err
}
}
return nil
}