github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

simplify validity map

pull/82/head
v2ray 2016-02-01 23:21:44 +01:00
parent 5df2a1c6e6
commit 526c4558ca
1 changed files with 24 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
common/collect/validity_map.go
View File

@ -2,7 +2,7 @@ package collect
import (
"sync"
"time"
"sync/atomic"
"github.com/v2ray/v2ray-core/common/serial"
)
@ -19,20 +19,17 @@ type entry struct {
type ValidityMap struct {
sync.RWMutex
cache map[string]Validity
cleanupIntervalSec int
opCount int32
}
func NewValidityMap(cleanupIntervalSec int) *ValidityMap {
instance := &ValidityMap{
cache: make(map[string]Validity),
cleanupIntervalSec: cleanupIntervalSec,
}
go instance.cleanup()
return instance
}
func (this *ValidityMap) cleanup() {
for range time.Tick(time.Duration(this.cleanupIntervalSec) * time.Second) {
entry2Remove := make([]entry, 0, 128)
this.RLock()
for key, value := range this.cache {
@ -53,12 +50,16 @@ func (this *ValidityMap) cleanup() {
}
}
}
}
func (this *ValidityMap) Set(key serial.String, value Validity) {
this.Lock()
this.cache[key.String()] = value
this.Unlock()
opCount := atomic.AddInt32(&this.opCount, 1)
if opCount > 1000 {
atomic.StoreInt32(&this.opCount, 0)
go this.cleanup()
}
}
func (this *ValidityMap) Get(key serial.String) Validity {