history-museum/Clash.Meta
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: rule-provider now read yaml line-by-line

Browse Source
This commit is contained in:
wwqgtxx
2023-04-01 14:11:09 +08:00
parent 54cad53f5f
commit 54c2fa98b4
4 changed files with 148 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
rules/provider/domain_strategy.go
View File

@ -7,8 +7,9 @@ import (
)
type domainStrategy struct {
count int
domainRules *trie.DomainSet
count int
domainTrie *trie.DomainTrie[struct{}]
domainSet *trie.DomainSet
}
func (d *domainStrategy) ShouldFindProcess() bool {
@ -16,7 +17,7 @@ func (d *domainStrategy) ShouldFindProcess() bool {
}
func (d *domainStrategy) Match(metadata *C.Metadata) bool {
return d.domainRules != nil && d.domainRules.Has(metadata.RuleHost())
return d.domainSet != nil && d.domainSet.Has(metadata.RuleHost())
}
func (d *domainStrategy) Count() int {
@ -27,16 +28,24 @@ func (d *domainStrategy) ShouldResolveIP() bool {
return false
}
func (d *domainStrategy) OnUpdate(rules []string) {
domainTrie := trie.New[struct{}]()
for _, rule := range rules {
err := domainTrie.Insert(rule, struct{}{})
if err != nil {
log.Warnln("invalid domain:[%s]", rule)
}
func (d *domainStrategy) Reset() {
d.domainTrie = trie.New[struct{}]()
d.domainSet = nil
d.count = 0
}
func (d *domainStrategy) Insert(rule string) {
err := d.domainTrie.Insert(rule, struct{}{})
if err != nil {
log.Warnln("invalid domain:[%s]", rule)
} else {
d.count++
}
d.domainRules = domainTrie.NewDomainSet()
d.count = len(rules)
}
func (d *domainStrategy) FinishInsert() {
d.domainSet = d.domainTrie.NewDomainSet()
d.domainTrie = nil
}
func NewDomainStrategy() *domainStrategy {