Compare commits

...

3 Commits

Author SHA1 Message Date
Kaive Young
ae2b86a748
Merge adb4d3d33d1e174e252acf1d55a07758a6c3d0e1 into 002831ea823edfa2ab3695b3f4fc0404979bcedf 2024-03-20 02:52:29 +00:00
Kaive Young
adb4d3d33d
add header for http health check 2024-03-20 10:52:09 +08:00
Kaive Young
bb7d51b0db
add header for http health check 2024-03-20 10:45:36 +08:00
3 changed files with 4 additions and 10 deletions

View File

@ -42,7 +42,6 @@ type Monitor struct {
// For http // For http
url string url string
header http.Header header http.Header
host string
failedTimes uint64 failedTimes uint64
statusOK bool statusOK bool
statusNormalFn func() statusNormalFn func()
@ -75,7 +74,7 @@ func NewMonitor(ctx context.Context, cfg v1.HealthCheckConfig, addr string,
url = s + cfg.Path url = s + cfg.Path
} }
header := make(http.Header) header := make(http.Header)
for _, h := range cfg.Headers { for _, h := range cfg.HTTPHeaders {
header.Set(h.Name, h.Value) header.Set(h.Name, h.Value)
} }
@ -87,7 +86,6 @@ func NewMonitor(ctx context.Context, cfg v1.HealthCheckConfig, addr string,
addr: addr, addr: addr,
url: url, url: url,
header: header, header: header,
host: cfg.HostHeaderRewrite,
statusOK: false, statusOK: false,
statusNormalFn: statusNormalFn, statusNormalFn: statusNormalFn,
statusFailedFn: statusFailedFn, statusFailedFn: statusFailedFn,
@ -172,7 +170,7 @@ func (monitor *Monitor) doHTTPCheck(ctx context.Context) error {
return err return err
} }
req.Header = monitor.header req.Header = monitor.header
req.Host = monitor.host req.Host = monitor.header.Get("Host")
resp, err := http.DefaultClient.Do(req) resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
return err return err

View File

@ -217,10 +217,9 @@ healthCheck.intervalSeconds = 10
healthCheck.maxFailed = 3 healthCheck.maxFailed = 3
healthCheck.timeoutSeconds = 3 healthCheck.timeoutSeconds = 3
# set health check headers # set health check headers
healthCheck.headers=[ healthCheck.httpheaders=[
{ name = "x-from-where", value = "frp" } { name = "x-from-where", value = "frp" }
] ]
healthCheck.hostHeaderRewrite = "example.com"
[[proxies]] [[proxies]]
name = "web02" name = "web02"

View File

@ -99,10 +99,7 @@ type HealthCheckConfig struct {
Path string `json:"path,omitempty"` Path string `json:"path,omitempty"`
// Headers specifies the headers to send health checks to if the // Headers specifies the headers to send health checks to if the
// health check type is "http". // health check type is "http".
Headers []HTTPHeader `json:"headers,omitempty"` HTTPHeaders []HTTPHeader `json:"httpHeaders,omitempty"`
// HostHeaderRewrite specifies the request host to send health checks to if the
// health check type is "http".
HostHeaderRewrite string `json:"hostHeaderRewrite,omitempty"`
} }
type DomainConfig struct { type DomainConfig struct {