TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-07 18:50:26 +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,14 +52,18 @@ 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)
}
// 查找启用中的条目 // 查找启用中的条目
func (this *HTTPFirewallPolicyDAO) FindEnabledHTTPFirewallPolicy(tx *dbs.Tx, id int64) (*HTTPFirewallPolicy, error) { func (this *HTTPFirewallPolicyDAO) FindEnabledHTTPFirewallPolicy(tx *dbs.Tx, id int64) (*HTTPFirewallPolicy, error) {
result, err := this.Query(tx). result, err := this.Query(tx).
@@ -127,9 +131,13 @@ 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
func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicyInbound(tx *dbs.Tx, policyId int64, inboundJSON []byte) error { func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicyInbound(tx *dbs.Tx, policyId int64, inboundJSON []byte) error {
if policyId <= 0 { if policyId <= 0 {
@@ -143,9 +151,13 @@ 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)
}
// 修改策略 // 修改策略
func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicy(tx *dbs.Tx, policyId int64, isOn bool, name string, description string, inboundJSON []byte, outboundJSON []byte, blockOptionsJSON []byte) error { func (this *HTTPFirewallPolicyDAO) UpdateFirewallPolicy(tx *dbs.Tx, policyId int64, isOn bool, name string, description string, inboundJSON []byte, outboundJSON []byte, blockOptionsJSON []byte) error {
if policyId <= 0 { if policyId <= 0 {
@@ -170,9 +182,13 @@ 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)
}
// 计算所有可用的策略数量 // 计算所有可用的策略数量
func (this *HTTPFirewallPolicyDAO) CountAllEnabledFirewallPolicies(tx *dbs.Tx) (int64, error) { func (this *HTTPFirewallPolicyDAO) CountAllEnabledFirewallPolicies(tx *dbs.Tx) (int64, error) {
return this.Query(tx). return this.Query(tx).