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

增加服务流量统计

Browse Source
This commit is contained in:
GoEdgeLab
2021-06-08 11:24:41 +08:00
parent a29f90beda
commit 9ca60abdc1
11 changed files with 104 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/nodes/http_request.go
View File

@@ -61,6 +61,7 @@ type HTTPRequest struct {
rewriteIsExternalURL bool // 重写目标是否为外部URL
cacheRef *serverconfigs.HTTPCacheRef // 缓存设置
cacheKey string // 缓存使用的Key
isCached bool // 是否已经被缓存
// WAF相关
firewallPolicyId int64
@@ -231,7 +232,11 @@ func (this *HTTPRequest) doEnd() {
// 流量统计
// TODO 增加是否开启开关
if this.Server != nil {
stats.SharedTrafficStatManager.Add(this.Server.Id, this.writer.sentBodyBytes)
if this.isCached {
stats.SharedTrafficStatManager.Add(this.Server.Id, this.writer.sentBodyBytes, this.writer.sentBodyBytes, 1, 1)
} else {
stats.SharedTrafficStatManager.Add(this.Server.Id, this.writer.sentBodyBytes, 0, 1, 0)
}
}
}