TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-06 18:10:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

限制单次处理的服务城市数量

Browse Source
This commit is contained in:
刘祥超
2023-04-19 22:06:39 +08:00
parent 7e8c09a684
commit fc401a1426
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
internal/stats/http_request_stat_manager.go
View File

@@ -204,22 +204,22 @@ func (this *HTTPRequestStatManager) Loop() error {
if len(pieces) < 4 { if len(pieces) < 4 {
return nil return nil
} }
var serverId = pieces[0] var serverIdString = pieces[0]
var ip = pieces[1] var ip = pieces[1]
var result = iplib.LookupIP(ip) var result = iplib.LookupIP(ip)
if result != nil && result.IsOk() { if result != nil && result.IsOk() {
this.locker.Lock() this.locker.Lock()
if result.CountryId() > 0 { if result.CountryId() > 0 {
var key = serverId + "@" + types.String(result.CountryId()) + "@" + types.String(result.ProvinceId()) + "@" + types.String(result.CityId()) var key = serverIdString + "@" + types.String(result.CountryId()) + "@" + types.String(result.ProvinceId()) + "@" + types.String(result.CityId())
stat, ok := this.cityMap[key] stat, ok := this.cityMap[key]
if !ok { if !ok {
// 检查数量 // 检查数量
if this.serverCityCountMap[key] > 128 { // 限制单个服务的城市数量，防止数量过多 if this.serverCityCountMap[serverIdString] > 128 { // 限制单个服务的城市数量，防止数量过多
this.locker.Unlock() this.locker.Unlock()
return nil return nil
} }
this.serverCityCountMap[key]++ // 需要放在限制之后，因为使用的是int16 this.serverCityCountMap[serverIdString]++ // 需要放在限制之后，因为使用的是int16
stat = &StatItem{} stat = &StatItem{}
this.cityMap[key] = stat this.cityMap[key] = stat
@@ -233,9 +233,9 @@ func (this *HTTPRequestStatManager) Loop() error {
} }
if result.ProviderId() > 0 { if result.ProviderId() > 0 {
this.providerMap[serverId+"@"+types.String(result.ProviderId())]++ this.providerMap[serverIdString+"@"+types.String(result.ProviderId())]++
} else if utils.IsLocalIP(ip) { // 局域网IP } else if utils.IsLocalIP(ip) { // 局域网IP
this.providerMap[serverId+"@258"]++ this.providerMap[serverIdString+"@258"]++
} }
this.locker.Unlock() this.locker.Unlock()
} }
@@ -244,7 +244,7 @@ func (this *HTTPRequestStatManager) Loop() error {
if atIndex < 0 { if atIndex < 0 {
return nil return nil
} }
var serverId = userAgentString[:atIndex] var serverIdString = userAgentString[:atIndex]
var userAgent = userAgentString[atIndex+1:] var userAgent = userAgentString[atIndex+1:]
var result = SharedUserAgentParser.Parse(userAgent) var result = SharedUserAgentParser.Parse(userAgent)
@@ -256,11 +256,11 @@ func (this *HTTPRequestStatManager) Loop() error {
} }
this.locker.Lock() this.locker.Lock()
var systemKey = serverId + "@" + osInfo.Name + "@" + osInfo.Version var systemKey = serverIdString + "@" + osInfo.Name + "@" + osInfo.Version
_, ok := this.systemMap[systemKey] _, ok := this.systemMap[systemKey]
if !ok { if !ok {
if this.serverSystemCountMap[serverId] < 128 { // 限制最大数据，防止攻击 if this.serverSystemCountMap[serverIdString] < 128 { // 限制最大数据，防止攻击
this.serverSystemCountMap[serverId]++ this.serverSystemCountMap[serverIdString]++
ok = true ok = true
} }
} }
@@ -278,11 +278,11 @@ func (this *HTTPRequestStatManager) Loop() error {
} }
this.locker.Lock() this.locker.Lock()
var browserKey = serverId + "@" + browser + "@" + browserVersion var browserKey = serverIdString + "@" + browser + "@" + browserVersion
_, ok := this.browserMap[browserKey] _, ok := this.browserMap[browserKey]
if !ok { if !ok {
if this.serverBrowserCountMap[serverId] < 256 { // 限制最大数据，防止攻击 if this.serverBrowserCountMap[serverIdString] < 256 { // 限制最大数据，防止攻击
this.serverBrowserCountMap[serverId]++ this.serverBrowserCountMap[serverIdString]++
ok = true ok = true
} }
} }