fix: pagination not working correctly

SumoLogic · Jun 11, 2024 · 5393410 · 5393410
1 parent ea0b004
commit 5393410
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/sumologic/resource_sumologic_s3_data_forwarding_destination.go b/sumologic/resource_sumologic_s3_data_forwarding_destination.go
@@ -80,7 +80,7 @@ func resourceSumologicS3DataForwardingDestinationCreate(d *schema.ResourceData,
 		d.SetId(createdDfd.ID)
 	}
 
-	return resourceSumologicS3DataForwardingDestinationUpdate(d, meta)
+	return resourceSumologicS3DataForwardingDestinationRead(d, meta)
 }
 
 func resourceSumologicS3DataForwardingDestinationRead(d *schema.ResourceData, meta interface{}) error {

diff --git a/sumologic/resource_sumologic_s3_data_forwarding_rule.go b/sumologic/resource_sumologic_s3_data_forwarding_rule.go
@@ -57,7 +57,7 @@ func resourceSumologicS3DataForwardingRuleCreate(d *schema.ResourceData, meta in
 		d.SetId(createdDfd.IndexID)
 	}
 
-	return resourceSumologicS3DataForwardingRuleUpdate(d, meta)
+	return resourceSumologicS3DataForwardingRuleRead(d, meta)
 }
 
 func resourceSumologicS3DataForwardingRuleRead(d *schema.ResourceData, meta interface{}) error {

diff --git a/sumologic/sumologic_partition.go b/sumologic/sumologic_partition.go
@@ -11,10 +11,15 @@ type ListPartitionResp struct {
 	Next string      `json:"next"`
 }
 
+func (s *ListPartitionResp) Reset() {
+	s.Data = nil
+	s.Next = ""
+}
+
 func (s *Client) ListPartitions() ([]Partition, error) {
 	var listPartitionResp ListPartitionResp
 
-	data, _, err := s.Get("v1/partitions")
+	data, _, err := s.Get("v1/partitions?limit=1000")
 	if err != nil {
 		return nil, err
 	}
@@ -32,6 +37,8 @@ func (s *Client) ListPartitions() ([]Partition, error) {
 			return nil, err
 		}
 
+		listPartitionResp.Reset()
+
 		err = json.Unmarshal(data, &listPartitionResp)
 		if err != nil {
 			return nil, err