fix: custom plan modifier

EnterpriseDB · Aug 19, 2024 · c165509 · c165509
1 parent 4b97360
commit c165509
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pkg/provider/resource_cluster.go b/pkg/provider/resource_cluster.go
@@ -530,7 +530,7 @@ func (c *clusterResource) Schema(ctx context.Context, req resource.SchemaRequest
 					},
 				},
 				PlanModifiers: []planmodifier.Set{
-					setplanmodifier.UseStateForUnknown(),
+					plan_modifier.CustomAssignTags(),
 				},
 			},
 			"transparent_data_encryption": schema.SingleNestedAttribute{

diff --git a/pkg/provider/resource_pgd.go b/pkg/provider/resource_pgd.go
@@ -125,7 +125,7 @@ func PgdSchema(ctx context.Context) schema.Schema {
 					},
 				},
 				PlanModifiers: []planmodifier.Set{
-					setplanmodifier.UseStateForUnknown(),
+					plan_modifier.CustomAssignTags(),
 				},
 			},
 			"data_groups": schema.ListNestedAttribute{

diff --git a/pkg/provider/resource_project.go b/pkg/provider/resource_project.go
@@ -8,6 +8,7 @@ import (
 	"github.com/EnterpriseDB/terraform-provider-biganimal/pkg/api"
 	"github.com/EnterpriseDB/terraform-provider-biganimal/pkg/models"
 	commonTerraform "github.com/EnterpriseDB/terraform-provider-biganimal/pkg/models/common/terraform"
+	"github.com/EnterpriseDB/terraform-provider-biganimal/pkg/plan_modifier"
 	"github.com/hashicorp/terraform-plugin-framework/resource"
 	"github.com/hashicorp/terraform-plugin-framework/resource/schema"
 	"github.com/hashicorp/terraform-plugin-framework/resource/schema/int64planmodifier"
@@ -123,7 +124,7 @@ func (p projectResource) Schema(ctx context.Context, req resource.SchemaRequest,
 					},
 				},
 				PlanModifiers: []planmodifier.Set{
-					setplanmodifier.UseStateForUnknown(),
+					plan_modifier.CustomAssignTags(),
 				},
 			},
 		},