floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'fix deployinstance logics' (#276) from tzwang/pcm-coordinator:master into master 

Former-commit-id: c6ea68e62c9850d50985fff5c388125a61f29c46
This commit is contained in:
tzwang 2024-07-27 15:08:15 +08:00
parent 1b475c5b63 1f868c58c4
commit 764ce65e8d
4 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/logic/inference/startdeployinstancelistlogic.go
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")

2
internal/logic/inference/stopdeployinstancelogic.go
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")

6
internal/scheduler/database/aiStorage.go
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())

8
internal/scheduler/service/updater/deployInstance.go
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
}
}