patch - tco_policy - support iterations over fields (#161)

* patch - tco_policy - support iterations over fields
coralogix · Oct 18, 2023 · 4d031c6 · 4d031c6
1 parent a52bd60
commit 4d031c6
Show file tree

Hide file tree

Showing 5 changed files with 313 additions and 152 deletions.
diff --git a/...logix/data_source_coralogix_tco_policy.go → .../data_source_coralogix_tco_policy_logs.go b/...logix/data_source_coralogix_tco_policy.go → .../data_source_coralogix_tco_policy_logs.go
@@ -56,7 +56,7 @@ func (d *TCOPolicyDataSource) Schema(_ context.Context, _ datasource.SchemaReque
 }
 
 func (d *TCOPolicyDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
-	var data TCOPolicyResourceModel
+	var data *TCOPolicyResourceModel
 	resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
 	if resp.Diagnostics.HasError() {
 		return
@@ -84,8 +84,11 @@ func (d *TCOPolicyDataSource) Read(ctx context.Context, req datasource.ReadReque
 	}
 	log.Printf("[INFO] Received tco-policy: %#v", getPolicyResp)
 
-	data = flattenTCOPolicy(getPolicyResp.GetPolicy())
-
+	data, diags := flattenTCOPolicy(ctx, getPolicyResp.GetPolicy())
+	if diags.HasError() {
+		resp.Diagnostics = diags
+		return
+	}
 	// Save data into Terraform state
-	resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
+	resp.Diagnostics.Append(resp.State.Set(ctx, data)...)
 }
diff --git a/.../data_source_coralogix_tco_policy_test.go → ..._source_coralogix_tco_policy_logs_test.go b/.../data_source_coralogix_tco_policy_test.go → ..._source_coralogix_tco_policy_logs_test.go
diff --git a/coralogix/data_source_coralogix_tco_policy_traces.go b/coralogix/data_source_coralogix_tco_policy_traces.go
@@ -56,7 +56,7 @@ func (d *TCOPolicyTracesDataSource) Schema(_ context.Context, _ datasource.Schem
 }
 
 func (d *TCOPolicyTracesDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
-	var data TCOPolicyTracesResourceModel
+	var data *TCOPolicyTracesResourceModel
 	resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
 	if resp.Diagnostics.HasError() {
 		return
@@ -84,8 +84,11 @@ func (d *TCOPolicyTracesDataSource) Read(ctx context.Context, req datasource.Rea
 	}
 	log.Printf("[INFO] Received tco-policy: %#v", getPolicyResp)
 
-	data = flattenTCOPolicyTraces(ctx, getPolicyResp.GetPolicy())
-
+	data, diags := flattenTCOPolicyTraces(ctx, getPolicyResp.GetPolicy())
+	if diags.HasError() {
+		resp.Diagnostics = diags
+		return
+	}
 	// Save data into Terraform state
 	resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
 }