From 91622428555c2fe3342fe174bf16388be1698df4 Mon Sep 17 00:00:00 2001
From: re_sh_cloud_arch <94049389+re-sh-cloud-arch@users.noreply.github.com>
Date: Wed, 4 Dec 2024 13:52:04 +0100
Subject: [PATCH] fix: Storage postgres replica parameter (#384)

* fix storage postgres replica

* pre-commit

* pre-commit docs
---
 postgres_flexible_server_replica/01_main.tf      | 4 ++--
 postgres_flexible_server_replica/04_variables.tf | 6 +++++-
 postgres_flexible_server_replica/README.md       | 1 +
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/postgres_flexible_server_replica/01_main.tf b/postgres_flexible_server_replica/01_main.tf
index 75cf80cc..73cc96cc 100644
--- a/postgres_flexible_server_replica/01_main.tf
+++ b/postgres_flexible_server_replica/01_main.tf
@@ -20,8 +20,8 @@ resource "azurerm_postgresql_flexible_server" "this" {
   #  private_dns_zobe_id will be required when setting a delegated_subnet_id
   private_dns_zone_id = var.private_endpoint_enabled ? var.private_dns_zone_id : null
 
-  sku_name = var.sku_name
-
+  sku_name         = var.sku_name
+  storage_mb       = var.storage_mb
   source_server_id = var.source_server_id
 
   dynamic "high_availability" {
diff --git a/postgres_flexible_server_replica/04_variables.tf b/postgres_flexible_server_replica/04_variables.tf
index ebf9691e..ad7c27b9 100644
--- a/postgres_flexible_server_replica/04_variables.tf
+++ b/postgres_flexible_server_replica/04_variables.tf
@@ -74,7 +74,11 @@ variable "sku_name" {
   description = "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)."
 }
 
-
+variable "storage_mb" {
+  type        = number
+  description = "The max storage allowed for the PostgreSQL Flexible Server. Possible values are 32768, 65536, 131072, 262144, 524288, 1048576, 2097152, 4194304, 8388608, 16777216, and 33554432."
+  default     = null
+}
 
 
 variable "zone" {
diff --git a/postgres_flexible_server_replica/README.md b/postgres_flexible_server_replica/README.md
index a76f2357..91cc9a62 100644
--- a/postgres_flexible_server_replica/README.md
+++ b/postgres_flexible_server_replica/README.md
@@ -190,6 +190,7 @@ No modules.
 | <a name="input_sku_name"></a> [sku\_name](#input\_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). | `string` | n/a | yes |
 | <a name="input_source_server_id"></a> [source\_server\_id](#input\_source\_server\_id) | (Required) Id of the source server to be replicated | `string` | n/a | yes |
 | <a name="input_standby_availability_zone"></a> [standby\_availability\_zone](#input\_standby\_availability\_zone) | (Optional) Specifies the Availability Zone in which the standby Flexible Server should be located. | `number` | `null` | no |
+| <a name="input_storage_mb"></a> [storage\_mb](#input\_storage\_mb) | The max storage allowed for the PostgreSQL Flexible Server. Possible values are 32768, 65536, 131072, 262144, 524288, 1048576, 2097152, 4194304, 8388608, 16777216, and 33554432. | `number` | `null` | no |
 | <a name="input_tags"></a> [tags](#input\_tags) | n/a | `map(any)` | n/a | yes |
 | <a name="input_zone"></a> [zone](#input\_zone) | (Optional) Specifies the Availability Zone in which the PostgreSQL Flexible Server should be located. | `number` | `null` | no |