Merge pull request #113 from trickest/fix/machine-type-conversion

Fix machine type conversion
trickest · Mar 8, 2024 · 64458eb · 64458eb
2 parents 9752fd3 + f775800
commit 64458eb
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/cmd/execute/execute.go b/cmd/execute/execute.go
@@ -189,7 +189,16 @@ func handleSingleMachineType(fleet types.Fleet, machines types.Machines) (types.
 		defaultOrSelfHosted = *machines.Default
 	} else {
 		// Backward-compatibility with the small-medium-large format
-		defaultOrSelfHosted = *machines.Small + *machines.Medium + *machines.Large
+		if machines.Small != nil {
+			defaultOrSelfHosted += *machines.Small
+		}
+		if machines.Medium != nil {
+			defaultOrSelfHosted += *machines.Medium
+		}
+		if machines.Large != nil {
+			defaultOrSelfHosted += *machines.Large
+		}
+
 		fmt.Printf("Warning: You have one type of machine in your fleet. %d identical or self-hosted machines will be used.\n", defaultOrSelfHosted)
 	}