diff --git a/internal/nodes/http_request.go b/internal/nodes/http_request.go index f7dee16..03c97de 100644 --- a/internal/nodes/http_request.go +++ b/internal/nodes/http_request.go @@ -177,6 +177,14 @@ func (this *HTTPRequest) doBegin() { } } + // 处理健康检查 + var healthCheckKey = this.RawReq.Header.Get(serverconfigs.HealthCheckHeaderName) + if len(healthCheckKey) > 0 { + if this.doHealthCheck(healthCheckKey) { + return + } + } + // 统计 if this.web.StatRef != nil && this.web.StatRef.IsOn { this.doStat() diff --git a/internal/nodes/http_request_health_check.go b/internal/nodes/http_request_health_check.go new file mode 100644 index 0000000..47b5a6d --- /dev/null +++ b/internal/nodes/http_request_health_check.go @@ -0,0 +1,27 @@ +// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved. + +package nodes + +import ( + "github.com/TeaOSLab/EdgeCommon/pkg/nodeutils" + "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs" + "github.com/TeaOSLab/EdgeNode/internal/remotelogs" +) + +// 健康检查 +func (this *HTTPRequest) doHealthCheck(key string) (stop bool) { + this.tags = append(this.tags, "healthCheck") + + this.RawReq.Header.Del(serverconfigs.HealthCheckHeaderName) + + data, err := nodeutils.DecryptData(sharedNodeConfig.NodeId, sharedNodeConfig.Secret, key) + if err != nil { + remotelogs.Error("HTTP_REQUEST_HEALTH_CHECK", "decode key failed: "+err.Error()) + return + } + + if data.GetBool("onlyBasicRequest") { + return true + } + return +}