diff --git a/main.go b/main.go index 2b5840b..11d9f9f 100644 --- a/main.go +++ b/main.go @@ -125,7 +125,7 @@ func main() { for { select { case msg := <-global.GWAF_CHAN_MSG: - if wafEngine.HostCode[msg.HostCode] != "" && wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]] != nil { + if wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]] != nil && wafEngine.HostCode[msg.HostCode] != "" { switch msg.Type { case enums.ChanTypeWhiteIP: wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]].IPWhiteLists = msg.Content.([]model.IPWhiteList) diff --git a/service/waf_service/waf_host.go b/service/waf_service/waf_host.go index 190b493..a650465 100644 --- a/service/waf_service/waf_host.go +++ b/service/waf_service/waf_host.go @@ -88,6 +88,20 @@ func (receiver *WafHostService) DelHostApi(req request.WafHostDelReq) error { return err } err = global.GWAF_LOCAL_DB.Where("CODE = ?", req.CODE).Delete(model.Hosts{}).Error + //删除规则 + err = global.GWAF_LOCAL_DB.Where("HostCode = ?", req.CODE).Delete(model.Rules{}).Error + //删除Anticc + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.AntiCC{}).Error + //删除禁用ip + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.IPBlockList{}).Error + //删除禁用url + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.URLBlockList{}).Error + //删除隐私保护url + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.LDPUrl{}).Error + //删除白名单ip + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.IPWhiteList{}).Error + //删除白名单URL + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.URLWhiteList{}).Error return err } func (receiver *WafHostService) ModifyGuardStatusApi(req request.WafHostGuardStatusReq) error {