lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-01-27 19:56:22 -05:00
Code Issues Releases Wiki Activity

don't alloc new maps until required

Browse Source
This commit is contained in:
Darien Raymond 2018-08-22 22:49:02 +02:00
parent 4e30ac33d0
commit e6b5356ea9
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
common/strmatcher/domain_matcher.go
View File

@ -19,18 +19,19 @@ type DomainMatcherGroup struct {
func (g *DomainMatcherGroup) Add(domain string, value uint32) { func (g *DomainMatcherGroup) Add(domain string, value uint32) {
if g.root == nil { if g.root == nil {
g.root = &node{ g.root = new(node)
sub: make(map[string]*node),
}
} }
current := g.root current := g.root
parts := breakDomain(domain) parts := breakDomain(domain)
for i := len(parts) - 1; i >= 0; i-- { for i := len(parts) - 1; i >= 0; i-- {
part := parts[i] part := parts[i]
if current.sub == nil {
current.sub = make(map[string]*node)
}
next := current.sub[part] next := current.sub[part]
if next == nil { if next == nil {
next = &node{sub: make(map[string]*node)} next = new(node)
current.sub[part] = next current.sub[part] = next
} }
current = next current = next
@ -52,6 +53,9 @@ func (g *DomainMatcherGroup) Match(domain string) uint32 {
parts := breakDomain(domain) parts := breakDomain(domain)
for i := len(parts) - 1; i >= 0; i-- { for i := len(parts) - 1; i >= 0; i-- {
part := parts[i] part := parts[i]
if current.sub == nil {
break
}
next := current.sub[part] next := current.sub[part]
if next == nil { if next == nil {
break break