GoDoxy/internal/watcher/health/monitor/raw.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
}