diff options
Diffstat (limited to 'middleware/loadbalance/handler.go')
| -rw-r--r-- | middleware/loadbalance/handler.go | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/middleware/loadbalance/handler.go b/middleware/loadbalance/handler.go index bb8619543..119b4d5d2 100644 --- a/middleware/loadbalance/handler.go +++ b/middleware/loadbalance/handler.go @@ -3,6 +3,7 @@ package loadbalance import ( "github.com/miekg/coredns/middleware" + "github.com/miekg/dns" "golang.org/x/net/context" ) @@ -12,7 +13,7 @@ type RoundRobin struct { Next middleware.Handler } -// ServeHTTP implements the middleware.Handler interface. +// ServeDNS implements the middleware.Handler interface. func (rr RoundRobin) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { wrr := NewRoundRobinResponseWriter(w) return rr.Next.ServeDNS(ctx, wrr, r) |