mirror of
https://github.com/coredns/coredns.git
synced 2025-10-30 17:53:21 -04:00
5b844b5017
* 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
68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
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
|
|
}
|