diff --git a/models/cloudbrain.go b/models/cloudbrain.go index 78efdc1b2..89b7abd4f 100755 --- a/models/cloudbrain.go +++ b/models/cloudbrain.go @@ -550,18 +550,18 @@ type GetResourceSpecsResult struct { ErrorMsg string `json:"error_msg"` IsSuccess bool `json:"is_success"` SpecTotalCount int `json:"spec_total_count"` - Specs []Specs `json:"specs"` + Specs []Specs `json:"specs"` } type Specs struct { - ErrorCode string `json:"core"` - ErrorMsg string `json:"cpu"` - IsSuccess bool `json:"no_resource"` - JobName string `json:"gpu_type"` - JobID int64 `json:"spec_id"` - Status int `json:"gpu_num"` - ResourceID string `json:"spec_code"` - VersionName string `json:"storage"` + Core string `json:"core"` + Cpu string `json:"cpu"` + IsNoResource bool `json:"no_resource"` + GpuType string `json:"gpu_type"` + SpecID int64 `json:"spec_id"` + GpuNum int `json:"gpu_num"` + SpecCode string `json:"spec_code"` + Storage string `json:"storage"` MaxNum int `json:"max_num"` UnitNum int `json:"unit_num"` InterfaceType int `json:"interface_type"` diff --git a/modules/modelarts/resty.go b/modules/modelarts/resty.go index 10267c549..bb514d0ba 100755 --- a/modules/modelarts/resty.go +++ b/modules/modelarts/resty.go @@ -319,6 +319,11 @@ sendjob: goto sendjob } + if res.StatusCode() != http.StatusOK { + log.Error("createTrainJob failed(%d)", res.StatusCode()) + return &result, fmt.Errorf("createTrainJob failed(%d)", res.StatusCode()) + } + if !result.IsSuccess { log.Error("createTrainJob failed(%s): %s", result.ErrorCode, result.ErrorMsg) return &result, fmt.Errorf("createTrainJob failed(%s): %s", result.ErrorCode, result.ErrorMsg) @@ -351,6 +356,8 @@ sendjob: goto sendjob } + log.Info("", res.StatusCode(), res.RawResponse.Body) + if res.StatusCode() != http.StatusOK { log.Error("GetResourceSpecs failed(%d)", res.StatusCode()) return &result, fmt.Errorf("GetResourceSpecs failed(%d)", res.StatusCode()) diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index b58398759..e6bc5b49d 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -325,6 +325,16 @@ func TrainJobNew(ctx *context.Context) { return } ctx.Data["flavor_infos"] = flavorInfos.Info + + res, err := modelarts.GetResourceSpecs() + if err != nil { + log.Error("GetResourceSpecs failed: %v", err) + ctx.ServerError("GetResourceSpecs failed:", err) + return + } + + log.Info("", res.SpecTotalCount) + ctx.HTML(200, tplModelArtsTrainJobNew) }