12b2ff9740
* update docs * plugins: use plugin specific logging Hooking up pkg/log also changed NewWithPlugin to just take a string instead of a plugin.Handler as that is more flexible and for instance the Root "plugin" doesn't implement it fully. Same logging from the reload plugin: .:1043 2018/04/22 08:56:37 [INFO] CoreDNS-1.1.1 2018/04/22 08:56:37 [INFO] linux/amd64, go1.10.1, CoreDNS-1.1.1 linux/amd64, go1.10.1, 2018/04/22 08:56:37 [INFO] plugin/reload: Running configuration MD5 = ec4c9c55cd19759ea1c46b8c45742b06 2018/04/22 08:56:54 [INFO] Reloading 2018/04/22 08:56:54 [INFO] plugin/reload: Running configuration MD5 = 9e2bfdd85bdc9cceb740ba9c80f34c1a 2018/04/22 08:56:54 [INFO] Reloading complete * update docs * better doc
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
// Package loadbalance shuffles A and AAAA records.
|
|
package loadbalance
|
|
|
|
import (
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// RoundRobinResponseWriter is a response writer that shuffles A and AAAA records.
|
|
type RoundRobinResponseWriter struct {
|
|
dns.ResponseWriter
|
|
}
|
|
|
|
// WriteMsg implements the dns.ResponseWriter interface.
|
|
func (r *RoundRobinResponseWriter) WriteMsg(res *dns.Msg) error {
|
|
if res.Rcode != dns.RcodeSuccess {
|
|
return r.ResponseWriter.WriteMsg(res)
|
|
}
|
|
|
|
res.Answer = roundRobin(res.Answer)
|
|
res.Ns = roundRobin(res.Ns)
|
|
res.Extra = roundRobin(res.Extra)
|
|
|
|
return r.ResponseWriter.WriteMsg(res)
|
|
}
|
|
|
|
func roundRobin(in []dns.RR) []dns.RR {
|
|
cname := []dns.RR{}
|
|
address := []dns.RR{}
|
|
mx := []dns.RR{}
|
|
rest := []dns.RR{}
|
|
for _, r := range in {
|
|
switch r.Header().Rrtype {
|
|
case dns.TypeCNAME:
|
|
cname = append(cname, r)
|
|
case dns.TypeA, dns.TypeAAAA:
|
|
address = append(address, r)
|
|
case dns.TypeMX:
|
|
mx = append(mx, r)
|
|
default:
|
|
rest = append(rest, r)
|
|
}
|
|
}
|
|
|
|
roundRobinShuffle(address)
|
|
roundRobinShuffle(mx)
|
|
|
|
out := append(cname, rest...)
|
|
out = append(out, address...)
|
|
out = append(out, mx...)
|
|
return out
|
|
}
|
|
|
|
func roundRobinShuffle(records []dns.RR) {
|
|
switch l := len(records); l {
|
|
case 0, 1:
|
|
break
|
|
case 2:
|
|
if dns.Id()%2 == 0 {
|
|
records[0], records[1] = records[1], records[0]
|
|
}
|
|
default:
|
|
for j := 0; j < l*(int(dns.Id())%4+1); j++ {
|
|
q := int(dns.Id()) % l
|
|
p := int(dns.Id()) % l
|
|
if q == p {
|
|
p = (p + 1) % l
|
|
}
|
|
records[q], records[p] = records[p], records[q]
|
|
}
|
|
}
|
|
}
|
|
|
|
// Write implements the dns.ResponseWriter interface.
|
|
func (r *RoundRobinResponseWriter) Write(buf []byte) (int, error) {
|
|
// Should we pack and unpack here to fiddle with the packet... Not likely.
|
|
log.Warning("RoundRobin called with Write: not shuffling records")
|
|
n, err := r.ResponseWriter.Write(buf)
|
|
return n, err
|
|
}
|
|
|
|
// Hijack implements the dns.ResponseWriter interface.
|
|
func (r *RoundRobinResponseWriter) Hijack() {
|
|
r.ResponseWriter.Hijack()
|
|
return
|
|
}
|