105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package helm
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/compose-spec/compose-go/types"
|
|
)
|
|
|
|
// Probe is a struct that can be used to create a Liveness or Readiness probe.
|
|
type Probe struct {
|
|
HttpGet *HttpGet `yaml:"httpGet,omitempty"`
|
|
Exec *Exec `yaml:"exec,omitempty"`
|
|
TCP *TCP `yaml:"tcpSocket,omitempty"`
|
|
Period float64 `yaml:"periodSeconds"`
|
|
InitialDelay float64 `yaml:"initialDelaySeconds"`
|
|
Success uint64 `yaml:"successThreshold"`
|
|
Failure uint64 `yaml:"failureThreshold"`
|
|
}
|
|
|
|
// Create a new Probe object that can be apply to HttpProbe or TCPProbe.
|
|
func NewProbe(period, initialDelaySeconds float64, success, failure uint64) *Probe {
|
|
probe := &Probe{
|
|
Period: period,
|
|
Success: success,
|
|
Failure: failure,
|
|
InitialDelay: initialDelaySeconds,
|
|
}
|
|
|
|
// fix default values from
|
|
// https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-startup-probes/
|
|
if period == 0 {
|
|
probe.Period = 10
|
|
}
|
|
if success == 0 {
|
|
probe.Success = 1
|
|
}
|
|
if failure == 0 {
|
|
probe.Failure = 3
|
|
}
|
|
return probe
|
|
}
|
|
|
|
// NewProbeWithDuration creates a new Probe object with the given duration from types.
|
|
func NewProbeWithDuration(period, initialDelaySeconds *types.Duration, success, failure *uint64) *Probe {
|
|
|
|
if period == nil {
|
|
d := types.Duration(0 * time.Second)
|
|
period = &d
|
|
}
|
|
|
|
if initialDelaySeconds == nil {
|
|
d := types.Duration(0 * time.Second)
|
|
initialDelaySeconds = &d
|
|
}
|
|
|
|
if success == nil {
|
|
s := uint64(0)
|
|
success = &s
|
|
}
|
|
|
|
if failure == nil {
|
|
f := uint64(0)
|
|
failure = &f
|
|
}
|
|
|
|
p, err := time.ParseDuration(period.String())
|
|
if err != nil {
|
|
p = time.Second * 10
|
|
}
|
|
|
|
i, err := time.ParseDuration(initialDelaySeconds.String())
|
|
if err != nil {
|
|
i = time.Second * 0
|
|
}
|
|
|
|
return NewProbe(p.Seconds(), i.Seconds(), *success, *failure)
|
|
|
|
}
|
|
|
|
// NewProbeFromService creates a new Probe object from a ServiceConfig.
|
|
func NewProbeFromService(s *types.ServiceConfig) *Probe {
|
|
if s == nil || s.HealthCheck == nil {
|
|
return NewProbe(0, 0, 0, 0)
|
|
}
|
|
|
|
return NewProbeWithDuration(s.HealthCheck.Interval, s.HealthCheck.StartPeriod, nil, s.HealthCheck.Retries)
|
|
|
|
}
|
|
|
|
// HttpGet is a Probe configuration to check http health.
|
|
type HttpGet struct {
|
|
Path string `yaml:"path"`
|
|
Port int `yaml:"port"`
|
|
}
|
|
|
|
// Execis a Probe configuration to check exec health.
|
|
type Exec struct {
|
|
Command []string `yaml:"command"`
|
|
}
|
|
|
|
// TCP is a Probe configuration to check tcp health.
|
|
type TCP struct {
|
|
Port int `yaml:"port"`
|
|
}
|