CDPCP-11600 - Revisit Availability zone on datahub

docs/resources/datahub_azure_cluster.md

@@ -231,6 +231,7 @@ Required:
 
 Optional:
 
+- `availability_zones` (Set of String) List of availability zones that this instance group is associated with.
 - `recipes` (Set of String) The set of recipe names that are going to be applied on the given instance group.
 
 <a id="nestedatt--instance_group--attached_volume_configuration"></a>

resources/datahub/common_scheme.go

@@ -146,3 +146,76 @@ var instanceGroupSchemaAttributes = map[string]schema.Attribute{
 		},
 	},
 }
+
+var azureInstanceGroupSchemaAttributes = map[string]schema.Attribute{
+	"instance_group": schema.ListNestedAttribute{
+		Optional: true,
+		NestedObject: schema.NestedAttributeObject{
+			Attributes: map[string]schema.Attribute{
+				"node_count": schema.Int64Attribute{
+					MarkdownDescription: "The cluster node count. Has to be greater or equal than 0 and less than 100,000.",
+					Required:            true,
+				},
+				"availability_zones": schema.SetAttribute{
+					MarkdownDescription: "List of availability zones that this instance group is associated with.",
+					ElementType:         types.StringType,
+					Optional:            true,
+				},
+				"instance_group_name": schema.StringAttribute{
+					MarkdownDescription: "The name of the instance group.",
+					Required:            true,
+				},
+				"instance_group_type": schema.StringAttribute{
+					MarkdownDescription: "The type of the instance group.",
+					Required:            true,
+				},
+				"instance_type": schema.StringAttribute{
+					MarkdownDescription: "The cloud provider-side instance type.",
+					Required:            true,
+				},
+				"root_volume_size": schema.Int64Attribute{
+					MarkdownDescription: "The size of the root volume in GB",
+					Required:            true,
+				},
+				"recipes": schema.SetAttribute{
+					MarkdownDescription: "The set of recipe names that are going to be applied on the given instance group.",
+					ElementType:         types.StringType,
+					Optional:            true,
+				},
+				"attached_volume_configuration": schema.ListNestedAttribute{
+					Required:            true,
+					MarkdownDescription: "Configuration regarding the attached volume to the specific instance group.",
+					NestedObject: schema.NestedAttributeObject{
+						Attributes: map[string]schema.Attribute{
+							"volume_size": schema.Int64Attribute{
+								MarkdownDescription: "The size of the volume in GB.",
+								Required:            true,
+							},
+							"volume_count": schema.Int64Attribute{
+								MarkdownDescription: "The number of volumes to be attached.",
+								Required:            true,
+							},
+							"volume_type": schema.StringAttribute{
+								MarkdownDescription: "The - cloud provider - type of the volume.",
+								Required:            true,
+							},
+						},
+					},
+				},
+				"recovery_mode": schema.StringAttribute{
+					MarkdownDescription: "The type of the recovery mode.",
+					Required:            true,
+				},
+				"volume_encryption": schema.SingleNestedAttribute{
+					MarkdownDescription: "The volume encryption related configuration.",
+					Required:            true,
+					Attributes: map[string]schema.Attribute{
+						"encryption": schema.BoolAttribute{
+							Required: true,
+						},
+					},
+				},
+			},
+		},
+	},
+}

resources/datahub/converter.go

@@ -127,6 +127,12 @@ func fromModelToAzureRequest(model azureDatahubResourceModel, ctx context.Contex
 				igRecipes = append(igRecipes, recipe.ValueString())
 			}
 		}
+		var azs []string
+		if group.AvailabilityZones != nil && len(group.AvailabilityZones) > 0 {
+			for _, az := range group.AvailabilityZones {
+				azs = append(azs, az.ValueString())
+			}
+		}
 		rootVolumeSize := int32(group.RootVolumeSize.ValueInt64())
 		ig := &datahubmodels.AzureInstanceGroupRequest{
 			AttachedVolumeConfiguration: volReqs,
@@ -137,6 +143,7 @@ func fromModelToAzureRequest(model azureDatahubResourceModel, ctx context.Contex
 			RecipeNames:                 igRecipes,
 			RecoveryMode:                group.RecoveryMode.ValueString(),
 			RootVolumeSize:              &rootVolumeSize,
+			AvailabilityZones:           azs,
 		}
 		igs = append(igs, ig)
 	}

resources/datahub/model_datahub.go

@@ -24,6 +24,7 @@ type InstanceGroup struct {
 	RecoveryMode                types.String                  `tfsdk:"recovery_mode"`
 	VolumeEncryption            VolumeEncryption              `tfsdk:"volume_encryption"`
 	Recipes                     []types.String                `tfsdk:"recipes"`
+	AvailabilityZones           []types.String                `tfsdk:"availability_zones"`
 }
 
 type AttachedVolumeConfiguration struct {

resources/datahub/schema_azure_datahub.go

@@ -27,7 +27,7 @@ import (
 func (r *azureDatahubResource) Schema(_ context.Context, _ resource.SchemaRequest, resp *resource.SchemaResponse) {
 	attr := map[string]schema.Attribute{}
 	utils.Append(attr, generalAttributes)
-	utils.Append(attr, instanceGroupSchemaAttributes)
+	utils.Append(attr, azureInstanceGroupSchemaAttributes)
 	utils.Append(attr, map[string]schema.Attribute{
 		"cluster_template": schema.StringAttribute{
 			MarkdownDescription: "The name of the cluster template.",