46 lines
727 B
Go
46 lines
727 B
Go
package sessdata
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/bjdgyc/anylink/base"
|
|
)
|
|
|
|
const limitAllKey = "__ALL__"
|
|
|
|
var (
|
|
limitClient = map[string]int{limitAllKey: 0}
|
|
limitMux = sync.Mutex{}
|
|
)
|
|
|
|
func LimitClient(user string, close bool) bool {
|
|
limitMux.Lock()
|
|
defer limitMux.Unlock()
|
|
|
|
_all := limitClient[limitAllKey]
|
|
c, ok := limitClient[user]
|
|
if !ok { // 不存在用户
|
|
limitClient[user] = 0
|
|
}
|
|
|
|
if close {
|
|
limitClient[user] = c - 1
|
|
limitClient[limitAllKey] = _all - 1
|
|
return true
|
|
}
|
|
|
|
// 全局判断
|
|
if _all >= base.Cfg.MaxClient {
|
|
return false
|
|
}
|
|
|
|
// 超出同一个用户限制
|
|
if c >= base.Cfg.MaxUserClient {
|
|
return false
|
|
}
|
|
|
|
limitClient[user] = c + 1
|
|
limitClient[limitAllKey] = _all + 1
|
|
return true
|
|
}
|