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:
GoEdgeLab
2023-09-12 10:45:44 +08:00
parent 33fb5cfb11
commit f43ad5a9d5
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
internal/nodes/http_request.go
View File

@@ -38,15 +38,16 @@ type HTTPRequest struct {
requestId string requestId string
// 外部参数 // 外部参数
RawReq *http.Request RawReq *http.Request
RawWriter http.ResponseWriter RawWriter http.ResponseWriter
ReqServer *serverconfigs.ServerConfig ReqServer *serverconfigs.ServerConfig
ReqHost string // 请求的Host ReqHost string // 请求的Host
ServerName string // 实际匹配到的Host ServerName string // 实际匹配到的Host
ServerAddr string // 实际启动的服务器监听地址 ServerAddr string // 实际启动的服务器监听地址
IsHTTP bool IsHTTP bool
IsHTTPS bool IsHTTPS bool
IsHTTP3 bool IsHTTP3 bool
isHealthCheck bool
// 共享参数 // 共享参数
nodeConfig *nodeconfigs.NodeConfig nodeConfig *nodeconfigs.NodeConfig
@@ -168,10 +169,9 @@ func (this *HTTPRequest) Do() {
} }
// 处理健康检查 // 处理健康检查
var isHealthCheck = false
var healthCheckKey = this.RawReq.Header.Get(serverconfigs.HealthCheckHeaderName) var healthCheckKey = this.RawReq.Header.Get(serverconfigs.HealthCheckHeaderName)
if len(healthCheckKey) > 0 { if len(healthCheckKey) > 0 {
if this.doHealthCheck(healthCheckKey, &isHealthCheck) { if this.doHealthCheck(healthCheckKey, &this.isHealthCheck) {
this.doEnd() this.doEnd()
return return
} }
@@ -205,7 +205,7 @@ func (this *HTTPRequest) Do() {
} }
// UAM // UAM
if !isHealthCheck { if !this.isHealthCheck {
if this.web.UAM != nil { if this.web.UAM != nil {
if this.web.UAM.IsOn { if this.web.UAM.IsOn {
if this.doUAM() { if this.doUAM() {
@@ -223,7 +223,7 @@ func (this *HTTPRequest) Do() {
} }
// CC // CC
if !isHealthCheck { if !this.isHealthCheck {
if this.web.CC != nil { if this.web.CC != nil {
if this.web.CC.IsOn { if this.web.CC.IsOn {
if this.doCC() { if this.doCC() {
@@ -388,7 +388,7 @@ func (this *HTTPRequest) doEnd() {
// 流量统计 // 流量统计
// TODO 增加是否开启开关 // TODO 增加是否开启开关
if this.ReqServer != nil && this.ReqServer.Id > 0 { if this.ReqServer != nil && this.ReqServer.Id > 0 && !this.isHealthCheck /** 健康检查时不统计 **/ {
var totalBytes int64 = 0 var totalBytes int64 = 0
var requestConn = this.RawReq.Context().Value(HTTPConnContextKey) var requestConn = this.RawReq.Context().Value(HTTPConnContextKey)