refactor(probes): rename ProbeFunction
to Probe
This commit is contained in:
parent
24cb5d816a
commit
8dc7f27400
26
README.md
26
README.md
@ -21,22 +21,22 @@ Default paths can be overriden with options described below. Accessing any other
|
|||||||
|
|
||||||
The `kubeprobes.New` function accepts following options as arguments:
|
The `kubeprobes.New` function accepts following options as arguments:
|
||||||
|
|
||||||
- `kubeprobes.WithLivenessProbes(...)` - adds particular [probe functions](#probe-functions) to the list of liveness probes;
|
- `kubeprobes.WithLivenessProbes(...)` - adds particular [probes](#probes) to the list of liveness probes;
|
||||||
- `kubeprobes.WithLivenessPath("/some/liveness/path")` - sets liveness probe path to given path (default is `/live`);
|
- `kubeprobes.WithLivenessPath("/some/liveness/path")` - sets liveness probe path to given path (default is `/live`);
|
||||||
- `kubeprobes.WithReadinessProbes(...)` - adds particular [probe functions](#probe-functions) to the list of readiness probes;
|
- `kubeprobes.WithReadinessProbes(...)` - adds particular [probes](#probes) to the list of readiness probes;
|
||||||
- `kubeprobes.WithReadinessPath("/some/readiness/path")` - sets readiness probe path to given path (default is `/ready`);
|
- `kubeprobes.WithReadinessPath("/some/readiness/path")` - sets readiness probe path to given path (default is `/ready`);
|
||||||
- `kubeprobes.WithVerboseOutput()` - enables verbose output by default (returns both failed and passed probes).
|
- `kubeprobes.WithVerboseOutput()` - enables verbose output by default (returns both failed and passed probes).
|
||||||
|
|
||||||
## Probes
|
## Probes
|
||||||
|
|
||||||
In order to determine the state of particular element of application, probes need to be implemented either by creating [status determining function](#probe-functions) or by using simple and thread-safe [manual probes](#manual-probes).
|
In order to determine the state of particular element of application, probes need to be implemented either by creating [probes from functions](#standard-probes) or by using simple and thread-safe [manual probes](#manual-probes).
|
||||||
|
|
||||||
### Probe functions
|
### Standard probes
|
||||||
|
|
||||||
Probe functions (instances of `ProbeFunction` interface) are wrappers for functions that performs user defined logic with given interval of updates in order to determine whether the probe should be marked as healthy or not. Those functions should take no arguments and return error (if no error is returned, the probe is considered to be healthy; if error is returned, the probe is considered to be unhealthy). If given interval is less or equal zero, then function is only checked on probe creation and remains in determined state forever.
|
Probes (instances of `Probe` interface) are wrappers for functions that performs user defined logic with given interval of updates in order to determine whether the probe should be marked as healthy or not. Those functions should take no arguments and return error (if no error is returned, the probe is considered to be healthy; if error is returned, the probe is considered to be unhealthy). If given interval is less or equal zero, then function is only checked on probe creation and remains in determined state forever.
|
||||||
|
|
||||||
```go
|
```go
|
||||||
someProbe := kubeprobes.NewProbeFunction("live", func() error {
|
someProbe := kubeprobes.NewProbe("live", func() error {
|
||||||
// Some logic here
|
// Some logic here
|
||||||
if time.Now().Weekday() == time.Wednesday {
|
if time.Now().Weekday() == time.Wednesday {
|
||||||
// Fail only on wednesday!
|
// Fail only on wednesday!
|
||||||
@ -45,7 +45,7 @@ someProbe := kubeprobes.NewProbeFunction("live", func() error {
|
|||||||
return nil
|
return nil
|
||||||
}, 1 * time.Hour)
|
}, 1 * time.Hour)
|
||||||
|
|
||||||
someOtherProbe := kubeprobes.NewProbeFunction("ready", func() error {
|
someOtherProbe := kubeprobes.NewProbe("ready", func() error {
|
||||||
// Always healthy
|
// Always healthy
|
||||||
return nil
|
return nil
|
||||||
}, 0) // This probe is checked once
|
}, 0) // This probe is checked once
|
||||||
@ -59,7 +59,7 @@ kp, _ := kubeprobes.New(
|
|||||||
|
|
||||||
### Manual probes
|
### Manual probes
|
||||||
|
|
||||||
Manual probes (instances of `ManualProbe` interface) are objects that can be marked either as healthy or unhealthy and implement `ProbeFunction` for easy integration. Those objects utilize `sync.RMutex` mechanism to ensure thread-safety.
|
Manual probes (instances of `ManualProbe` interface) are objects that can be marked either as healthy or unhealthy and implement `Probe` for easy integration. Those objects utilize `sync.RMutex` mechanism to ensure thread-safety.
|
||||||
|
|
||||||
Those probes can be changed by user with provided methods:
|
Those probes can be changed by user with provided methods:
|
||||||
|
|
||||||
@ -78,7 +78,9 @@ kp, _ := kubeprobes.New(
|
|||||||
kubeprobes.WithReadinessProbes(someOtherProbe),
|
kubeprobes.WithReadinessProbes(someOtherProbe),
|
||||||
)
|
)
|
||||||
|
|
||||||
// Can be later marked according
|
// Can be later marked according to needs
|
||||||
|
someProbe.Pass()
|
||||||
|
someOtherProbe.FailWithCause(errors.New("I'm not doing anything!"))
|
||||||
```
|
```
|
||||||
|
|
||||||
## Direct handler access
|
## Direct handler access
|
||||||
@ -105,9 +107,9 @@ appProbe := func() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create stateful probes
|
// Create manual probes
|
||||||
live := kubeprobes.NewStatefulProbe()
|
live := kubeprobes.NewManualProbe()
|
||||||
ready := kubeprobes.NewStatefulProbe()
|
ready := kubeprobes.NewManualProbe()
|
||||||
|
|
||||||
// Prepare handler
|
// Prepare handler
|
||||||
kp, err := kubeprobes.New(
|
kp, err := kubeprobes.New(
|
||||||
|
@ -19,8 +19,8 @@ type Kubeprobes interface {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type kubeprobes struct {
|
type kubeprobes struct {
|
||||||
livenessProbes []ProbeFunction
|
livenessProbes []Probe
|
||||||
readinessProbes []ProbeFunction
|
readinessProbes []Probe
|
||||||
|
|
||||||
verbose bool
|
verbose bool
|
||||||
|
|
||||||
@ -31,8 +31,8 @@ type kubeprobes struct {
|
|||||||
// New returns a new instance of a Kubernetes probes with given options.
|
// New returns a new instance of a Kubernetes probes with given options.
|
||||||
func New(options ...Option) (Kubeprobes, error) {
|
func New(options ...Option) (Kubeprobes, error) {
|
||||||
kp := &kubeprobes{
|
kp := &kubeprobes{
|
||||||
livenessProbes: []ProbeFunction{},
|
livenessProbes: []Probe{},
|
||||||
readinessProbes: []ProbeFunction{},
|
readinessProbes: []Probe{},
|
||||||
pathLive: defaultLivenessPath,
|
pathLive: defaultLivenessPath,
|
||||||
pathReady: defaultReadinessPath,
|
pathReady: defaultReadinessPath,
|
||||||
}
|
}
|
||||||
|
@ -12,7 +12,7 @@ func (o option) apply(kp *kubeprobes) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// WithLivenessProbes adds given probe functions to the set of liveness probes.
|
// WithLivenessProbes adds given probe functions to the set of liveness probes.
|
||||||
func WithLivenessProbes(probes ...ProbeFunction) Option {
|
func WithLivenessProbes(probes ...Probe) Option {
|
||||||
return option(func(kp *kubeprobes) {
|
return option(func(kp *kubeprobes) {
|
||||||
kp.livenessProbes = append(kp.livenessProbes, probes...)
|
kp.livenessProbes = append(kp.livenessProbes, probes...)
|
||||||
})
|
})
|
||||||
@ -26,7 +26,7 @@ func WithLivenessPath(path string) Option {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// WithReadinessProbes adds given probe functions to the set of readiness probes.
|
// WithReadinessProbes adds given probe functions to the set of readiness probes.
|
||||||
func WithReadinessProbes(probes ...ProbeFunction) Option {
|
func WithReadinessProbes(probes ...Probe) Option {
|
||||||
return option(func(kp *kubeprobes) {
|
return option(func(kp *kubeprobes) {
|
||||||
kp.readinessProbes = append(kp.readinessProbes, probes...)
|
kp.readinessProbes = append(kp.readinessProbes, probes...)
|
||||||
})
|
})
|
||||||
|
@ -5,15 +5,15 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
// ProbeFunction is a wrapper for a function that determines whether
|
// Probe is a wrapper for a function that determines whether
|
||||||
// the given metric may be marked as correctly functioning.
|
// the given metric may be marked as correctly functioning.
|
||||||
// It not, the error should be returned.
|
// It not, the error should be returned.
|
||||||
type ProbeFunction interface {
|
type Probe interface {
|
||||||
name() string
|
name() string
|
||||||
status() error
|
status() error
|
||||||
}
|
}
|
||||||
|
|
||||||
type probeFunction struct {
|
type probe struct {
|
||||||
probeName string
|
probeName string
|
||||||
probeFunc func() error
|
probeFunc func() error
|
||||||
refreshInterval time.Duration
|
refreshInterval time.Duration
|
||||||
@ -22,20 +22,20 @@ type probeFunction struct {
|
|||||||
err error
|
err error
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewProbeFunction returns new instance of [ProbeFunction].
|
// NewProbe returns new instance of [Probe].
|
||||||
//
|
//
|
||||||
// If update interval is less or equal zero then probe is updated only
|
// If update interval is less or equal zero then probe is updated only
|
||||||
// on its creation and remains in the same state forever.
|
// on its creation and remains in the same state forever.
|
||||||
func NewProbeFunction(
|
func NewProbe(
|
||||||
name string,
|
name string,
|
||||||
fn func() error,
|
fn func() error,
|
||||||
updateInterval time.Duration,
|
updateInterval time.Duration,
|
||||||
) (ProbeFunction, error) {
|
) (Probe, error) {
|
||||||
if name == "" {
|
if name == "" {
|
||||||
return nil, errProbeNameEmpty
|
return nil, errProbeNameEmpty
|
||||||
}
|
}
|
||||||
|
|
||||||
pf := &probeFunction{
|
pf := &probe{
|
||||||
probeName: name,
|
probeName: name,
|
||||||
probeFunc: fn,
|
probeFunc: fn,
|
||||||
refreshInterval: updateInterval,
|
refreshInterval: updateInterval,
|
||||||
@ -47,25 +47,25 @@ func NewProbeFunction(
|
|||||||
}
|
}
|
||||||
|
|
||||||
// name implements ProbeFunction.
|
// name implements ProbeFunction.
|
||||||
func (pf *probeFunction) name() string {
|
func (pf *probe) name() string {
|
||||||
return pf.probeName
|
return pf.probeName
|
||||||
}
|
}
|
||||||
|
|
||||||
// status implements ProbeFunction.
|
// status implements ProbeFunction.
|
||||||
func (pf *probeFunction) status() error {
|
func (pf *probe) status() error {
|
||||||
pf.mux.RLock()
|
pf.mux.RLock()
|
||||||
defer pf.mux.RUnlock()
|
defer pf.mux.RUnlock()
|
||||||
return pf.err
|
return pf.err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pf *probeFunction) update() {
|
func (pf *probe) update() {
|
||||||
err := pf.probeFunc()
|
err := pf.probeFunc()
|
||||||
pf.mux.Lock()
|
pf.mux.Lock()
|
||||||
pf.err = err
|
pf.err = err
|
||||||
pf.mux.Unlock()
|
pf.mux.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pf *probeFunction) autoUpdate() {
|
func (pf *probe) autoUpdate() {
|
||||||
pf.update()
|
pf.update()
|
||||||
if pf.refreshInterval <= 0 {
|
if pf.refreshInterval <= 0 {
|
||||||
return
|
return
|
@ -7,7 +7,7 @@ import (
|
|||||||
// ManualProbe represents the simple probe that can be either
|
// ManualProbe represents the simple probe that can be either
|
||||||
// marked as "up" (healthy) or "down" (unhealthy).
|
// marked as "up" (healthy) or "down" (unhealthy).
|
||||||
type ManualProbe interface {
|
type ManualProbe interface {
|
||||||
ProbeFunction
|
Probe
|
||||||
|
|
||||||
// Pass marks the probe as healthy.
|
// Pass marks the probe as healthy.
|
||||||
Pass()
|
Pass()
|
||||||
|
2
query.go
2
query.go
@ -20,7 +20,7 @@ func (sq *statusQuery) wait() {
|
|||||||
sq.wg.Wait()
|
sq.wg.Wait()
|
||||||
}
|
}
|
||||||
|
|
||||||
func newStatusQuery(probes []ProbeFunction) *statusQuery {
|
func newStatusQuery(probes []Probe) *statusQuery {
|
||||||
sq := &statusQuery{
|
sq := &statusQuery{
|
||||||
ok: true,
|
ok: true,
|
||||||
passed: make([]statusEntry, 0, len(probes)),
|
passed: make([]statusEntry, 0, len(probes)),
|
||||||
|
@ -6,28 +6,28 @@ import (
|
|||||||
|
|
||||||
func TestStatusQueryIsAllGreen(t *testing.T) {
|
func TestStatusQueryIsAllGreen(t *testing.T) {
|
||||||
var (
|
var (
|
||||||
probePassing, _ = NewProbeFunction("pass", func() error {
|
probePassing, _ = NewProbe("pass", func() error {
|
||||||
return nil
|
return nil
|
||||||
}, 0)
|
}, 0)
|
||||||
probeFailing, _ = NewProbeFunction("fail", func() error {
|
probeFailing, _ = NewProbe("fail", func() error {
|
||||||
return errProbeFailed
|
return errProbeFailed
|
||||||
}, 0)
|
}, 0)
|
||||||
)
|
)
|
||||||
|
|
||||||
tests := map[string]struct {
|
tests := map[string]struct {
|
||||||
probes []ProbeFunction
|
probes []Probe
|
||||||
expectedStatus bool
|
expectedStatus bool
|
||||||
}{
|
}{
|
||||||
"all green": {
|
"all green": {
|
||||||
probes: []ProbeFunction{probePassing},
|
probes: []Probe{probePassing},
|
||||||
expectedStatus: true,
|
expectedStatus: true,
|
||||||
},
|
},
|
||||||
"some failed": {
|
"some failed": {
|
||||||
probes: []ProbeFunction{probePassing, probeFailing},
|
probes: []Probe{probePassing, probeFailing},
|
||||||
expectedStatus: false,
|
expectedStatus: false,
|
||||||
},
|
},
|
||||||
"all failed": {
|
"all failed": {
|
||||||
probes: []ProbeFunction{probeFailing},
|
probes: []Probe{probeFailing},
|
||||||
expectedStatus: false,
|
expectedStatus: false,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user