Add service topology key logic

pkg/controller/service.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"strconv"
 
 	"github.com/containous/maesh/pkg/k8s"
 	"github.com/sirupsen/logrus"
@@ -75,6 +76,13 @@ func (s *ShadowServiceManager) Create(userSvc *corev1.Service) error {
 		},
 	}
 
+	major, minor := parseKubernetesServerVersion(s.kubeClient)
+
+	// If the kuberentes server version is 1.17+, then use the topology key.
+	if major == 1 && minor >= 17 {
+		svc.Spec.TopologyKeys = []string{"kubernetes.io/hostname", "*"}
+	}
+
 	if _, err = s.kubeClient.CoreV1().Services(s.namespace).Create(context.TODO(), svc, metav1.CreateOptions{}); err != nil {
 		return fmt.Errorf("unable to create kubernetes service: %w", err)
 	}
@@ -248,3 +256,22 @@ func (s *ShadowServiceManager) getTCPPort(svcName, svcNamespace string, svcPort
 
 	return port, nil
 }
+
+func parseKubernetesServerVersion(kubeClient kubernetes.Interface) (major, minor int) {
+	kubeVersion, err := kubeClient.Discovery().ServerVersion()
+	if err != nil {
+		return 0, 0
+	}
+
+	major, err = strconv.Atoi(kubeVersion.Major)
+	if err != nil {
+		return 0, 0
+	}
+
+	minor, err = strconv.Atoi(kubeVersion.Minor)
+	if err != nil {
+		return 0, 0
+	}
+
+	return major, minor
+}