package settings import ( "fmt" "os" "time" "github.com/qdm12/gosettings" "github.com/qdm12/gotree" "github.com/qdm12/govalid/address" ) // Health contains settings for the healthcheck and health server. type Health struct { // ServerAddress is the listening address // for the health check server. // It cannot be the empty string in the internal state. ServerAddress string // ReadHeaderTimeout is the HTTP server header read timeout // duration of the HTTP server. It defaults to 100 milliseconds. ReadHeaderTimeout time.Duration // ReadTimeout is the HTTP read timeout duration of the // HTTP server. It defaults to 500 milliseconds. ReadTimeout time.Duration // TargetAddress is the address (host or host:port) // to TCP dial to periodically for the health check. // It cannot be the empty string in the internal state. TargetAddress string // SuccessWait is the duration to wait to re-run the // healthcheck after a successful healthcheck. // It defaults to 5 seconds and cannot be zero in // the internal state. SuccessWait time.Duration // VPN has health settings specific to the VPN loop. VPN HealthyWait } func (h Health) Validate() (err error) { uid := os.Getuid() err = address.Validate(h.ServerAddress, address.OptionListening(uid)) if err != nil { return fmt.Errorf("server listening address is not valid: %w", err) } err = h.VPN.validate() if err != nil { return fmt.Errorf("health VPN settings: %w", err) } return nil } func (h *Health) copy() (copied Health) { return Health{ ServerAddress: h.ServerAddress, ReadHeaderTimeout: h.ReadHeaderTimeout, ReadTimeout: h.ReadTimeout, TargetAddress: h.TargetAddress, SuccessWait: h.SuccessWait, VPN: h.VPN.copy(), } } // MergeWith merges the other settings into any // unset field of the receiver settings object. func (h *Health) MergeWith(other Health) { h.ServerAddress = gosettings.MergeWithString(h.ServerAddress, other.ServerAddress) h.ReadHeaderTimeout = gosettings.MergeWithNumber(h.ReadHeaderTimeout, other.ReadHeaderTimeout) h.ReadTimeout = gosettings.MergeWithNumber(h.ReadTimeout, other.ReadTimeout) h.TargetAddress = gosettings.MergeWithString(h.TargetAddress, other.TargetAddress) h.SuccessWait = gosettings.MergeWithNumber(h.SuccessWait, other.SuccessWait) h.VPN.mergeWith(other.VPN) } // OverrideWith overrides fields of the receiver // settings object with any field set in the other // settings. func (h *Health) OverrideWith(other Health) { h.ServerAddress = gosettings.OverrideWithString(h.ServerAddress, other.ServerAddress) h.ReadHeaderTimeout = gosettings.OverrideWithNumber(h.ReadHeaderTimeout, other.ReadHeaderTimeout) h.ReadTimeout = gosettings.OverrideWithNumber(h.ReadTimeout, other.ReadTimeout) h.TargetAddress = gosettings.OverrideWithString(h.TargetAddress, other.TargetAddress) h.SuccessWait = gosettings.OverrideWithNumber(h.SuccessWait, other.SuccessWait) h.VPN.overrideWith(other.VPN) } func (h *Health) SetDefaults() { h.ServerAddress = gosettings.DefaultString(h.ServerAddress, "127.0.0.1:9999") const defaultReadHeaderTimeout = 100 * time.Millisecond h.ReadHeaderTimeout = gosettings.DefaultNumber(h.ReadHeaderTimeout, defaultReadHeaderTimeout) const defaultReadTimeout = 500 * time.Millisecond h.ReadTimeout = gosettings.DefaultNumber(h.ReadTimeout, defaultReadTimeout) h.TargetAddress = gosettings.DefaultString(h.TargetAddress, "cloudflare.com:443") const defaultSuccessWait = 5 * time.Second h.SuccessWait = gosettings.DefaultNumber(h.SuccessWait, defaultSuccessWait) h.VPN.setDefaults() } func (h Health) String() string { return h.toLinesNode().String() } func (h Health) toLinesNode() (node *gotree.Node) { node = gotree.New("Health settings:") node.Appendf("Server listening address: %s", h.ServerAddress) node.Appendf("Target address: %s", h.TargetAddress) node.Appendf("Duration to wait after success: %s", h.SuccessWait) node.Appendf("Read header timeout: %s", h.ReadHeaderTimeout) node.Appendf("Read timeout: %s", h.ReadTimeout) node.AppendNode(h.VPN.toLinesNode("VPN")) return node }