From e16ef9be0aed3b9e774106c8cfc3c985ffd25efb Mon Sep 17 00:00:00 2001 From: samwaf Date: Wed, 24 May 2023 14:06:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E6=8F=90=E7=A4=BA=E9=94=99?= =?UTF-8?q?=E8=AF=AF=E6=B2=A1=E6=9C=89=E5=8F=8A=E6=97=B6return=E5=AF=BC?= =?UTF-8?q?=E8=87=B4=20=E5=90=8E=E7=BB=AD=E8=BF=98=E8=83=BD=E6=AD=A3?= =?UTF-8?q?=E5=B8=B8=E4=BD=BF=E7=94=A8=E7=9A=84=E6=83=85=E5=86=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/waf_rule.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/waf_rule.go b/api/waf_rule.go index 18c37fa..1a88b46 100644 --- a/api/waf_rule.go +++ b/api/waf_rule.go @@ -32,6 +32,7 @@ func (w *WafRuleAPi) AddApi(c *gin.Context) { err = wafRuleService.CheckIsExistApi(ruleInfo.RuleBase.RuleName, ruleInfo.RuleBase.RuleDomainCode) if err != nil { response.FailWithMessage("当前规则名称已存在", c) + return } chsName := ruleInfo.RuleBase.RuleName var ruleCode = uuid.NewV4().String() @@ -52,12 +53,15 @@ func (w *WafRuleAPi) AddApi(c *gin.Context) { if err == nil { w.NotifyWaf(ruleInfo.RuleBase.RuleDomainCode) response.OkWithMessage("添加成功", c) + return } else { response.FailWithMessage("添加失败", c) + return } } else { response.FailWithMessage("解析失败", c) + return } } func (w *WafRuleAPi) GetDetailApi(c *gin.Context) { @@ -110,6 +114,7 @@ func (w *WafRuleAPi) DelRuleApi(c *gin.Context) { //TODO 通知引擎重新加载某个网站的规则信息 if err != nil && errors.Is(err, gorm.ErrRecordNotFound) { response.FailWithMessage("请检测参数", c) + return } else if err != nil { response.FailWithMessage("发生错误", c) } else {