TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-28 02:40:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

实现WAF统计数据接口

Browse Source
This commit is contained in:
GoEdgeLab
2021-01-26 18:41:02 +08:00
parent 2f9fe99088
commit e7a01c567f
12 changed files with 395 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
internal/rpc/services/service_server_.go
View File

@@ -12,12 +12,13 @@ import (
)
// HTTP请求统计缓存队列
var serverHTTPCountryStatMap = map[string]int64{} // serverId@countryId@month => count
var serverHTTPProvinceStatMap = map[string]int64{} // serverId@provinceId@month => count
var serverHTTPCityStatMap = map[string]int64{} // serverId@cityId@month => count
var serverHTTPProviderStatMap = map[string]int64{} // serverId@providerId@month => count
var serverHTTPSystemStatMap = map[string]int64{} // serverId@systemId@version@month => count
var serverHTTPBrowserStatMap = map[string]int64{} // serverId@browserId@version@month => count
var serverHTTPCountryStatMap = map[string]int64{} // serverId@countryId@month => count
var serverHTTPProvinceStatMap = map[string]int64{} // serverId@provinceId@month => count
var serverHTTPCityStatMap = map[string]int64{} // serverId@cityId@month => count
var serverHTTPProviderStatMap = map[string]int64{} // serverId@providerId@month => count
var serverHTTPSystemStatMap = map[string]int64{} // serverId@systemId@version@month => count
var serverHTTPBrowserStatMap = map[string]int64{} // serverId@browserId@version@month => count
var serverHTTPFirewallRuleGroupStatMap = map[string]int64{} // serverId@firewallRuleGroupId@action@day => count
var serverStatLocker = sync.Mutex{}
func init() {
@@ -29,7 +30,7 @@ func init() {
var duration = 30 * time.Minute
if Tea.IsTesting() {
// 测试条件下缩短时间，以便进行观察
duration = 1 * time.Minute
duration = 10 * time.Second
}
ticker := time.NewTicker(duration)
for range ticker.C {
@@ -151,5 +152,24 @@ func (this *ServerService) dumpServerHTTPStats() error {
}
}
// 防火墙
{
serverStatLocker.Lock()
m := serverHTTPFirewallRuleGroupStatMap
serverHTTPFirewallRuleGroupStatMap = map[string]int64{}
serverStatLocker.Unlock()
for k, count := range m {
pieces := strings.Split(k, "@")
if len(pieces) != 4 {
continue
}
err := stats.SharedServerHTTPFirewallDailyStatDAO.IncreaseDailyCount(nil, types.Int64(pieces[0]), types.Int64(pieces[1]), pieces[2], pieces[3], count)
if err != nil {
return err
}
}
}
return nil
}