TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-07 10:40:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加WAF更新通知

Browse Source
This commit is contained in:
GoEdgeLab
2021-02-02 16:22:47 +08:00
parent 04215db205
commit 6c4cf4eb30
1 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/db/models/http_firewall_policy_dao.go
View File

@@ -52,12 +52,16 @@ func (this *HTTPFirewallPolicyDAO) EnableHTTPFirewallPolicy(tx *dbs.Tx, id int64
} }
// 禁用条目 // 禁用条目
func (this *HTTPFirewallPolicyDAO) DisableHTTPFirewallPolicy(tx *dbs.Tx, id int64) error { func (this *HTTPFirewallPolicyDAO) DisableHTTPFirewallPolicy(tx *dbs.Tx, policyId int64) error {
_, err := this.Query(tx). _, err := this.Query(tx).
Pk(id). Pk(policyId).
Set("state", HTTPFirewallPolicyStateDisabled). Set("state", HTTPFirewallPolicyStateDisabled).
Update() Update()
if err != nil {
return err return err
}
return this.NotifyUpdate(tx, policyId)
} }
// 查找启用中的条目 // 查找启用中的条目
@@ -127,7 +131,11 @@ func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicyInboundAndOutbound(tx *db
op.Outbound = "null" op.Outbound = "null"
} }
err := this.Save(tx, op) err := this.Save(tx, op)
if err != nil {
return err return err
}
return this.NotifyUpdate(tx, policyId)
} }
// 修改策略的Inbound // 修改策略的Inbound
@@ -143,7 +151,11 @@ func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicyInbound(tx *dbs.Tx, polic
op.Inbound = "null" op.Inbound = "null"
} }
err := this.Save(tx, op) err := this.Save(tx, op)
if err != nil {
return err return err
}
return this.NotifyUpdate(tx, policyId)
} }
// 修改策略 // 修改策略
@@ -170,7 +182,11 @@ func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicy(tx *dbs.Tx, policyId int
op.BlockOptions = blockOptionsJSON op.BlockOptions = blockOptionsJSON
} }
err := this.Save(tx, op) err := this.Save(tx, op)
if err != nil {
return err return err
}
return this.NotifyUpdate(tx, policyId)
} }
// 计算所有可用的策略数量 // 计算所有可用的策略数量