修正SQL注入的查询问题

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

models/ai_model_manage.go

@@ -237,13 +237,10 @@ func QueryModelConvertByName(name string, repoId int64) ([]*AiModelConvert, erro
 func QueryModelConvertById(id string) (*AiModelConvert, error) {
 	sess := x.NewSession()
 	defer sess.Close()
-	sess.Select("*").Table(new(AiModelConvert)).Where("id='" + id + "'")
-	aiModelManageConvertList := make([]*AiModelConvert, 0)
-	err := sess.Find(&aiModelManageConvertList)
-	if err == nil {
-		if len(aiModelManageConvertList) == 1 {
-			return aiModelManageConvertList[0], nil
-		}
+	re := new(AiModelConvert)
+	isExist, err := sess.Table(new(AiModelConvert)).ID(id).Get(re)
+	if err == nil && isExist {
+		return re, nil
 	}
 	return nil, err
 }
@@ -251,16 +248,10 @@ func QueryModelConvertById(id string) (*AiModelConvert, error) {
 func QueryModelById(id string) (*AiModelManage, error) {
 	sess := x.NewSession()
 	defer sess.Close()
-	sess.Select("*").Table("ai_model_manage").
-		Where("id='" + id + "'")
-	aiModelManageList := make([]*AiModelManage, 0)
-	err := sess.Find(&aiModelManageList)
-	if err == nil {
-		if len(aiModelManageList) == 1 {
-			return aiModelManageList[0], nil
-		}
-	} else {
-		log.Info("error=" + err.Error())
+	re := new(AiModelManage)
+	isExist, err := sess.Table(new(AiModelManage)).ID(id).Get(re)
+	if err == nil && isExist {
+		return re, nil
 	}
 	return nil, err
 }