mirror of
https://github.com/coredns/coredns.git
synced 2025-11-20 19:02:16 -05:00
pkg/up: add generic run-this-functions (#1481)
This adds a generic way of start a check function to check a backend. This package can be used to kick off healthchecks. The package makes sure only 1 is run at any one time. It should allow for: See upstream error -> kick off healthcheck and not to worry about overwhelming the upstream with a barrage of queries.
This commit is contained in:
66
plugin/pkg/up/up.go
Normal file
66
plugin/pkg/up/up.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package up
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Probe is used to run a single Func until it returns true (indicating a target is healthy). If an Func
|
||||
// is already in progress no new one will be added, i.e. there is always a maximum of 1 checks in flight.
|
||||
type Probe struct {
|
||||
do chan Func
|
||||
stop chan bool
|
||||
|
||||
target string
|
||||
|
||||
sync.Mutex
|
||||
inprogress bool
|
||||
}
|
||||
|
||||
// Func is used to determine if a target is alive. If so this function must return true.
|
||||
type Func func(target string) bool
|
||||
|
||||
// New returns a pointer to an intialized Probe.
|
||||
func New() *Probe {
|
||||
return &Probe{stop: make(chan bool), do: make(chan Func)}
|
||||
}
|
||||
|
||||
// Do will probe target, if a probe is already in progress this is a noop.
|
||||
func (p *Probe) Do(f Func) { p.do <- f }
|
||||
|
||||
// Stop stops the probing.
|
||||
func (p *Probe) Stop() { p.stop <- true }
|
||||
|
||||
// Start will start the probe manager, after which probes can be initialized with Do.
|
||||
func (p *Probe) Start(target string, interval time.Duration) { go p.start(target, interval) }
|
||||
|
||||
func (p *Probe) start(target string, interval time.Duration) {
|
||||
for {
|
||||
select {
|
||||
case <-p.stop:
|
||||
return
|
||||
case f := <-p.do:
|
||||
p.Lock()
|
||||
if p.inprogress {
|
||||
p.Unlock()
|
||||
continue
|
||||
}
|
||||
p.inprogress = true
|
||||
p.Unlock()
|
||||
|
||||
// Passed the lock. Now run f for as long it returns false. If a true is returned
|
||||
// we return from the goroutine and we can accept another Func to run.
|
||||
go func() {
|
||||
for {
|
||||
if ok := f(target); ok {
|
||||
break
|
||||
}
|
||||
time.Sleep(interval)
|
||||
}
|
||||
p.Lock()
|
||||
p.inprogress = false
|
||||
p.Unlock()
|
||||
}()
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user