Mirrors/coredns
1
0
Fork 0
mirror of https://github.com/coredns/coredns.git synced 2025-11-09 05:26:21 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use weights from xDS

Browse Source 
use the weights as reported. Set prio to 0 to signal it's not used.

Signed-off-by: Miek Gieben <miek@miek.nl>
This commit is contained in:
Miek Gieben
2020-07-24 10:33:52 +02:00
parent 6afcfb8390
commit ccc6e0cff9
4 changed files with 21 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
plugin/traffic/xds/assignment.go
View File

@@ -126,21 +126,27 @@ func (a *assignment) Select(cluster string, healthy bool) (*SocketAddress, bool)
return nil, true
}
// All returns all healthy endpoints.
func (a *assignment) All(cluster string, healthy bool) ([]*SocketAddress, bool) {
// All returns all healthy endpoints, together with their weights.
func (a *assignment) All(cluster string, healthy bool) ([]*SocketAddress, []uint32, bool) {
cla := a.ClusterLoadAssignment(cluster)
if cla == nil {
return nil, false
return nil, nil, false
}
sa := []*SocketAddress{}
we := []uint32{}
for _, ep := range cla.Endpoints {
for _, lb := range ep.GetLbEndpoints() {
if healthy && lb.GetHealthStatus() != corepb2.HealthStatus_HEALTHY {
continue
}
weight := lb.GetLoadBalancingWeight().GetValue()
if weight > 2^16 {
log.Warning("Weight in cluster %q > %d, truncating to %d in SRV responses", cluster, weight, uint16(weight))
}
we = append(we, weight)
sa = append(sa, &SocketAddress{lb.GetEndpoint().GetAddress().GetSocketAddress(), lb.GetHealthStatus()})
}
}
return sa, true
return sa, we, true
}