fix: update flattenAndSetQuery interface

observeinc · Jan 17, 2024 · 03d19a4 · 03d19a4
1 parent 95081ba
commit 03d19a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/observe/resource_dataset.go b/observe/resource_dataset.go
@@ -248,11 +248,7 @@ func datasetToResourceData(d *gql.Dataset, data *schema.ResourceData) (diags dia
 	}
 
 	if d.Transform != nil && d.Transform.Current != nil {
-		stages := make([]*gql.StageQuery, 0)
-		for _, stage := range d.Transform.Current.Query.Stages {
-			stages = append(stages, &stage)
-		}
-		_, err := flattenAndSetQuery(data, stages, d.Transform.Current.Query.OutputStage)
+		_, err := flattenAndSetQuery(data, d.Transform.Current.Query.Stages, d.Transform.Current.Query.OutputStage)
 		if err != nil {
 			diags = append(diags, diag.FromErr(err)...)
 		}

diff --git a/observe/resource_monitor.go b/observe/resource_monitor.go
@@ -819,11 +819,7 @@ func resourceMonitorRead(ctx context.Context, data *schema.ResourceData, meta in
 		diags = append(diags, diag.FromErr(err)...)
 	}
 
-	stages := make([]*gql.StageQuery, 0)
-	for _, stage := range monitor.Query.Stages {
-		stages = append(stages, &stage)
-	}
-	stageIds, err := flattenAndSetQuery(data, stages, monitor.Query.OutputStage)
+	stageIds, err := flattenAndSetQuery(data, monitor.Query.Stages, monitor.Query.OutputStage)
 	if err != nil {
 		diags = append(diags, diag.FromErr(err)...)
 	}