Add data source sumologic_partitions

Author: ErikAtSumo

CHANGELOG.md

@@ -3,6 +3,7 @@
 
 FEATURES:
 * **New Data Source:** sumologic_partition
+* **New Data Source:** sumologic_partitions
 
 ENHANCEMENTS:
 * Add support for searchable time for dashboard field time_source

sumologic/data_source_sumologic_partitions.go

@@ -0,0 +1,76 @@
+package sumologic
+
+import (
+	"crypto/sha256"
+	"encoding/hex"
+	"fmt"
+	"sort"
+	"strings"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+)
+
+func dataSourceSumologicPartitions() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceSumologicPartitionsRead,
+
+		Schema: map[string]*schema.Schema{
+			"partitions": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: dataSourcePartitionSchema(),
+					Read:   dataSourceSumologicPartitionRead,
+				},
+			},
+		},
+	}
+}
+
+func dataSourceSumologicPartitionsRead(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	spartitions, err := c.ListPartitions()
+	if err != nil {
+		return fmt.Errorf("error retrieving partitions: %v", err)
+	}
+
+	partitions := make([]map[string]interface{}, 0, len(spartitions))
+
+	for _, spartition := range spartitions {
+		partition := map[string]interface{}{
+			"id":                            spartition.ID,
+			"name":                          spartition.Name,
+			"routing_expression":            spartition.RoutingExpression,
+			"analytics_tier":                spartition.AnalyticsTier,
+			"retention_period":              spartition.RetentionPeriod,
+			"is_compliant":                  spartition.IsCompliant,
+			"total_bytes":                   spartition.TotalBytes,
+			"data_forwarding_id":            spartition.DataForwardingId,
+			"is_active":                     spartition.IsActive,
+			"index_type":                    spartition.IndexType,
+			"is_included_in_default_search": spartition.IsIncludedInDefaultSearch,
+		}
+
+		partitions = append(partitions, partition)
+	}
+
+	d.Set("partitions", partitions)
+	d.SetId(generatePartitionsId(spartitions))
+
+	return nil
+}
+
+func generatePartitionsId(partitions []Partition) string {
+	// Collect a sorted list of partition IDs
+	ids := []string{"partition_ids"}
+	for _, partition := range partitions {
+		ids = append(ids, partition.ID)
+	}
+	sort.Strings(ids)
+
+	// Join all IDs and create a hash
+	idString := strings.Join(ids, "|")
+	hash := sha256.Sum256([]byte(idString))
+	return hex.EncodeToString(hash[:])
+}

sumologic/data_source_sumologic_partitions_test.go

@@ -0,0 +1,26 @@
+package sumologic
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+)
+
+func TestAccDataSourcePartitions_basic(t *testing.T) {
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testDataSourceSumologicPartitionsConfig,
+				Check: resource.ComposeTestCheckFunc(
+					checkResourceAttrGreaterThanZero("data.sumologic_partitions.test", "partitions.#"),
+				),
+			},
+		},
+	})
+}
+
+var testDataSourceSumologicPartitionsConfig = `
+	data "sumologic_partitions" "test" {}
+`

sumologic/provider.go

@@ -135,6 +135,7 @@ func Provider() terraform.ResourceProvider {
 			"sumologic_folder":                         dataSourceSumologicFolder(),
 			"sumologic_my_user_id":                     dataSourceSumologicMyUserId(),
 			"sumologic_partition":                      dataSourceSumologicPartition(),
+			"sumologic_partitions":                     dataSourceSumologicPartitions(),
 			"sumologic_role":                           dataSourceSumologicRole(),
 			"sumologic_role_v2":                        dataSourceSumologicRoleV2(),
 			"sumologic_user":                           dataSourceSumologicUser(),

website/docs/d/partitions.html.markdown

@@ -0,0 +1,33 @@
+---
+layout: "sumologic"
+page_title: "SumoLogic: sumologic_partition"
+description: |-
+  Provides a way to retrieve all Sumo Logic partitions.
+---
+
+# sumologic_partition
+
+Provides a way to retrieve all [Sumologic Partitions][1].
+
+## Example Usage
+
+```hcl
+data "sumologic_partitions" "partitions" {}
+```
+
+## Attributes reference
+
+The following attributes are exported:
+
+- `name` - The name of the partition.
+- `routing_expression` - The query that defines the data to be included in the partition.
+- `analytics_tier` - The Data Tier where the data in the partition will reside. Possible values are: `continuous`, `frequent`, `infrequent`
+- `retention_period` - The number of days to retain data in the partition.
+- `is_compliant` - Whether the partition is used for compliance or audit purposes.
+- `is_included_in_default_search` - Whether the partition is included in the default search scope.
+- `total_bytes` - The size of the data in the partition in bytes.
+- `is_active` - Whether the partition is currently active or decommissioned.
+- `index_type` - The type of partition index. Possible values are: `DefaultIndex`, `AuditIndex`or `Partition`
+- `data_forwarding_id` - The ID of the data forwarding configuration to be used by the partition.
+
+[1]: https://help.sumologic.com/docs/manage/partitions/data-tiers/