mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-13 06:23:48 -04:00
87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
package core
|
|
|
|
import (
|
|
"time"
|
|
|
|
v2hash "github.com/v2ray/v2ray-core/hash"
|
|
)
|
|
|
|
const (
|
|
updateIntervalSec = 10
|
|
cacheDurationSec = 120
|
|
)
|
|
|
|
type UserSet interface {
|
|
AddUser(user User) error
|
|
GetUser(timeHash []byte) (*ID, int64, bool)
|
|
}
|
|
|
|
type TimedUserSet struct {
|
|
validUserIds []ID
|
|
userHashes map[string]indexTimePair
|
|
}
|
|
|
|
type indexTimePair struct {
|
|
index int
|
|
timeSec int64
|
|
}
|
|
|
|
type hashEntry struct {
|
|
hash string
|
|
timeSec int64
|
|
}
|
|
|
|
func NewTimedUserSet() UserSet {
|
|
vuSet := new(TimedUserSet)
|
|
vuSet.validUserIds = make([]ID, 0, 16)
|
|
vuSet.userHashes = make(map[string]indexTimePair)
|
|
|
|
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*3*len(us.validUserIds))
|
|
lastSec2Remove := now.Unix()
|
|
idHash := v2hash.NewTimeHash(v2hash.HMACHash{})
|
|
for {
|
|
now := <-tick
|
|
nowSec := now.UTC().Unix()
|
|
|
|
remove2Sec := nowSec - cacheDurationSec
|
|
if remove2Sec > lastSec2Remove {
|
|
for lastSec2Remove+1 < remove2Sec {
|
|
entry := <-hash2Remove
|
|
lastSec2Remove = entry.timeSec
|
|
delete(us.userHashes, entry.hash)
|
|
}
|
|
}
|
|
|
|
for lastSec < nowSec+cacheDurationSec {
|
|
for idx, id := range us.validUserIds {
|
|
idHash := idHash.Hash(id.Bytes, lastSec)
|
|
hash2Remove <- hashEntry{string(idHash), lastSec}
|
|
us.userHashes[string(idHash)] = indexTimePair{idx, lastSec}
|
|
}
|
|
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, int64, bool) {
|
|
pair, found := us.userHashes[string(userHash)]
|
|
if found {
|
|
return &us.validUserIds[pair.index], pair.timeSec, true
|
|
}
|
|
return nil, 0, false
|
|
}
|