Mirror/k3s
1
0
Fork 0
mirror of https://github.com/k3s-io/k3s.git synced 2024-06-07 19:41:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tweaked order of ingress IPs in ServiceLB (#8711)

Browse Source 
* 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>
This commit is contained in:
Jason Costello 2023-11-15 17:33:31 -05:00 committed by GitHub
parent 7ecd5874d2
commit 07ee854914
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/cloudprovider/servicelb.go
View File

@ -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{
@ -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:

26
pkg/cloudprovider/servicelb_test.go
View File

@ -1,6 +1,7 @@
package cloudprovider
import (
"math/rand"
"reflect"
"testing"
@ -8,8 +9,10 @@ import (
)
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) {
@ -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)
}
}