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

fix: update instance center of ai

This commit is contained in:
qiwang 2024-10-09 09:10:25 +08:00
parent d2cd5980e3
commit c005b8ab9c
3 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
desc/inference/inference.api
View File

@ -35,7 +35,8 @@ type (
/******************image inference*************************/ /******************image inference*************************/
/******************instance center*************************/ /******************instance center*************************/
InstanceCenterReq{ InstanceCenterReq{
InstanceType int32 `form:"instance_type"` InstanceType int32 `json:"instance_type"`
InstanceClass string `json:"instance_class"`
} }
InstanceCenterResp { InstanceCenterResp {
InstanceCenterList []InstanceCenterList `json:"instanceCenterList" copier:"InstanceCenterList"` InstanceCenterList []InstanceCenterList `json:"instanceCenterList" copier:"InstanceCenterList"`

5
internal/logic/inference/instancecenterlogic.go
View File

@ -33,9 +33,10 @@ func (l *InstanceCenterLogic) InstanceCenter(req *types.InstanceCenterReq) (*typ
if req.InstanceType == 0 { if req.InstanceType == 0 {
l.svcCtx.DbEngin.Raw("select * from ai_instance_center").Scan(&InstanceCenter) l.svcCtx.DbEngin.Raw("select * from ai_instance_center").Scan(&InstanceCenter)
} else { } else if req.InstanceType != 0 && req.InstanceClass == "" {
l.svcCtx.DbEngin.Raw("select * from ai_instance_center where instance_type = ? ", req.InstanceType).Scan(&InstanceCenter) l.svcCtx.DbEngin.Raw("select * from ai_instance_center where instance_type = ? ", req.InstanceType).Scan(&InstanceCenter)
} else {
l.svcCtx.DbEngin.Raw("select * from ai_instance_center where instance_type = ? and instance_class = ?", req.InstanceType, req.InstanceClass).Scan(&InstanceCenter)
} }
var instanceCenter = *InstanceCenter var instanceCenter = *InstanceCenter

3
internal/types/types.go
View File

@ -3666,7 +3666,8 @@ type InstanceCenterList struct {
} }
type InstanceCenterReq struct { type InstanceCenterReq struct {
InstanceType int32 `form:"instance_type"` InstanceType int32 `json:"instance_type"`
InstanceClass string `json:"instance_class"`
} }
type InstanceCenterResp struct { type InstanceCenterResp struct {