data.azurerm_postgresql_flexible_server - add support for zone and high_availability (hashicorp#31034)

[ENHANCEMENT] * `data.azurerm_postgresql_flexible_server` - add support for `zone` and `high_availability`

Author: tspearconquest

internal/services/postgres/postgresql_flexible_server_data_source.go

@@ -82,6 +82,26 @@ func dataSourcePostgresqlFlexibleServer() *pluginsdk.Resource {
 			},
 
 			"tags": commonschema.TagsDataSource(),
+
+			"high_availability": {
+				Type:     pluginsdk.TypeList,
+				Computed: true,
+				Elem: &pluginsdk.Resource{
+					Schema: map[string]*pluginsdk.Schema{
+						"mode": {
+							Type:     pluginsdk.TypeString,
+							Computed: true,
+						},
+
+						"standby_availability_zone": {
+							Type:     pluginsdk.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+
+			"zone": commonschema.ZoneSingleComputed(),
 		},
 	}
 }
@@ -116,6 +136,7 @@ func dataSourceArmPostgresqlFlexibleServerRead(d *pluginsdk.ResourceData, meta i
 			d.Set("administrator_login", props.AdministratorLogin)
 			d.Set("version", string(pointer.From(props.Version)))
 			d.Set("fqdn", props.FullyQualifiedDomainName)
+			d.Set("zone", props.AvailabilityZone)
 
 			if storage := props.Storage; storage != nil {
 				if storage.AutoGrow != nil {
@@ -139,6 +160,10 @@ func dataSourceArmPostgresqlFlexibleServerRead(d *pluginsdk.ResourceData, meta i
 				}
 				d.Set("public_network_access_enabled", publicNetworkAccess)
 			}
+
+			if err := d.Set("high_availability", flattenFlexibleServerHighAvailability(props.HighAvailability)); err != nil {
+				return fmt.Errorf("flattening `high_availability`: %+v", err)
+			}
 		}
 
 		sku, err := flattenFlexibleServerSku(model.Sku)

internal/services/postgres/postgresql_flexible_server_data_source_test.go

@@ -29,6 +29,7 @@ func TestAccDataSourcePostgresqlflexibleServer_basic(t *testing.T) {
 				check.That(data.ResourceName).Key("backup_retention_days").Exists(),
 				check.That(data.ResourceName).Key("fqdn").Exists(),
 				check.That(data.ResourceName).Key("public_network_access_enabled").Exists(),
+				check.That(data.ResourceName).Key("zone").Exists(),
 			),
 		},
 	})

website/docs/d/postgresql_flexible_server.html.markdown

@@ -49,6 +49,8 @@ In addition to the Arguments listed above - the following Attributes are exporte
 
 * `fqdn` - The FQDN of the PostgreSQL Flexible Server.
 
+* `high_availability` - A `high_availability` block for this PostgreSQL Flexible Server as defined below.
+
 * `public_network_access_enabled` - Is public network access enabled?
 
 * `sku_name` - The SKU Name for the PostgreSQL Flexible Server. The name of the SKU, follows the `tier` + `name` pattern (e.g. `B_Standard_B1ms`, `GP_Standard_D2s_v3`, `MO_Standard_E4s_v3`).
@@ -57,8 +59,18 @@ In addition to the Arguments listed above - the following Attributes are exporte
 
 * `version` - The version of PostgreSQL Flexible Server to use.
 
+* `zone` - The Availability Zones where this PostgreSQL Flexible Server is located.
+
 * `tags` - A mapping of tags assigned to the PostgreSQL Flexible Server.
 
+---
+
+A `high_availability` block exports the following:
+
+* `mode` - The high availability mode of the PostgreSQL Flexible Server.
+
+* `standby_availability_zone` - The availability zone of the standby Flexible Server.
+
 ## Timeouts
 
 The `timeouts` block allows you to specify [timeouts](https://developer.hashicorp.com/terraform/language/resources/configure#define-operation-timeouts) for certain actions: