Skip to content

Commit

Permalink
Tweaked order of ingress IPs in ServiceLB (#8711)
Browse files Browse the repository at this point in the history
* Tweaked order of ingress IPs in ServiceLB
    Previously, ingress IPs were only string-sorted when returned
    Sorted by IP family and string-sorted in each family as part of
    filterByIPFamily method
* Update pkg/cloudprovider/servicelb.go
* Formatting

Signed-off-by: Jason Costello <jason@hazy.com>
Co-authored-by: Brad Davidson <brad@oatmail.org>
(cherry picked from commit 07ee854)
Signed-off-by: Brad Davidson <brad.davidson@rancher.com>
  • Loading branch information
jsnctl and brandond committed Nov 16, 2023
1 parent 57ed39a commit 91622fc
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 4 deletions.
5 changes: 3 additions & 2 deletions pkg/cloudprovider/servicelb.go
Original file line number Diff line number Diff line change
Expand Up @@ -285,8 +285,6 @@ func (k *k3s) getStatus(svc *core.Service) (*core.LoadBalancerStatus, error) {
return nil, err
}

sort.Strings(expectedIPs)

loadbalancer := &core.LoadBalancerStatus{}
for _, ip := range expectedIPs {
loadbalancer.Ingress = append(loadbalancer.Ingress, core.LoadBalancerIngress{
Expand Down Expand Up @@ -393,6 +391,9 @@ func filterByIPFamily(ips []string, svc *core.Service) ([]string, error) {
}
}

sort.Strings(ipv4Addresses)
sort.Strings(ipv6Addresses)

for _, ipFamily := range svc.Spec.IPFamilies {
switch ipFamily {
case core.IPv4Protocol:
Expand Down
26 changes: 24 additions & 2 deletions pkg/cloudprovider/servicelb_test.go
Original file line number Diff line number Diff line change
@@ -1,15 +1,18 @@
package cloudprovider

import (
"math/rand"
"reflect"
"testing"

core "k8s.io/api/core/v1"
)

const (
addrv4 = "1.2.3.4"
addrv6 = "2001:db8::1"
addrv4 = "1.2.3.4"
addrv4_2 = "2.3.4.5"
addrv6 = "2001:db8::1"
addrv6_2 = "3001:db8::1"
)

func Test_UnitFilterByIPFamily(t *testing.T) {
Expand Down Expand Up @@ -89,3 +92,22 @@ func Test_UnitFilterByIPFamily(t *testing.T) {
})
}
}

func Test_UnitFilterByIPFamily_Ordering(t *testing.T) {
want := []string{addrv4, addrv4_2, addrv6, addrv6_2}
ips := []string{addrv4, addrv4_2, addrv6, addrv6_2}
rand.Shuffle(len(ips), func(i, j int) {
ips[i], ips[j] = ips[j], ips[i]
})
svc := &core.Service{
Spec: core.ServiceSpec{
IPFamilies: []core.IPFamily{core.IPv4Protocol, core.IPv6Protocol},
},
}

got, _ := filterByIPFamily(ips, svc)

if !reflect.DeepEqual(got, want) {
t.Errorf("filterByIPFamily() = %+v\nWant = %+v", got, want)
}
}

0 comments on commit 91622fc

Please sign in to comment.