leftyer/SamWaf
1
0
Fork 0
mirror of https://gitee.com/samwaf/SamWaf.git synced 2025-12-06 14:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

修正提示错误没有及时return导致 后续还能正常使用的情况

Browse Source
This commit is contained in:
samwaf
2023-05-24 14:06:34 +08:00
parent 24ced8fdb4
commit e16ef9be0a
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/waf_rule.go
View File

@@ -32,6 +32,7 @@ func (w *WafRuleAPi) AddApi(c *gin.Context) {
err = wafRuleService.CheckIsExistApi(ruleInfo.RuleBase.RuleName, ruleInfo.RuleBase.RuleDomainCode) err = wafRuleService.CheckIsExistApi(ruleInfo.RuleBase.RuleName, ruleInfo.RuleBase.RuleDomainCode)
if err != nil { if err != nil {
response.FailWithMessage("当前规则名称已存在", c) response.FailWithMessage("当前规则名称已存在", c)
return
} }
chsName := ruleInfo.RuleBase.RuleName chsName := ruleInfo.RuleBase.RuleName
var ruleCode = uuid.NewV4().String() var ruleCode = uuid.NewV4().String()
@@ -52,12 +53,15 @@ func (w *WafRuleAPi) AddApi(c *gin.Context) {
if err == nil { if err == nil {
w.NotifyWaf(ruleInfo.RuleBase.RuleDomainCode) w.NotifyWaf(ruleInfo.RuleBase.RuleDomainCode)
response.OkWithMessage("添加成功", c) response.OkWithMessage("添加成功", c)
return
} else { } else {
response.FailWithMessage("添加失败", c) response.FailWithMessage("添加失败", c)
return
} }
} else { } else {
response.FailWithMessage("解析失败", c) response.FailWithMessage("解析失败", c)
return
} }
} }
func (w *WafRuleAPi) GetDetailApi(c *gin.Context) { func (w *WafRuleAPi) GetDetailApi(c *gin.Context) {
@@ -110,6 +114,7 @@ func (w *WafRuleAPi) DelRuleApi(c *gin.Context) {
//TODO 通知引擎重新加载某个网站的规则信息 //TODO 通知引擎重新加载某个网站的规则信息
if err != nil && errors.Is(err, gorm.ErrRecordNotFound) { if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
response.FailWithMessage("请检测参数", c) response.FailWithMessage("请检测参数", c)
return
} else if err != nil { } else if err != nil {
response.FailWithMessage("发生错误", c) response.FailWithMessage("发生错误", c)
} else { } else {