Add data source sumologic_partition

Author: ErikAtSumo

CHANGELOG.md

@@ -1,6 +1,9 @@
 ## X.Y.Z (Unreleased)
 * Add new change notes here
 
+FEATURES:
+* **New Data Source:** sumologic_partition
+
 ENHANCEMENTS:
 * Add support for searchable time for dashboard field time_source
 

sumologic/data_source_sumologic_partition.go

@@ -0,0 +1,104 @@
+package sumologic
+
+import (
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+)
+
+func dataSourceSumologicPartition() *schema.Resource {
+	return &schema.Resource{
+		Read:   dataSourceSumologicPartitionRead,
+		Schema: dataSourcePartitionSchema(),
+	}
+}
+
+func dataSourcePartitionSchema() map[string]*schema.Schema {
+	return map[string]*schema.Schema{
+		"id": {
+			Type:     schema.TypeString,
+			Required: true,
+		},
+		"name": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"routing_expression": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"analytics_tier": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"retention_period": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+		"is_compliant": {
+			Type:     schema.TypeBool,
+			Computed: true,
+		},
+		"total_bytes": {
+			Type:     schema.TypeInt,
+			Computed: true,
+		},
+		"data_forwarding_id": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"is_active": {
+			Type:     schema.TypeBool,
+			Computed: true,
+		},
+		"index_type": {
+			Type:     schema.TypeString,
+			Computed: true,
+		},
+		"reduce_retention_period_immediately": {
+			Type:     schema.TypeBool,
+			Computed: true,
+		},
+		"is_included_in_default_search": {
+			Type:     schema.TypeBool,
+			Computed: true,
+		},
+	}
+}
+
+func dataSourceSumologicPartitionRead(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	var err error
+	var spartition *Partition
+
+	rid, ok := d.GetOk("id")
+
+	if !ok {
+		return fmt.Errorf("please specify the partition id")
+	}
+
+	id := rid.(string)
+	spartition, err = c.GetPartition(id)
+	if err != nil {
+		return fmt.Errorf("error retrieving partition with id %v: %v", id, err)
+	}
+	if spartition == nil {
+		return fmt.Errorf("partition with id %v not found", id)
+	}
+
+	d.SetId(spartition.ID)
+	d.Set("routing_expression", spartition.RoutingExpression)
+	d.Set("name", spartition.Name)
+	d.Set("analytics_tier", spartition.AnalyticsTier)
+	d.Set("retention_period", spartition.RetentionPeriod)
+	d.Set("is_compliant", spartition.IsCompliant)
+	d.Set("data_forwarding_id", spartition.DataForwardingId)
+	d.Set("is_active", spartition.IsActive)
+	d.Set("total_bytes", spartition.TotalBytes)
+	d.Set("index_type", spartition.IndexType)
+	d.Set("reduce_retention_period_immediately", spartition.ReduceRetentionPeriodImmediately)
+	d.Set("is_included_in_default_search", spartition.IsIncludedInDefaultSearch)
+
+	return nil
+}

sumologic/data_source_sumologic_partition_test.go

@@ -0,0 +1,45 @@
+package sumologic
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+)
+
+func TestAccDataSourcePartition_basic(t *testing.T) {
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testDataSourceSumologicPartitionConfig,
+				Check: resource.ComposeTestCheckFunc(
+					testAccDataSourcePartitionCheck("data.sumologic_partition.test"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourcePartitionCheck(name string) resource.TestCheckFunc {
+	return resource.ComposeTestCheckFunc(
+		resource.TestCheckResourceAttr(name, "id", "00000000000AD88D"),
+		resource.TestCheckResourceAttr(name, "name", "apache"),
+		resource.TestCheckResourceAttr(name, "routing_expression", "_sourcecategory=*/Apache"),
+		resource.TestCheckResourceAttr(name, "analytics_tier", "continuous"),
+		resource.TestCheckResourceAttr(name, "retention_period", "365"),
+		resource.TestCheckResourceAttr(name, "is_compliant", "false"),
+		resource.TestCheckResourceAttr(name, "is_active", "true"),
+		resource.TestCheckResourceAttr(name, "total_bytes", "0"),
+		resource.TestCheckResourceAttr(name, "data_forwarding_id", ""),
+		resource.TestCheckResourceAttr(name, "index_type", "Partition"),
+		resource.TestCheckResourceAttr(name, "reduce_retention_period_immediately", "false"),
+		resource.TestCheckResourceAttr(name, "is_included_in_default_search", "true"),
+	)
+}
+
+var testDataSourceSumologicPartitionConfig = `
+	data "sumologic_partition" "test" {
+		id = "00000000000AD88D"
+	}
+`

sumologic/provider.go

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

website/docs/d/partition.html.markdown

@@ -0,0 +1,39 @@
+---
+layout: "sumologic"
+page_title: "SumoLogic: sumologic_partition"
+description: |-
+  Provides a way to retrieve Sumo Logic partition details (id, name, etc) for a partition managed outside of terraform.
+---
+
+# sumologic_partition
+
+Provides a way to retrieve a [Sumologic Partition][1]'s details (id, name, etc) for a partition
+managed by another terraform stack.
+
+
+## Example Usage
+
+```hcl
+data "sumologic_partition" "this" {
+  id = "1234567890"
+}
+```
+
+A partition can be looked up by `id`.
+
+## 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/