提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

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

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