mirror of
https://github.com/coredns/coredns.git
synced 2025-11-01 18:53:43 -04:00
plugin/forward: add it (#1447)
* plugin/forward: add it This moves coredns/forward into CoreDNS. Fixes as a few bugs, adds a policy option and more tests to the plugin. Update the documentation, test IPv6 address and add persistent tests. * Always use random policy when spraying * include scrub fix here as well * use correct var name * Code review * go vet * Move logging to metrcs * Small readme updates * Fix readme
This commit is contained in:
67
plugin/forward/health.go
Normal file
67
plugin/forward/health.go
Normal file
@@ -0,0 +1,67 @@
|
||||
package forward
|
||||
|
||||
import (
|
||||
"log"
|
||||
"sync/atomic"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
// For HC we send to . IN NS +norec message to the upstream. Dial timeouts and empty
|
||||
// replies are considered fails, basically anything else constitutes a healthy upstream.
|
||||
|
||||
func (h *host) Check() {
|
||||
h.Lock()
|
||||
|
||||
if h.checking {
|
||||
h.Unlock()
|
||||
return
|
||||
}
|
||||
|
||||
h.checking = true
|
||||
h.Unlock()
|
||||
|
||||
err := h.send()
|
||||
if err != nil {
|
||||
log.Printf("[INFO] healtheck of %s failed with %s", h.addr, err)
|
||||
|
||||
HealthcheckFailureCount.WithLabelValues(h.addr).Add(1)
|
||||
|
||||
atomic.AddUint32(&h.fails, 1)
|
||||
} else {
|
||||
atomic.StoreUint32(&h.fails, 0)
|
||||
}
|
||||
|
||||
h.Lock()
|
||||
h.checking = false
|
||||
h.Unlock()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (h *host) send() error {
|
||||
hcping := new(dns.Msg)
|
||||
hcping.SetQuestion(".", dns.TypeNS)
|
||||
hcping.RecursionDesired = false
|
||||
|
||||
m, _, err := h.client.Exchange(hcping, h.addr)
|
||||
// If we got a header, we're alright, basically only care about I/O errors 'n stuff
|
||||
if err != nil && m != nil {
|
||||
// Silly check, something sane came back
|
||||
if m.Response || m.Opcode == dns.OpcodeQuery {
|
||||
err = nil
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// down returns true is this host has more than maxfails fails.
|
||||
func (h *host) down(maxfails uint32) bool {
|
||||
if maxfails == 0 {
|
||||
return false
|
||||
}
|
||||
|
||||
fails := atomic.LoadUint32(&h.fails)
|
||||
return fails > maxfails
|
||||
}
|
||||
Reference in New Issue
Block a user