fix: count error

2022-04-28 08:54:33 +08:00
parent ffb49ba4c5
commit 30eaa8add6
3 changed files with 9 additions and 3 deletions
--- a/rule/provider/ipcidr_strategy.go
+++ b/rule/provider/ipcidr_strategy.go
@ -26,16 +26,18 @@ func (i *ipcidrStrategy) ShouldResolveIP() bool {

 func (i *ipcidrStrategy) OnUpdate(rules []string) {
 	ipCidrTrie := trie.NewIpCidrTrie()
+	count := 0
 	for _, rule := range rules {
 		err := ipCidrTrie.AddIpCidrForString(rule)
 		if err != nil {
 			log.Warnln("invalid Ipcidr:[%s]", rule)
 		} else {
-			i.count++
+			count++
 		}
 	}

 	i.trie = ipCidrTrie
+	i.count = count
 	i.shouldResolveIP = i.count > 0
 }