mirror of
https://github.com/coredns/coredns.git
synced 2025-10-31 10:13:14 -04:00
20 lines
569 B
Go
20 lines
569 B
Go
|
// Package loadbalance is middleware for rewriting responses to do "load balancing"
|
||
|
|
package loadbalance
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/miekg/coredns/middleware"
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RoundRobin is middleware to rewrite responses for "load balancing".
|
||
|
|
type RoundRobin struct {
|
||
|
|
Next middleware.Handler
|
||
|
|
}
|
||
|
|
|
||
|
|
// ServeHTTP 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)
|
||
|
|
}
|