mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
1.9 KiB
89 lines
1.9 KiB
package core
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/v2ray/v2ray-core/log"
|
|
)
|
|
|
|
const (
|
|
updateIntervalSec = 10
|
|
cacheDurationSec = 120
|
|
)
|
|
|
|
type UserSet interface {
|
|
AddUser(user User) error
|
|
GetUser(timeHash []byte) (*ID, bool)
|
|
}
|
|
|
|
type TimedUserSet struct {
|
|
validUserIds []ID
|
|
userHashes map[string]int
|
|
}
|
|
|
|
type hashEntry struct {
|
|
hash string
|
|
timeSec int64
|
|
}
|
|
|
|
func NewTimedUserSet() UserSet {
|
|
vuSet := new(TimedUserSet)
|
|
vuSet.validUserIds = make([]ID, 0, 16)
|
|
vuSet.userHashes = make(map[string]int)
|
|
|
|
go vuSet.updateUserHash(time.Tick(updateIntervalSec * time.Second))
|
|
return vuSet
|
|
}
|
|
|
|
func (us *TimedUserSet) updateUserHash(tick <-chan time.Time) {
|
|
now := time.Now().UTC()
|
|
lastSec := now.Unix() - cacheDurationSec
|
|
|
|
hash2Remove := make(chan hashEntry, cacheDurationSec*2*len(us.validUserIds))
|
|
lastSec2Remove := now.Unix()
|
|
log.Debug("Start updating")
|
|
for {
|
|
now := <-tick
|
|
nowSec := now.UTC().Unix()
|
|
|
|
remove2Sec := nowSec - cacheDurationSec
|
|
|
|
log.Debug("remove2Sec %d, to %d", lastSec2Remove, remove2Sec)
|
|
|
|
|
|
if remove2Sec > lastSec2Remove {
|
|
for lastSec2Remove+1 < remove2Sec {
|
|
entry := <-hash2Remove
|
|
lastSec2Remove = entry.timeSec
|
|
delete(us.userHashes, entry.hash)
|
|
}
|
|
}
|
|
|
|
log.Debug("LastSec %d, to %d", lastSec, nowSec + updateIntervalSec)
|
|
|
|
for lastSec < nowSec + cacheDurationSec {
|
|
for idx, id := range us.validUserIds {
|
|
idHash := id.TimeHash(lastSec)
|
|
hash2Remove <- hashEntry{string(idHash), lastSec}
|
|
log.Debug("Hash: %v", idHash)
|
|
us.userHashes[string(idHash)] = idx
|
|
}
|
|
lastSec ++
|
|
}
|
|
}
|
|
}
|
|
|
|
func (us *TimedUserSet) AddUser(user User) error {
|
|
id := user.Id
|
|
us.validUserIds = append(us.validUserIds, id)
|
|
return nil
|
|
}
|
|
|
|
func (us TimedUserSet) GetUser(userHash []byte) (*ID, bool) {
|
|
idIndex, found := us.userHashes[string(userHash)]
|
|
if found {
|
|
return &us.validUserIds[idIndex], true
|
|
}
|
|
return nil, false
|
|
}
|