refactor for loop updateRootSize and use const

Author: Laure-di

internal/services/baremetal/easy_partitioning_data_source.go

@@ -19,6 +19,13 @@ import (
 const (
 	partitionSize     = 20000000000
 	defaultMountpoint = "/data"
+	md0               = "/dev/md0"
+	md1               = "/dev/md1"
+	raidLevel1        = "raid_level_1"
+	nvme0p2           = "/dev/nvme0n1p2"
+	nvme0p3           = "/dev/nvme0n1p3"
+	nvme1p1           = "/dev/nvme1n1p1"
+	nvme1p2           = "/dev/nvme1n1p2"
 )
 
 func DataEasyPartitioning() *schema.Resource {
@@ -152,19 +159,19 @@ func dataEasyPartitioningRead(ctx context.Context, d *schema.ResourceData, m int
 func updateRaidRemoveSwap(partitionSchema *baremetal.Schema) {
 	raidSchema := []*baremetal.SchemaRAID{
 		{
-			Name:  "/dev/md0",
-			Level: "raid_level_1",
+			Name:  md0,
+			Level: raidLevel1,
 			Devices: []string{
-				"/dev/nvme0n1p2",
-				"/dev/nvme1n1p1",
+				nvme0p2,
+				nvme1p1,
 			},
 		},
 		{
-			Name:  "/dev/md1",
-			Level: "raid_level_1",
+			Name:  md1,
+			Level: raidLevel1,
 			Devices: []string{
-				"/dev/nvme0n1p3",
-				"/dev/nvme1n1p2",
+				nvme0p3,
+				nvme1p2,
 			},
 		},
 	}
@@ -211,12 +218,11 @@ func updateSizeRoot(originalDisks []*baremetal.SchemaDisk, hasExtraPartition boo
 	for _, disk := range originalDisks {
 		for _, partition := range disk.Partitions {
 			if partition.Label == "root" {
+				partition.Size = 0
+				partition.UseAllAvailableSpace = true
 				if hasExtraPartition {
 					partition.Size = partitionSize
 					partition.UseAllAvailableSpace = false
-				} else {
-					partition.Size = 0
-					partition.UseAllAvailableSpace = true
 				}
 			}
 		}