提交代码。

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

models/ai_model_manage.go

@@ -56,14 +56,19 @@ func SaveModelToDb(model *AiModelManage) error {
 	return nil
 }
 
-func QueryModelById(id string) AiModelManage {
+func QueryModelById(id string) (*AiModelManage, error) {
 	sess := x.NewSession()
 	defer sess.Close()
 	sess.Select("*").Table("ai_model_manage").
 		Where("id='" + id + "'")
-	var aiModelManage AiModelManage
-	sess.Find(&aiModelManage)
-	return aiModelManage
+	aiModelManageList := make([]*AiModelManage, 0)
+	err := sess.Find(&aiModelManageList)
+	if err != nil {
+		if len(aiModelManageList) == 1 {
+			return aiModelManageList[0], nil
+		}
+	}
+	return nil, err
 }
 
 func DeleteModelById(id string) error {

routers/repo/ai_model_manage.go

@@ -135,14 +135,17 @@ func DeleteModel(ctx *context.Context) {
 func DeleteModelByID(id string) error {
 	log.Info("delete model start. id=" + id)
 
-	model := models.QueryModelById(id)
-	log.Info("bucket=" + setting.Bucket + " path=" + model.Path)
-	//err := storage.ObsRemoveObject(setting.Bucket, model.Path)
-	//if err != nil {
-	//	log.Info("Failed to delete model. id=" + id)
-	//	return err
-	//}
-	return models.DeleteModelById(id)
+	model, err := models.QueryModelById(id)
+	if err == nil {
+		log.Info("bucket=" + setting.Bucket + " path=" + model.Path)
+		//err := storage.ObsRemoveObject(setting.Bucket, model.Path)
+		//if err != nil {
+		//	log.Info("Failed to delete model. id=" + id)
+		//	return err
+		//}
+		return models.DeleteModelById(id)
+	}
+	return err
 }
 
 func DownloadModel(ctx *context.Context) {