Add data source sumologic_partition

Author: ErikAtSumo

sumologic/sumologic_partition.go

@@ -63,6 +63,47 @@ func (s *Client) UpdatePartition(spartition Partition) error {
 	return err
 }
 
+func (s *Client) ListPartitions() ([]Partition, error) {
+	var listPartitionResp ListPartitionResp
+
+	data, err := s.Get("v1/partitions")
+	if err != nil {
+		return nil, err
+	}
+
+	err = json.Unmarshal(data, &listPartitionResp)
+	if err != nil {
+		return nil, err
+	}
+
+	spartitions := listPartitionResp.Data
+
+	for listPartitionResp.Next != "" {
+		data, err = s.Get("v1/partitions?token=" + listPartitionResp.Next)
+		if err != nil {
+			return nil, err
+		}
+
+		listPartitionResp.Reset()
+
+		err = json.Unmarshal(data, &listPartitionResp)
+		if err != nil {
+			return nil, err
+		}
+
+		spartitions = append(spartitions, listPartitionResp.Data...)
+	}
+
+	var activePartitions []Partition
+	for _, partition := range spartitions {
+		if partition.IsActive {
+			activePartitions = append(activePartitions, partition)
+		}
+	}
+
+	return activePartitions, nil
+}
+
 type Partition struct {
 	ID                               string  `json:"id,omitempty"`
 	Name                             string  `json:"name"`
@@ -77,3 +118,13 @@ type Partition struct {
 	ReduceRetentionPeriodImmediately bool    `json:"reduceRetentionPeriodImmediately,omitempty"`
 	IsIncludedInDefaultSearch        *bool   `json:"isIncludedInDefaultSearch,omitempty"`
 }
+
+type ListPartitionResp struct {
+	Data []Partition `json:"data"`
+	Next string      `json:"next"`
+}
+
+func (s *ListPartitionResp) Reset() {
+	s.Data = nil
+	s.Next = ""
+}