fixup! refactor: run a script to configure kube-vip

nutanix-cloud-native · Jun 7, 2024 · 0d5c982 · 0d5c982
1 parent 7aa1443
commit 0d5c982
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 24 deletions.
diff --git a/pkg/handlers/generic/mutation/controlplanevirtualip/inject.go b/pkg/handlers/generic/mutation/controlplanevirtualip/inject.go
@@ -134,12 +134,11 @@ func (h *ControlPlaneVirtualIP) Mutate(
 		selectors.ControlPlane(),
 		log,
 		func(obj *controlplanev1.KubeadmControlPlaneTemplate) error {
-			virtualIPProviderFiles, preKubeadmCommands, postKubeadmCommands, generateErr :=
-				virtualIPProvider.GenerateFilesAndCommands(
-					ctx,
-					controlPlaneEndpointVar,
-					cluster,
-				)
+			files, preKubeadmCommands, postKubeadmCommands, generateErr := virtualIPProvider.GenerateFilesAndCommands(
+				ctx,
+				controlPlaneEndpointVar,
+				cluster,
+			)
 			if generateErr != nil {
 				return generateErr
 			}
@@ -153,7 +152,7 @@ func (h *ControlPlaneVirtualIP) Mutate(
 			))
 			obj.Spec.Template.Spec.KubeadmConfigSpec.Files = append(
 				obj.Spec.Template.Spec.KubeadmConfigSpec.Files,
-				virtualIPProviderFiles...,
+				files...,
 			)
 
 			if len(preKubeadmCommands) > 0 {

diff --git a/pkg/handlers/generic/mutation/controlplanevirtualip/providers/kubevip.go b/pkg/handlers/generic/mutation/controlplanevirtualip/providers/kubevip.go
@@ -30,16 +30,16 @@ var (
 	configureForKubeVIPScriptOnRemote = common.ConfigFilePathOnRemote(
 		"configure-for-kube-vip.sh")
 
-	configureForKubeVIPScriptOnRemotePreKubeadmCommand  = "/bin/bash " + configureForKubeVIPScriptOnRemote + " use-super-admin.conf"
-	configureForKubeVIPScriptOnRemotePostKubeadmCommand = "/bin/bash " + configureForKubeVIPScriptOnRemote + " use-admin.conf"
+	configureForKubeVIPScriptOnRemotePreKubeadmCommand = "/bin/bash " +
+		configureForKubeVIPScriptOnRemote + " use-super-admin.conf"
+	configureForKubeVIPScriptOnRemotePostKubeadmCommand = "/bin/bash " +
+		configureForKubeVIPScriptOnRemote + " use-admin.conf"
 
 	setHostAliasesScriptOnRemoteCommand = "/bin/bash " + configureForKubeVIPScriptOnRemote + " set-host-aliases"
 )
 
-var (
-	//go:embed templates/configure-for-kube-vip.sh
-	configureForKubeVIPScript []byte
-)
+//go:embed templates/configure-for-kube-vip.sh
+var configureForKubeVIPScript []byte
 
 type kubeVIPFromConfigMapProvider struct {
 	client client.Reader
@@ -72,7 +72,7 @@ func (p *kubeVIPFromConfigMapProvider) GenerateFilesAndCommands(
 	ctx context.Context,
 	spec v1alpha1.ControlPlaneEndpointSpec,
 	cluster *clusterv1.Cluster,
-) ([]bootstrapv1.File, []string, []string, error) {
+) (files []bootstrapv1.File, preKubeadmCommands, postKubeadmCommands []string, err error) {
 	data, err := getTemplateFromConfigMap(ctx, p.client, p.configMapKey)
 	if err != nil {
 		return nil, nil, nil, fmt.Errorf("failed getting template data: %w", err)
@@ -83,7 +83,7 @@ func (p *kubeVIPFromConfigMapProvider) GenerateFilesAndCommands(
 		return nil, nil, nil, fmt.Errorf("failed templating static Pod: %w", err)
 	}
 
-	files := []bootstrapv1.File{
+	files = []bootstrapv1.File{
 		{
 			Content:     kubeVIPStaticPod,
 			Owner:       kubeVIPFileOwner,
@@ -114,7 +114,10 @@ func (p *kubeVIPFromConfigMapProvider) GenerateFilesAndCommands(
 	// See https://github.com/kubernetes/kubernetes/issues/122420#issuecomment-1864609518
 	needCommands, err := needHackCommands(cluster)
 	if err != nil {
-		return nil, nil, nil, fmt.Errorf("failed to determine if kube-vip commands are needed: %w", err)
+		return nil, nil, nil, fmt.Errorf(
+			"failed to determine if kube-vip commands are needed: %w",
+			err,
+		)
 	}
 	if !needCommands {
 		return files, nil, nil, nil
@@ -129,11 +132,11 @@ func (p *kubeVIPFromConfigMapProvider) GenerateFilesAndCommands(
 		},
 	)
 
-	preKubeadmCommands := []string{
+	preKubeadmCommands = []string{
 		setHostAliasesScriptOnRemoteCommand,
 		configureForKubeVIPScriptOnRemotePreKubeadmCommand,
 	}
-	postKubeadmCommands := []string{configureForKubeVIPScriptOnRemotePostKubeadmCommand}
+	postKubeadmCommands = []string{configureForKubeVIPScriptOnRemotePostKubeadmCommand}
 
 	return files, preKubeadmCommands, postKubeadmCommands, nil
 }

diff --git a/pkg/handlers/generic/mutation/controlplanevirtualip/providers/kubevip_test.go b/pkg/handlers/generic/mutation/controlplanevirtualip/providers/kubevip_test.go
@@ -118,12 +118,11 @@ func Test_GenerateFilesAndCommands(t *testing.T) {
 				configMapKey: client.ObjectKeyFromObject(tt.configMap),
 			}
 
-			files, preKubeadmCommands, postKubeadmCommands, err :=
-				provider.GenerateFilesAndCommands(
-					context.TODO(),
-					tt.controlPlaneEndpointSpec,
-					tt.cluster,
-				)
+			files, preKubeadmCommands, postKubeadmCommands, err := provider.GenerateFilesAndCommands(
+				context.TODO(),
+				tt.controlPlaneEndpointSpec,
+				tt.cluster,
+			)
 			require.Equal(t, tt.expectedErr, err)
 			assert.Equal(t, tt.expectedFiles, files)
 			assert.Equal(t, tt.expectedPreKubeadmCommands, preKubeadmCommands)