simplify validity map

common/collect/validity_map.go

@@ -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 {
@@ -52,13 +49,17 @@ func (this *ValidityMap) cleanup() {
 			this.Unlock()
 		}
 	}
-	}
 }
 
 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 {