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

Merge pull request 'fix noticeinfo bugs' (#235) from tzwang/pcm-coordinator:master into master 

Former-commit-id: 5f1f61275d53b4463951a0e0dcca2a497548bc79
This commit is contained in:
tzwang 2024-06-22 19:23:41 +08:00
parent 8f1fd4af5b f193d6ab1f
commit ae657125d1
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/internal/logic/inference/imageinferencelogic.go
View File

@ -333,9 +333,9 @@ func infer(opt *option.InferOption, clusters []*strategy.AssignedCluster, ts []s
}
if len(cs) == successStatusCount {
svcCtx.Scheduler.AiStorages.AddNoticeInfo(opt.AdapterId, adapterName, "", "", opt.TaskName, "failed", "任务失败")
} else {
svcCtx.Scheduler.AiStorages.AddNoticeInfo(opt.AdapterId, adapterName, "", "", opt.TaskName, "completed", "任务完成")
} else {
svcCtx.Scheduler.AiStorages.AddNoticeInfo(opt.AdapterId, adapterName, "", "", opt.TaskName, "failed", "任务失败")
}
//save ai sub tasks

12
api/internal/scheduler/database/aiStorage.go
View File

@ -311,12 +311,16 @@ func (s *AiStorage) GetStrategyCode(name string) (int64, error) {
func (s *AiStorage) AddNoticeInfo(adapterId string, adapterName string, clusterId string, clusterName string, taskName string, noticeType string, incident string) {
aId, err := strconv.ParseInt(adapterId, 10, 64)
if err != nil {
return
logx.Errorf("adapterId convert failure, err: %v", err)
}
cId, err := strconv.ParseInt(clusterId, 10, 64)
if err != nil {
return
var cId int64
if clusterId != "" {
cId, err = strconv.ParseInt(clusterId, 10, 64)
if err != nil {
logx.Errorf("clusterId convert failure, err: %v", err)
}
}
noticeInfo := clientCore.NoticeInfo{
AdapterId: aId,
AdapterName: adapterName,