Browse Source

修正SQL注入的问题

Signed-off-by: zouap <zouap@pcl.ac.cn>
pull/3589/head
zouap 2 years ago
parent
commit
00b3f9dc46
1 changed files with 12 additions and 8 deletions
  1. +12
    -8
      models/ai_model_manage.go

+ 12
- 8
models/ai_model_manage.go View File

@@ -237,15 +237,19 @@ func QueryModelConvertByName(name string, repoId int64) ([]*AiModelConvert, erro
func QueryModelConvertById(id string) (*AiModelConvert, error) { func QueryModelConvertById(id string) (*AiModelConvert, error) {
sess := x.NewSession() sess := x.NewSession()
defer sess.Close() defer sess.Close()
sess.Table(new(AiModelConvert)).ID(id)
//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 isExist {
return re, nil
} }
//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
// }
//}
return nil, err return nil, err
} }




Loading…
Cancel
Save