tcjlj/V2bX
1
0
Fork 0
mirror of https://github.com/wyx2685/V2bX.git synced 2025-01-22 09:58:14 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

add time check for not realtime clear online ip

Browse Source
This commit is contained in:
yuzuki999 2023-05-19 08:47:28 +08:00
parent 30549a4ad1
commit 8890b566d5
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
limiter/conn.go
View File

@ -2,6 +2,7 @@ package limiter
import (
"sync"
"time"
)
type ConnLimiter struct {
@ -46,7 +47,7 @@ func (c *ConnLimiter) AddConnCount(user string, ip string, isTcp bool) (limit bo
ipMap.Store(ip, 1)
}
} else {
ipMap.Store(ip, struct{}{})
ipMap.Store(ip, time.Now())
}
// check user online ip
if v, ok := c.ip.LoadOrStore(user, ipMap); ok {
@ -60,6 +61,8 @@ func (c *ConnLimiter) AddConnCount(user string, ip string, isTcp bool) (limit bo
// count add
ips.Store(ip, online.(int)+2)
}
} else {
ips.Store(ip, time.Now())
}
} else {
// not online ip
@ -81,7 +84,7 @@ func (c *ConnLimiter) AddConnCount(user string, ip string, isTcp bool) (limit bo
ips.Store(ip, 1)
}
} else {
ips.Store(ip, struct{}{})
ips.Store(ip, time.Now())
}
}
}
@ -130,14 +133,18 @@ func (c *ConnLimiter) ClearOnlineIP() {
c.ip.Range(func(_, v any) bool {
userIp := v.(*sync.Map)
userIp.Range(func(ip, v any) bool {
if c.realtime {
// clear not realtime ip
userIp.Delete(ip)
if _, ok := v.(int); ok {
if v.(int) == 1 {
// clear packet ip for realtime
userIp.Delete(ip)
}
return true
}
if v.(int) == 1 {
// clear packet ip for realtime
userIp.Delete(ip)
} else {
// clear ip for not realtime
if v.(time.Time).Before(time.Now().Add(time.Minute)) {
// 1 minute no active
userIp.Delete(ip)
}
}
return true
})