51 lines
722 B
Go
51 lines
722 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type backoff struct {
|
||
|
|
userlist map[string]int64
|
||
|
|
sync.Mutex
|
||
|
|
}
|
||
|
|
|
||
|
|
var Delay = backoff{userlist: make(map[string]int64)}
|
||
|
|
|
||
|
|
func (u *backoff) AddUser(address string) {
|
||
|
|
|
||
|
|
u.Lock()
|
||
|
|
defer u.Unlock()
|
||
|
|
|
||
|
|
u.userlist[address] = time.Now().UnixMilli()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *backoff) DelUser(address string) {
|
||
|
|
delete(u.userlist, address)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *backoff) CheckUser(address string) bool {
|
||
|
|
|
||
|
|
u.Lock()
|
||
|
|
defer u.Unlock()
|
||
|
|
|
||
|
|
if _, ok := u.userlist[address]; ok {
|
||
|
|
return true
|
||
|
|
} else {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u *backoff) CheckBackoff() {
|
||
|
|
|
||
|
|
u.Lock()
|
||
|
|
defer u.Unlock()
|
||
|
|
|
||
|
|
for i, e := range u.userlist {
|
||
|
|
added := time.UnixMilli(e)
|
||
|
|
if time.Since(added) >= time.Minute*2 {
|
||
|
|
Delay.DelUser(i)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|