Browse Source

修正SQL注入问题

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

+ 2
- 2
models/ai_model_manage.go View File

@@ -239,7 +239,7 @@ func QueryModelConvertById(id string) (*AiModelConvert, error) {
defer sess.Close()
re := new(AiModelConvert)
isExist, err := sess.Table(new(AiModelConvert)).ID(id).Get(re)
if isExist {
if err == nil && isExist {
return re, nil
}
return nil, err
@@ -250,7 +250,7 @@ func QueryModelById(id string) (*AiModelManage, error) {
defer sess.Close()
re := new(AiModelManage)
isExist, err := sess.Table(new(AiModelManage)).ID(id).Get(re)
if isExist {
if err == nil && isExist {
return re, nil
}
return nil, err


Loading…
Cancel
Save