feat: wal storage

examples/resources/biganimal_cluster/ha/resource.tf

@@ -68,6 +68,12 @@ resource "biganimal_cluster" "ha_cluster" {
     size              = "4 Gi"
   }
 
+#  wal_storage = {
+#    volume_type       = "gp3"
+#    volume_properties = "gp3"
+#    size              = "4 Gi"
+#  }
+
   maintenance_window = {
     is_enabled = true
     start_day  = 6

examples/resources/biganimal_cluster/single_node/aws/resource.tf

@@ -69,6 +69,12 @@ resource "biganimal_cluster" "single_node_cluster" {
     size              = "4 Gi"
   }
 
+#  wal_storage = {
+#    volume_type       = "gp3"
+#    volume_properties = "gp3"
+#    size              = "4 Gi"
+#  }
+
   maintenance_window = {
     is_enabled = true
     start_day  = 6

examples/resources/biganimal_cluster/single_node/azure/resource.tf

@@ -69,6 +69,12 @@ resource "biganimal_cluster" "single_node_cluster" {
     size              = "4 Gi"
   }
 
+#  wal_storage = {
+#    volume_type       = "azurepremiumstorage"
+#    volume_properties = "P1"
+#    size              = "4 Gi"
+#  }
+
   maintenance_window = {
     is_enabled = true
     start_day  = 6

examples/resources/biganimal_cluster/single_node/gcp/resource.tf

@@ -69,6 +69,12 @@ resource "biganimal_cluster" "single_node_cluster" {
     size              = "10 Gi"
   }
 
+#  wal_storage = {
+#    volume_type       = "pd-ssd"
+#    volume_properties = "pd-ssd"
+#    size              = "10 Gi"
+#  }
+
   maintenance_window = {
     is_enabled = true
     start_day  = 6

examples/resources/biganimal_faraway_replica/aws/resource.tf

@@ -66,6 +66,11 @@ resource "biganimal_faraway_replica" "faraway_replica" {
     volume_properties = "gp3"
     size              = "4 Gi"
   }
+#  wal_storage = {
+#    volume_type       = "gp3"
+#    volume_properties = "gp3"
+#    size              = "4 Gi"
+#  }
   private_networking = false
   region             = "ap-south-1"
 

examples/resources/biganimal_faraway_replica/azure/resource.tf

@@ -66,6 +66,11 @@ resource "biganimal_faraway_replica" "faraway_replica" {
     volume_properties = "P1"
     size              = "4 Gi"
   }
+#  wal_storage = {
+#    volume_type       = "azurepremiumstorage"
+#    volume_properties = "P1"
+#    size              = "4 Gi"
+#  }
   private_networking = false
   region             = "australiaeast"
 

examples/resources/biganimal_faraway_replica/cluster_and_faraway_replica/resource.tf

@@ -102,6 +102,11 @@ resource "biganimal_faraway_replica" "faraway_replica" {
     volume_properties = "P1"
     size              = "4 Gi"
   }
+#  wal_storage = {
+#    volume_type       = "azurepremiumstorage"
+#    volume_properties = "P1"
+#    size              = "4 Gi"
+#  }
   private_networking = false
   region             = "centralindia"
 

examples/resources/biganimal_faraway_replica/gcp/resource.tf

@@ -66,6 +66,11 @@ resource "biganimal_faraway_replica" "faraway_replica" {
     volume_properties = "pd-ssd"
     size              = "4 Gi"
   }
+#  wal_storage = {
+#    volume_type       = "pd-ssd"
+#    volume_properties = "pd-ssd"
+#    size              = "4 Gi"
+#  }
   private_networking = false
   region             = "us-east1"
 

examples/resources/biganimal_pgd/aws/data_group/resource.tf

@@ -77,6 +77,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "gp3"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "gp3"
+#        volume_properties = "gp3"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

examples/resources/biganimal_pgd/aws/data_groups_with_witness_group/resource.tf

@@ -77,6 +77,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "gp3"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "gp3"
+#        volume_properties = "gp3"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }
@@ -134,6 +139,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "gp3"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "gp3"
+#        volume_properties = "gp3"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

examples/resources/biganimal_pgd/azure/data_group/resource.tf

@@ -77,6 +77,16 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "P2"
         size              = "32 Gi"
       }
+      storage = {
+        volume_type       = "azurepremiumstorage"
+        volume_properties = "P2"
+        size              = "32 Gi"
+      }
+#      wal_storage = {
+#        volume_type       = "azurepremiumstorage"
+#        volume_properties = "P2"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

examples/resources/biganimal_pgd/azure/data_groups_with_witness_group/resource.tf

@@ -77,6 +77,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "P2"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "azurepremiumstorage"
+#        volume_properties = "P2"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }
@@ -134,6 +139,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "P2"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "azurepremiumstorage"
+#        volume_properties = "P2"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

examples/resources/biganimal_pgd/gcp/data_group/resource.tf

@@ -77,6 +77,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "pd-ssd"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "pd-ssd"
+#        volume_properties = "pd-ssd"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

examples/resources/biganimal_pgd/gcp/data_groups_with_witness_group/resource.tf

@@ -77,6 +77,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "pd-ssd"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "pd-ssd"
+#        volume_properties = "pd-ssd"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }
@@ -138,6 +143,11 @@ resource "biganimal_pgd" "pgd_cluster" {
         volume_properties = "pd-ssd"
         size              = "32 Gi"
       }
+#      wal_storage = {
+#        volume_type       = "pd-ssd"
+#        volume_properties = "pd-ssd"
+#        size              = "32 Gi"
+#      }
       pg_type = {
         pg_type_id = "epas" #valid values ["epas", "pgextended", "postgres]"
       }

pkg/models/cluster.go

@@ -186,6 +186,7 @@ type Cluster struct {
 	EncryptionKeyIdReq         *string                      `json:"keyId,omitempty"`
 	EncryptionKeyResp          *EncryptionKey               `json:"encryptionKey,omitempty"`
 	PgIdentity                 *string                      `json:"pgIdentity,omitempty"`
+	WalStorage                 *Storage                     `json:"walStorage,omitempty"`
 }
 
 // IsHealthy checks to see if the cluster has the right condition 'biganimal.com/deployed'

pkg/models/pgd/api/data_group.go

@@ -30,4 +30,5 @@ type DataGroup struct {
 	PeAllowedPrincipalIds *[]string                 `json:"peAllowedPrincipalIds,omitempty"`
 	RoConnectionUri       *string                   `json:"roConnectionUri,omitempty"`
 	ReadOnlyConnections   *bool                     `json:"readOnlyConnections,omitempty"`
+	WalStorage            *models.Storage           `json:"storage,omitempty"`
 }

pkg/models/pgd/terraform/data_group.go

@@ -33,4 +33,5 @@ type DataGroup struct {
 	PeAllowedPrincipalIds types.Set                 `tfsdk:"pe_allowed_principal_ids"`
 	RoConnectionUri       types.String              `tfsdk:"ro_connection_uri"`
 	ReadOnlyConnections   *bool                     `tfsdk:"read_only_connections"`
+	WalStorage            *Storage                  `tfsdk:"wal_storage"`
 }

pkg/plan_modifier/data_group_custom_diff.go

@@ -126,6 +126,11 @@ func (m CustomDataGroupDiffModifier) PlanModifyList(ctx context.Context, req pla
 				pDg.Storage.Iops = sDg.Storage.Iops
 				pDg.Storage.Throughput = sDg.Storage.Throughput
 
+				if sDg.WalStorage != nil {
+					pDg.WalStorage.Iops = sDg.WalStorage.Iops
+					pDg.WalStorage.Throughput = sDg.WalStorage.Throughput
+				}
+
 				// fix to set the correct allowed ip ranges to allow all if a PGD data group has private networking set as true
 				if pDg.PrivateNetworking != nil && *pDg.PrivateNetworking {
 					pDg.AllowedIpRanges = types.SetValueMust(pDg.AllowedIpRanges.ElementType(ctx), []attr.Value{

pkg/provider/common.go

@@ -3,6 +3,9 @@ package provider
 import (
 	commonApi "github.com/EnterpriseDB/terraform-provider-biganimal/pkg/models/common/api"
 	commonTerraform "github.com/EnterpriseDB/terraform-provider-biganimal/pkg/models/common/terraform"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier"
 	"github.com/hashicorp/terraform-plugin-framework/types"
 	"github.com/hashicorp/terraform-plugin-framework/types/basetypes"
 )
@@ -31,3 +34,35 @@ func buildAPIReqAssignTags(tfRsrcTags []commonTerraform.Tag) []commonApi.Tag {
 	}
 	return tags
 }
+
+var resourceWal = schema.SingleNestedAttribute{
+	Description: "Write-Ahead Logs (WAL) Storage.",
+	Optional:    true,
+	Attributes: map[string]schema.Attribute{
+		"iops": schema.StringAttribute{
+			Description:   "IOPS for the selected volume. It can be set to different values depending on your volume type and properties.",
+			Optional:      true,
+			Computed:      true,
+			PlanModifiers: []planmodifier.String{stringplanmodifier.UseStateForUnknown()},
+		},
+		"size": schema.StringAttribute{
+			Description:   "Size of the volume. It can be set to different values depending on your volume type and properties.",
+			Required:      true,
+			PlanModifiers: []planmodifier.String{stringplanmodifier.UseStateForUnknown()},
+		},
+		"throughput": schema.StringAttribute{
+			Description:   "Throughput is automatically calculated by BigAnimal based on the IOPS input if it's not provided.",
+			Optional:      true,
+			Computed:      true,
+			PlanModifiers: []planmodifier.String{stringplanmodifier.UseStateForUnknown()},
+		},
+		"volume_properties": schema.StringAttribute{
+			Description: "Volume properties in accordance with the selected volume type.",
+			Required:    true,
+		},
+		"volume_type": schema.StringAttribute{
+			Description: "Volume type. For Azure: \"azurepremiumstorage\" or \"ultradisk\". For AWS: \"gp3\", \"io2\", org s \"io2-block-express\". For Google Cloud: only \"pd-ssd\".",
+			Required:    true,
+		},
+	},
+}

pkg/provider/resource_cluster.go

@@ -82,6 +82,7 @@ type ClusterResourceModel struct {
 	VolumeSnapshot                  types.Bool                         `tfsdk:"volume_snapshot_backup"`
 	Tags                            []commonTerraform.Tag              `tfsdk:"tags"`
 	ServiceName                     types.String                       `tfsdk:"service_name"`
+	WalStorage                      *StorageResourceModel              `tfsdk:"wal_storage"`
 
 	Timeouts timeouts.Value `tfsdk:"timeouts"`
 }
@@ -575,6 +576,7 @@ func (c *clusterResource) Schema(ctx context.Context, req resource.SchemaRequest
 				Computed:            true,
 				PlanModifiers:       []planmodifier.String{stringplanmodifier.UseStateForUnknown()},
 			},
+			"wal_storage": resourceWal,
 		},
 	}
 }
@@ -863,6 +865,13 @@ func readCluster(ctx context.Context, client *api.ClusterClient, tfClusterResour
 	tfClusterResource.SuperuserAccess = types.BoolPointerValue(responseCluster.SuperuserAccess)
 	tfClusterResource.PgIdentity = types.StringPointerValue(responseCluster.PgIdentity)
 	tfClusterResource.VolumeSnapshot = types.BoolPointerValue(responseCluster.VolumeSnapshot)
+	tfClusterResource.WalStorage = &StorageResourceModel{
+		VolumeType:       types.StringPointerValue(responseCluster.WalStorage.VolumeTypeId),
+		VolumeProperties: types.StringPointerValue(responseCluster.WalStorage.VolumePropertiesId),
+		Size:             types.StringPointerValue(responseCluster.WalStorage.Size),
+		Iops:             types.StringPointerValue(responseCluster.WalStorage.Iops),
+		Throughput:       types.StringPointerValue(responseCluster.WalStorage.Throughput),
+	}
 
 	if responseCluster.EncryptionKeyResp != nil && *responseCluster.Phase != constants.PHASE_HEALTHY {
 		if !tfClusterResource.PgIdentity.IsNull() && tfClusterResource.PgIdentity.ValueString() != "" {
@@ -1102,6 +1111,16 @@ func (c *clusterResource) generateGenericClusterModel(ctx context.Context, clust
 		VolumeSnapshot:        clusterResource.VolumeSnapshot.ValueBoolPointer(),
 	}
 
+	if clusterResource.WalStorage != nil {
+		cluster.WalStorage = &models.Storage{
+			VolumePropertiesId: clusterResource.WalStorage.VolumeProperties.ValueStringPointer(),
+			VolumeTypeId:       clusterResource.WalStorage.VolumeType.ValueStringPointer(),
+			Iops:               clusterResource.WalStorage.Iops.ValueStringPointer(),
+			Size:               clusterResource.WalStorage.Size.ValueStringPointer(),
+			Throughput:         clusterResource.WalStorage.Throughput.ValueStringPointer(),
+		}
+	}
+
 	cluster.Extensions = &[]models.ClusterExtension{}
 	if clusterResource.Pgvector.ValueBool() {
 		*cluster.Extensions = append(*cluster.Extensions, models.ClusterExtension{Enabled: true, ExtensionId: "pgvector"})

pkg/provider/resource_fareplica.go

@@ -65,6 +65,7 @@ type FAReplicaResourceModel struct {
 	TransparentDataEncryptionAction types.String                      `tfsdk:"transparent_data_encryption_action"`
 	VolumeSnapshot                  types.Bool                        `tfsdk:"volume_snapshot_backup"`
 	Tags                            []commonTerraform.Tag             `tfsdk:"tags"`
+	WalStorage                      *StorageResourceModel             `tfsdk:"wal_storage"`
 
 	Timeouts timeouts.Value `tfsdk:"timeouts"`
 }
@@ -422,6 +423,7 @@ func (r *FAReplicaResource) Schema(ctx context.Context, req resource.SchemaReque
 					plan_modifier.CustomAssignTags(),
 				},
 			},
+			"wal_storage": resourceWal,
 		},
 	}
 }
@@ -636,6 +638,13 @@ func readFAReplica(ctx context.Context, client *api.ClusterClient, fAReplicaReso
 	fAReplicaResourceModel.PgVersion = types.StringValue(responseCluster.PgVersion.PgVersionId)
 	fAReplicaResourceModel.PgType = types.StringValue(responseCluster.PgType.PgTypeId)
 	fAReplicaResourceModel.VolumeSnapshot = types.BoolPointerValue(responseCluster.VolumeSnapshot)
+	fAReplicaResourceModel.WalStorage = &StorageResourceModel{
+		VolumeType:       types.StringPointerValue(responseCluster.WalStorage.VolumeTypeId),
+		VolumeProperties: types.StringPointerValue(responseCluster.WalStorage.VolumePropertiesId),
+		Size:             types.StringPointerValue(responseCluster.WalStorage.Size),
+		Iops:             types.StringPointerValue(responseCluster.WalStorage.Iops),
+		Throughput:       types.StringPointerValue(responseCluster.WalStorage.Throughput),
+	}
 
 	// pgConfig. If tf resource pg config elem matches with api response pg config elem then add the elem to tf resource pg config
 	newPgConfig := []PgConfigResourceModel{}
@@ -770,6 +779,16 @@ func (r *FAReplicaResource) generateGenericFAReplicaModel(ctx context.Context, f
 		BackupRetentionPeriod: fAReplicaResourceModel.BackupRetentionPeriod.ValueStringPointer(),
 	}
 
+	if fAReplicaResourceModel.WalStorage != nil {
+		cluster.WalStorage = &models.Storage{
+			VolumePropertiesId: fAReplicaResourceModel.WalStorage.VolumeProperties.ValueStringPointer(),
+			VolumeTypeId:       fAReplicaResourceModel.WalStorage.VolumeType.ValueStringPointer(),
+			Iops:               fAReplicaResourceModel.WalStorage.Iops.ValueStringPointer(),
+			Size:               fAReplicaResourceModel.WalStorage.Size.ValueStringPointer(),
+			Throughput:         fAReplicaResourceModel.WalStorage.Throughput.ValueStringPointer(),
+		}
+	}
+
 	allowedIpRanges := []models.AllowedIpRange{}
 	for _, ipRange := range fAReplicaResourceModel.AllowedIpRanges {
 		allowedIpRanges = append(allowedIpRanges, models.AllowedIpRange{