Mirrors/coredns
1
0
Fork 0
mirror of https://github.com/coredns/coredns.git synced 2025-11-01 10:43:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add weighted-round-robin policy to loadbalance plugin (#5662)

Browse Source 
* Add weighted-round-robin policy to loadbalance plugin

Signed-off-by: Gabor Dozsa <gabor.dozsa@ibm.com>
This commit is contained in:
Gabor Dozsa
2023-01-27 17:36:56 +01:00
committed by GitHub
parent bf7c2cf37b
commit 7da2cedaf0
8 changed files with 975 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
plugin/loadbalance/handler.go
View File

@@ -10,15 +10,16 @@ import (
)
// RoundRobin is a plugin to rewrite responses for "load balancing".
type RoundRobin struct {
Next plugin.Handler
type LoadBalance struct {
Next plugin.Handler
shuffle func(*dns.Msg) *dns.Msg
}
// ServeDNS implements the plugin.Handler interface.
func (rr RoundRobin) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
wrr := &RoundRobinResponseWriter{w}
return plugin.NextOrFailure(rr.Name(), rr.Next, ctx, wrr, r)
func (lb LoadBalance) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
rw := &LoadBalanceResponseWriter{ResponseWriter: w, shuffle: lb.shuffle}
return plugin.NextOrFailure(lb.Name(), lb.Next, ctx, rw, r)
}
// Name implements the Handler interface.
func (rr RoundRobin) Name() string { return "loadbalance" }
func (lb LoadBalance) Name() string { return "loadbalance" }