mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 04:42:33 +02:00
1a5f3735cf
* cleanup code for URL type * fix makefile for trace mode * refactor, merge Entry, RawEntry and Route into one. * Implement fileserver. * refactor: rename HTTPRoute to ReverseProxyRoute to avoid confusion * refactor: move metrics logger to middleware package - fix prometheus metrics for load balanced routes - route will now fail when health monitor fail to start * fix extra output of ls-* commands by defer initializaing stuff, speed up start time * add test for path traversal attack, small fix on FileServer.Start method * rename rule.on.bypass to pass * refactor and fixed map-to-map deserialization * updated route loading logic * schemas: add "add_prefix" option to modify_request middleware * updated route JSONMarshalling --------- Co-authored-by: yusing <yusing@6uo.me>
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package monitor
|
|
|
|
import (
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/yusing/go-proxy/internal/net/types"
|
|
"github.com/yusing/go-proxy/internal/watcher/health"
|
|
)
|
|
|
|
type (
|
|
RawHealthMonitor struct {
|
|
*monitor
|
|
dialer *net.Dialer
|
|
}
|
|
)
|
|
|
|
func NewRawHealthMonitor(url *types.URL, config *health.HealthCheckConfig) *RawHealthMonitor {
|
|
mon := new(RawHealthMonitor)
|
|
mon.monitor = newMonitor(url, config, mon.CheckHealth)
|
|
mon.dialer = &net.Dialer{
|
|
Timeout: config.Timeout,
|
|
FallbackDelay: -1,
|
|
}
|
|
return mon
|
|
}
|
|
|
|
func NewRawHealthChecker(url *types.URL, config *health.HealthCheckConfig) health.HealthChecker {
|
|
return NewRawHealthMonitor(url, config)
|
|
}
|
|
|
|
func (mon *RawHealthMonitor) CheckHealth() (result *health.HealthCheckResult, err error) {
|
|
ctx, cancel := mon.ContextWithTimeout("ping request timed out")
|
|
defer cancel()
|
|
|
|
url := mon.url.Load()
|
|
start := time.Now()
|
|
conn, dialErr := mon.dialer.DialContext(ctx, url.Scheme, url.Host)
|
|
result = &health.HealthCheckResult{
|
|
Latency: time.Since(start),
|
|
}
|
|
if dialErr != nil {
|
|
result.Detail = dialErr.Error()
|
|
return
|
|
}
|
|
conn.Close()
|
|
result.Healthy = true
|
|
return
|
|
}
|