Get previous disk usage only from existing machine configs (#6700)

Co-authored-by: Abhay Krishna Arunachalam <arnchlm@amazon.com>
aws · Sep 19, 2023 · b07acd0 · b07acd0
1 parent 0100084
commit b07acd0
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/pkg/providers/vsphere/vsphere.go b/pkg/providers/vsphere/vsphere.go
@@ -471,12 +471,14 @@ type datastoreUsage struct {
 }
 
 func (p *vsphereProvider) getPrevMachineConfigDatastoreUsage(ctx context.Context, machineConfig *v1alpha1.VSphereMachineConfig, cluster *types.Cluster, count int) (diskGiB float64, err error) {
-	em, err := p.providerKubectlClient.GetEksaVSphereMachineConfig(ctx, machineConfig.Name, cluster.KubeconfigFile, machineConfig.GetNamespace())
-	if err != nil {
-		return 0, err
-	}
-	if em != nil {
-		return float64(em.Spec.DiskGiB * count), nil
+	if count > 0 {
+		em, err := p.providerKubectlClient.GetEksaVSphereMachineConfig(ctx, machineConfig.Name, cluster.KubeconfigFile, machineConfig.GetNamespace())
+		if err != nil {
+			return 0, err
+		}
+		if em != nil {
+			return float64(em.Spec.DiskGiB * count), nil
+		}
 	}
 	return 0, nil
 }