Browse Source

修正SQL注入的问题

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

+ 4
- 18
models/ai_model_manage.go View File

@@ -242,30 +242,16 @@ func QueryModelConvertById(id string) (*AiModelConvert, error) {
if isExist { if isExist {
return re, nil 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
} }


func QueryModelById(id string) (*AiModelManage, error) { func QueryModelById(id string) (*AiModelManage, error) {
sess := x.NewSession() sess := x.NewSession()
defer sess.Close() 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 isExist {
return re, nil
} }
return nil, err return nil, err
} }


Loading…
Cancel
Save