From 3938cef7375a99ac800f1b4f37f9ce5d16a239b4 Mon Sep 17 00:00:00 2001 From: zouap Date: Thu, 4 Nov 2021 17:49:51 +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 --- models/ai_model_manage.go | 2 +- routers/repo/ai_model_manage.go | 13 +++++++------ 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index 65edcdc46..1f8953137 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -75,7 +75,7 @@ func QueryModelByName(name string, uid int64) []*AiModelManage { sess := x.NewSession() defer sess.Close() sess.Select("*").Table("ai_model_manage"). - Where("name=" + name + " and user_id=" + fmt.Sprint(uid)) + Where("name='" + name + "' and user_id=" + fmt.Sprint(uid)) aiModelManageList := make([]*AiModelManage, 0) sess.Find(&aiModelManageList) return aiModelManageList diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 47b3501a0..1cbbcb327 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -163,12 +163,13 @@ func downloadModelFromCloudBrainTwo(modelUUID string, jobName string, parentDir } 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.") - // } + destKeyNamePrefix := "aimodels/" + models.AttachmentRelativePath(modelUUID) + "/" + log.Info("copy file, bucket=" + setting.Bucket + ", src keyname=" + prefix + modelFile.FileName) + log.Info("Dest key name=" + destKeyNamePrefix + modelFile.FileName) + err := storage.ObsCopyFile(setting.Bucket, modelFile.ParenDir+modelFile.FileName, setting.Bucket, dataActualPath+modelFile.FileName) + if err != nil { + log.Info("copy failed.") + } } return dataActualPath, 0, nil