|
|
@@ -7,7 +7,9 @@ import ( |
|
|
|
"io" |
|
|
|
"io/ioutil" |
|
|
|
"os" |
|
|
|
"path" |
|
|
|
"path/filepath" |
|
|
|
"strings" |
|
|
|
|
|
|
|
"code.gitea.io/gitea/models" |
|
|
|
"code.gitea.io/gitea/modules/context" |
|
|
@@ -151,17 +153,18 @@ func downloadModelFromCloudBrainTwo(modelUUID string, jobName string, parentDir |
|
|
|
models.AttachmentRelativePath(modelUUID) + |
|
|
|
"/" |
|
|
|
|
|
|
|
models, err := storage.GetObsListObject(jobName, parentDir) |
|
|
|
modelDbResult, err := storage.GetObsListObject(jobName, parentDir) |
|
|
|
if err != nil { |
|
|
|
log.Info("get TrainJobListModel failed:", err) |
|
|
|
return "", 0, err |
|
|
|
} |
|
|
|
if len(models) == 0 { |
|
|
|
if len(modelDbResult) == 0 { |
|
|
|
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) |
|
|
|
// if err != nil { |
|
|
|
// log.Info("copy failed.") |
|
|
|