Impl flatternASKConfig for os_disk_type

castai · Dec 15, 2023 · 3c1707f · 3c1707f
1 parent a179b42
commit 3c1707f
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/castai/resource_node_configuration.go b/castai/resource_node_configuration.go
@@ -706,9 +706,29 @@ func flattenAKSConfig(config *sdk.NodeconfigV1AKSConfig) []map[string]interface{
 		m["max_pods_per_node"] = *config.MaxPodsPerNode
 	}
 
+	if v := config.MaxPodsPerNode; v != nil {
+		m["os_disk_type"] = fromAKSDiskType(config.OsDiskType)
+	}
+
 	return []map[string]interface{}{m}
 }
 
+func fromAKSDiskType(osDiskType *sdk.NodeconfigV1AKSConfigOsDiskType) string {
+	if osDiskType == nil {
+		return ""
+	}
+	switch *osDiskType {
+	case sdk.OSDISKTYPESTANDARD:
+		return "standard"
+	case sdk.OSDISKTYPESTANDARDSSD:
+		return "standard-ssd"
+	case sdk.OSDISKTYPEPREMIUMSSD:
+		return "premium-ssd"
+	default:
+		return ""
+	}
+}
+
 func toGKEConfig(obj map[string]interface{}) *sdk.NodeconfigV1GKEConfig {
 	if obj == nil {
 		return nil