diff --git a/desc/inference/inference.api b/desc/inference/inference.api index 3f12d5a1..87b9f7fd 100644 --- a/desc/inference/inference.api +++ b/desc/inference/inference.api @@ -35,7 +35,8 @@ type ( /******************image inference*************************/ /******************instance center*************************/ InstanceCenterReq{ - InstanceType int32 `form:"instance_type"` + InstanceType int32 `json:"instance_type"` + InstanceClass string `json:"instance_class"` } InstanceCenterResp { InstanceCenterList []InstanceCenterList `json:"instanceCenterList" copier:"InstanceCenterList"` diff --git a/internal/logic/inference/instancecenterlogic.go b/internal/logic/inference/instancecenterlogic.go index dd0d90ec..dd7deacf 100644 --- a/internal/logic/inference/instancecenterlogic.go +++ b/internal/logic/inference/instancecenterlogic.go @@ -33,9 +33,10 @@ func (l *InstanceCenterLogic) InstanceCenter(req *types.InstanceCenterReq) (*typ if req.InstanceType == 0 { l.svcCtx.DbEngin.Raw("select * from ai_instance_center").Scan(&InstanceCenter) + } else if req.InstanceType != 0 && req.InstanceClass == "" { + 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 = ?", req.InstanceType).Scan(&InstanceCenter) - + l.svcCtx.DbEngin.Raw("select * from ai_instance_center where instance_type = ? and instance_class = ?", req.InstanceType, req.InstanceClass).Scan(&InstanceCenter) } var instanceCenter = *InstanceCenter diff --git a/internal/types/types.go b/internal/types/types.go index 8183e7fa..271fcb36 100644 --- a/internal/types/types.go +++ b/internal/types/types.go @@ -3666,7 +3666,8 @@ type InstanceCenterList struct { } type InstanceCenterReq struct { - InstanceType int32 `form:"instance_type"` + InstanceType int32 `json:"instance_type"` + InstanceClass string `json:"instance_class"` } type InstanceCenterResp struct {