Browse Source

代码规范

tags/v1.22.6.2^2
liuzx 3 years ago
parent
commit
b8ec66f52c
1 changed files with 9 additions and 48 deletions
  1. +9
    -48
      routers/repo/modelarts.go

+ 9
- 48
routers/repo/modelarts.go View File

@@ -1101,15 +1101,15 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
Value: modelarts.Ascend,
})
}
DatasUrlList, dataUrl, datasetNames, isMultiDataset, err := GetDatasUrlListByUUIDS(uuid)
datasUrlList, dataUrl, datasetNames, isMultiDataset, err := getDatasUrlListByUUIDS(uuid)
if err != nil {
log.Error("Failed to GetDatasUrlListByUUIDS: %v", err)
log.Error("Failed to getDatasUrlListByUUIDS: %v", err)
trainJobErrorNewDataPrepare(ctx, form)
ctx.RenderWithErr("Failed to GetDatasUrlListByUUIDS:"+err.Error(), tplModelArtsTrainJobNew, &form)
ctx.RenderWithErr("Failed to getDatasUrlListByUUIDS:"+err.Error(), tplModelArtsTrainJobNew, &form)
return
}
dataPath := dataUrl
jsondatas, err := json.Marshal(DatasUrlList)
jsondatas, err := json.Marshal(datasUrlList)
if err != nil {
log.Error("Failed to Marshal: %v", err)
trainJobErrorNewDataPrepare(ctx, form)
@@ -1342,15 +1342,15 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
})
}

DatasUrlList, dataUrl, datasetNames, isMultiDataset, err := GetDatasUrlListByUUIDS(uuid)
datasUrlList, dataUrl, datasetNames, isMultiDataset, err := getDatasUrlListByUUIDS(uuid)
if err != nil {
log.Error("Failed to GetDatasUrlListByUUIDS: %v", err)
log.Error("Failed to getDatasUrlListByUUIDS: %v", err)
versionErrorDataPrepare(ctx, form)
ctx.RenderWithErr("Failed to GetDatasUrlListByUUIDS:"+err.Error(), tplModelArtsTrainJobVersionNew, &form)
ctx.RenderWithErr("Failed to getDatasUrlListByUUIDS:"+err.Error(), tplModelArtsTrainJobVersionNew, &form)
return
}
dataPath := dataUrl
jsondatas, err := json.Marshal(DatasUrlList)
jsondatas, err := json.Marshal(datasUrlList)
if err != nil {
log.Error("Failed to Marshal: %v", err)
versionErrorDataPrepare(ctx, form)
@@ -2482,7 +2482,7 @@ func TrainJobDownloadLogFile(ctx *context.Context) {
ctx.Resp.Header().Set("Cache-Control", "max-age=0")
http.Redirect(ctx.Resp, ctx.Req.Request, url, http.StatusMovedPermanently)
}
func GetDatasUrlListByUUIDS(uuidStr string) ([]models.Datasurl, string, string, bool, error) {
func getDatasUrlListByUUIDS(uuidStr string) ([]models.Datasurl, string, string, bool, error) {
var isMultiDataset bool
var dataUrl string
var datasetNames string
@@ -2528,43 +2528,4 @@ func GetDatasUrlListByUUIDS(uuidStr string) ([]models.Datasurl, string, string,
}

return datasUrlList, dataUrl, datasetNames, isMultiDataset, nil

// var dataUrl string
// var DatasUrlList []models.Datasurl
// uuidList := strings.Split(uuidStr, ";")
// isMultiDataset := false

// if len(uuidList) >= 1 {
// for _, uuid := range uuidList {
// attach, err := models.GetAttachmentByUUID(uuid)
// if err != nil {
// log.Error("GetAttachmentByUUID failed: %v", err)
// return nil, "", isMultiDataset, err
// }
// datasetName := attach.Name
// index := strings.LastIndex(datasetName, ".")
// if index <= 0 {
// index = 0
// }
// datasetNameHead := datasetName[:index]
// datasetUrl := "s3://" + setting.Bucket + "/" + setting.BasePath + path.Join(uuid[0:1], uuid[1:2]) + "/" + uuid + uuid + "/"
// DatasUrlList = append(DatasUrlList, models.Datasurl{
// DatasetUrl: datasetUrl,
// DatasetName: datasetNameHead,
// })
// }
// firstDataset := uuidList[0]
// dataUrl = "/" + setting.Bucket + "/" + setting.BasePath + path.Join(firstDataset[0:1], firstDataset[1:2]) + "/" + firstDataset + firstDataset + "/"
// isMultiDataset = true
// return DatasUrlList, dataUrl, isMultiDataset, nil
// }
// return nil, "", isMultiDataset, nil
}
func IsDatasetUseCountExceed(uuid string) bool {
uuidList := strings.Split(uuid, ";")
if len(uuidList) > setting.MaxDatasetNum {
return true
} else {
return false
}
}

Loading…
Cancel
Save