From 31589e17a02e16afb1243412bdbff2f76d9800a1 Mon Sep 17 00:00:00 2001 From: zouap Date: Thu, 4 Nov 2021 17:41:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/ai_model_manage.go | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 28286587f..47b3501a0 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -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.")