fix deployinstance status

This commit is contained in:
tzwang 2024-07-27 15:07:19 +08:00
parent c7b10f2aec
commit 8e34db7011
4 changed files with 7 additions and 11 deletions

View File

@ -33,8 +33,6 @@ func (l *StartDeployInstanceListLogic) StartDeployInstanceList(req *types.StartD
return nil, err
}
l.svcCtx.Scheduler.AiStorages.UpdateInferDeployInstance(ins)
success := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[req.AdapterId][req.ClusterId].StartInferDeployInstance(l.ctx, req.InstanceId)
if !success {
return nil, errors.New("start instance failed")

View File

@ -33,8 +33,6 @@ func (l *StopDeployInstanceLogic) StopDeployInstance(req *types.StopDeployInstan
return nil, err
}
l.svcCtx.Scheduler.AiStorages.UpdateInferDeployInstance(ins)
success := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[req.AdapterId][req.ClusterId].StopInferDeployInstance(l.ctx, req.InstanceId)
if !success {
return nil, errors.New("stop instance failed")

View File

@ -399,8 +399,10 @@ func (s *AiStorage) SaveInferDeployInstance(instanceId string, instanceName stri
return insModel.Id, nil
}
func (s *AiStorage) UpdateInferDeployInstance(instance *models.AiInferDeployInstance) error {
instance.UpdateTime = time.Now().Format(time.RFC3339)
func (s *AiStorage) UpdateInferDeployInstance(instance *models.AiInferDeployInstance, needUpdateTime bool) error {
if needUpdateTime {
instance.UpdateTime = time.Now().Format(time.RFC3339)
}
tx := s.DbEngin.Table("ai_infer_deploy_instance").Updates(instance)
if tx.Error != nil {
logx.Errorf(tx.Error.Error())

View File

@ -71,10 +71,8 @@ func UpdateDeployInstanceStatus(svc *svc.ServiceContext, instance *models.AiInfe
}
}
if updatetime {
err = svc.Scheduler.AiStorages.UpdateInferDeployInstance(instance)
if err != nil {
return
}
err = svc.Scheduler.AiStorages.UpdateInferDeployInstance(instance, updatetime)
if err != nil {
return
}
}