From a46e1f0613867f5e8c5a7c41a2e46035f399f505 Mon Sep 17 00:00:00 2001 From: Antonio Ojea Date: Tue, 30 Jun 2020 17:35:38 +0200 Subject: [PATCH] kube-proxy ShouldSkipService takes only one argument instead of receiving the service name and namespace we can obtain it from the service object directly. Signed-off-by: Antonio Ojea --- pkg/proxy/service.go | 7 ++++--- pkg/proxy/userspace/proxier.go | 11 +++++------ pkg/proxy/util/BUILD | 1 - pkg/proxy/util/utils.go | 6 +++--- pkg/proxy/util/utils_test.go | 10 +--------- 5 files changed, 13 insertions(+), 22 deletions(-) diff --git a/pkg/proxy/service.go b/pkg/proxy/service.go index 038eab23f59..9f66272463c 100644 --- a/pkg/proxy/service.go +++ b/pkg/proxy/service.go @@ -25,7 +25,7 @@ import ( "k8s.io/klog/v2" - "k8s.io/api/core/v1" + v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/types" "k8s.io/apimachinery/pkg/util/sets" "k8s.io/client-go/tools/record" @@ -307,8 +307,8 @@ func (sct *ServiceChangeTracker) serviceToServiceMap(service *v1.Service) Servic if service == nil { return nil } - svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} - if utilproxy.ShouldSkipService(svcName, service) { + + if utilproxy.ShouldSkipService(service) { return nil } @@ -322,6 +322,7 @@ func (sct *ServiceChangeTracker) serviceToServiceMap(service *v1.Service) Servic } serviceMap := make(ServiceMap) + svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} for i := range service.Spec.Ports { servicePort := &service.Spec.Ports[i] svcPortName := ServicePortName{NamespacedName: svcName, Port: servicePort.Name, Protocol: servicePort.Protocol} diff --git a/pkg/proxy/userspace/proxier.go b/pkg/proxy/userspace/proxier.go index 6b4f5ce451d..71c04391955 100644 --- a/pkg/proxy/userspace/proxier.go +++ b/pkg/proxy/userspace/proxier.go @@ -489,12 +489,11 @@ func (proxier *Proxier) mergeService(service *v1.Service) sets.String { if service == nil { return nil } - svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} - if utilproxy.ShouldSkipService(svcName, service) { - klog.V(3).Infof("Skipping service %s due to clusterIP = %q", svcName, service.Spec.ClusterIP) + if utilproxy.ShouldSkipService(service) { return nil } existingPorts := sets.NewString() + svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} for i := range service.Spec.Ports { servicePort := &service.Spec.Ports[i] serviceName := proxy.ServicePortName{NamespacedName: svcName, Port: servicePort.Name} @@ -551,12 +550,12 @@ func (proxier *Proxier) unmergeService(service *v1.Service, existingPorts sets.S if service == nil { return } - svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} - if utilproxy.ShouldSkipService(svcName, service) { - klog.V(3).Infof("Skipping service %s due to clusterIP = %q", svcName, service.Spec.ClusterIP) + + if utilproxy.ShouldSkipService(service) { return } staleUDPServices := sets.NewString() + svcName := types.NamespacedName{Namespace: service.Namespace, Name: service.Name} for i := range service.Spec.Ports { servicePort := &service.Spec.Ports[i] if existingPorts.Has(servicePort.Name) { diff --git a/pkg/proxy/util/BUILD b/pkg/proxy/util/BUILD index 3b5e3411b27..4f1ce9864b1 100644 --- a/pkg/proxy/util/BUILD +++ b/pkg/proxy/util/BUILD @@ -35,7 +35,6 @@ go_test( "//pkg/proxy/util/testing:go_default_library", "//staging/src/k8s.io/api/core/v1:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library", - "//staging/src/k8s.io/apimachinery/pkg/types:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/util/sets:go_default_library", ], ) diff --git a/pkg/proxy/util/utils.go b/pkg/proxy/util/utils.go index d7f94345e2d..c2bf2c97d15 100644 --- a/pkg/proxy/util/utils.go +++ b/pkg/proxy/util/utils.go @@ -146,15 +146,15 @@ func GetLocalAddrs() ([]net.IP, error) { } // ShouldSkipService checks if a given service should skip proxying -func ShouldSkipService(svcName types.NamespacedName, service *v1.Service) bool { +func ShouldSkipService(service *v1.Service) bool { // if ClusterIP is "None" or empty, skip proxying if !helper.IsServiceIPSet(service) { - klog.V(3).Infof("Skipping service %s due to clusterIP = %q", svcName, service.Spec.ClusterIP) + klog.V(3).Infof("Skipping service %s in namespace %s due to clusterIP = %q", service.Name, service.Namespace, service.Spec.ClusterIP) return true } // Even if ClusterIP is set, ServiceTypeExternalName services don't get proxied if service.Spec.Type == v1.ServiceTypeExternalName { - klog.V(3).Infof("Skipping service %s due to Type=ExternalName", svcName) + klog.V(3).Infof("Skipping service %s in namespace %s due to Type=ExternalName", service.Name, service.Namespace) return true } return false diff --git a/pkg/proxy/util/utils_test.go b/pkg/proxy/util/utils_test.go index 50f5239d21c..e563a8da888 100644 --- a/pkg/proxy/util/utils_test.go +++ b/pkg/proxy/util/utils_test.go @@ -25,7 +25,6 @@ import ( v1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" - "k8s.io/apimachinery/pkg/types" "k8s.io/apimachinery/pkg/util/sets" fake "k8s.io/kubernetes/pkg/proxy/util/testing" ) @@ -167,7 +166,6 @@ func TestIsProxyableHostname(t *testing.T) { func TestShouldSkipService(t *testing.T) { testCases := []struct { service *v1.Service - svcName types.NamespacedName shouldSkip bool }{ { @@ -178,7 +176,6 @@ func TestShouldSkipService(t *testing.T) { ClusterIP: v1.ClusterIPNone, }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: true, }, { @@ -189,7 +186,6 @@ func TestShouldSkipService(t *testing.T) { ClusterIP: "", }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: true, }, { @@ -201,7 +197,6 @@ func TestShouldSkipService(t *testing.T) { Type: v1.ServiceTypeExternalName, }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: true, }, { @@ -213,7 +208,6 @@ func TestShouldSkipService(t *testing.T) { Type: v1.ServiceTypeClusterIP, }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: false, }, { @@ -225,7 +219,6 @@ func TestShouldSkipService(t *testing.T) { Type: v1.ServiceTypeNodePort, }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: false, }, { @@ -237,13 +230,12 @@ func TestShouldSkipService(t *testing.T) { Type: v1.ServiceTypeLoadBalancer, }, }, - svcName: types.NamespacedName{Namespace: "foo", Name: "bar"}, shouldSkip: false, }, } for i := range testCases { - skip := ShouldSkipService(testCases[i].svcName, testCases[i].service) + skip := ShouldSkipService(testCases[i].service) if skip != testCases[i].shouldSkip { t.Errorf("case %d: expect %v, got %v", i, testCases[i].shouldSkip, skip) }