45 lines
824 B
Go
45 lines
824 B
Go
|
package kubeprobes
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
// ProbeFunction is a function that determines whether
|
||
|
|
// the given metric may be marked as correctly functioning.
|
||
|
|
// It not, the error should be returned.
|
||
|
|
type ProbeFunction func() error
|
||
|
|
|
||
|
|
type statusQuery struct {
|
||
|
|
allGreen bool
|
||
|
|
mux sync.Mutex
|
||
|
|
wg sync.WaitGroup
|
||
|
|
}
|
||
|
|
|
||
|
|
func (sq *statusQuery) isAllGreen() bool {
|
||
|
|
sq.wg.Wait()
|
||
|
|
sq.mux.Lock()
|
||
|
|
defer sq.mux.Unlock()
|
||
|
|
return sq.allGreen
|
||
|
|
}
|
||
|
|
|
||
|
|
func newStatusQuery(probes []ProbeFunction) *statusQuery {
|
||
|
|
sq := &statusQuery{
|
||
|
|
allGreen: true,
|
||
|
|
mux: sync.Mutex{},
|
||
|
|
wg: sync.WaitGroup{},
|
||
|
|
}
|
||
|
|
|
||
|
|
sq.wg.Add(len(probes))
|
||
|
|
for _, probe := range probes {
|
||
|
|
probe := probe
|
||
|
|
go func() {
|
||
|
|
defer sq.wg.Done()
|
||
|
|
if err := probe(); err != nil {
|
||
|
|
sq.mux.Lock()
|
||
|
|
sq.allGreen = false
|
||
|
|
sq.mux.Unlock()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
return sq
|
||
|
|
}
|