mirror of https://github.com/v2ray/v2ray-core
improve performance on domain matcher
parent
787f37363b
commit
9de03d088e
|
@ -51,22 +51,38 @@ func (g *DomainMatcherGroup) addMatcher(m domainMatcher, value uint32) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (g *DomainMatcherGroup) Match(domain string) uint32 {
|
func (g *DomainMatcherGroup) Match(domain string) uint32 {
|
||||||
|
if len(domain) == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
current := g.root
|
current := g.root
|
||||||
if current == nil {
|
if current == nil {
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
parts := breakDomain(domain)
|
nextPart := func(idx int) int {
|
||||||
for i := len(parts) - 1; i >= 0; i-- {
|
for i := idx - 1; i >= 0; i-- {
|
||||||
part := parts[i]
|
if domain[i] == '.' {
|
||||||
if current.sub == nil {
|
return i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
|
||||||
|
idx := len(domain)
|
||||||
|
for {
|
||||||
|
if idx == -1 || current.sub == nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
|
nidx := nextPart(idx)
|
||||||
|
part := domain[nidx+1 : idx]
|
||||||
next := current.sub[part]
|
next := current.sub[part]
|
||||||
if next == nil {
|
if next == nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
current = next
|
current = next
|
||||||
|
idx = nidx
|
||||||
}
|
}
|
||||||
return current.value
|
return current.value
|
||||||
}
|
}
|
||||||
|
|
|
@ -34,6 +34,22 @@ func TestDomainMatcherGroup(t *testing.T) {
|
||||||
Domain: "c.a.b.com",
|
Domain: "c.a.b.com",
|
||||||
Result: 4,
|
Result: 4,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Domain: "c.a..b.com",
|
||||||
|
Result: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Domain: ".com",
|
||||||
|
Result: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Domain: "com",
|
||||||
|
Result: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Domain: "",
|
||||||
|
Result: 0,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, testCase := range testCases {
|
for _, testCase := range testCases {
|
||||||
|
|
Loading…
Reference in New Issue