| @@ -7,7 +7,9 @@ import ( | |||||
| "io" | "io" | ||||
| "io/ioutil" | "io/ioutil" | ||||
| "os" | "os" | ||||
| "path" | |||||
| "path/filepath" | "path/filepath" | ||||
| "strings" | |||||
| "code.gitea.io/gitea/models" | "code.gitea.io/gitea/models" | ||||
| "code.gitea.io/gitea/modules/context" | "code.gitea.io/gitea/modules/context" | ||||
| @@ -151,17 +153,18 @@ func downloadModelFromCloudBrainTwo(modelUUID string, jobName string, parentDir | |||||
| models.AttachmentRelativePath(modelUUID) + | models.AttachmentRelativePath(modelUUID) + | ||||
| "/" | "/" | ||||
| models, err := storage.GetObsListObject(jobName, parentDir) | |||||
| modelDbResult, err := storage.GetObsListObject(jobName, parentDir) | |||||
| if err != nil { | if err != nil { | ||||
| log.Info("get TrainJobListModel failed:", err) | log.Info("get TrainJobListModel failed:", err) | ||||
| return "", 0, err | return "", 0, err | ||||
| } | } | ||||
| if len(models) == 0 { | |||||
| if len(modelDbResult) == 0 { | |||||
| return "", 0, errors.New("cannot create model, as model is empty.") | return "", 0, errors.New("cannot create model, as model is empty.") | ||||
| } | } | ||||
| for _, modelFile := range models { | |||||
| log.Info("copy file, bucket=%s, src keyname=%s,dest keyname=%s", setting.Bucket, modelFile.ParenDir+modelFile.FileName, dataActualPath+modelFile.FileName) | |||||
| prefix := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, jobName, setting.OutPutPath, parentDir), "/") | |||||
| for _, modelFile := range modelDbResult { | |||||
| destKeyName := "/aimodels/" + models.AttachmentRelativePath(modelUUID) + "/" | |||||
| log.Info("copy file, bucket=%s, src keyname=%s, dest keyname=%s,", setting.Bucket, prefix+modelFile.FileName, destKeyName) | |||||
| // err := storage.ObsCopyFile(setting.Bucket, modelFile.ParenDir+modelFile.FileName, setting.Bucket, dataActualPath+modelFile.FileName) | // err := storage.ObsCopyFile(setting.Bucket, modelFile.ParenDir+modelFile.FileName, setting.Bucket, dataActualPath+modelFile.FileName) | ||||
| // if err != nil { | // if err != nil { | ||||
| // log.Info("copy failed.") | // log.Info("copy failed.") | ||||