Import sumologic_ingest_budget_v2 by ID or by name

Author: ErikAtSumo

CHANGELOG.md

@@ -10,6 +10,7 @@ BUG FIXES:
 
 ENHANCEMENTS:
 * Improve error message when an API URL is constructed with missing parameters
+* sumologic_ingest_budget_v2 can be imported by ID or by name
 
 ## 3.0.8 (April 1, 2025)
 

sumologic/resource_sumologic_ingest_budget_v2.go

@@ -1,17 +1,7 @@
-// ----------------------------------------------------------------------------
-//
-//	***     AUTO GENERATED CODE    ***    AUTO GENERATED CODE     ***
-//
-// ----------------------------------------------------------------------------
-//
-//	This file is automatically generated by Sumo Logic and manual
-//	changes will be clobbered when the file is regenerated. Do not submit
-//	changes to this file.
-//
-// ----------------------------------------------------------------------------
 package sumologic
 
 import (
+	"fmt"
 	"log"
 
 	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
@@ -24,7 +14,7 @@ func resourceSumologicIngestBudgetV2() *schema.Resource {
 		Update: resourceSumologicIngestBudgetV2Update,
 		Delete: resourceSumologicIngestBudgetV2Delete,
 		Importer: &schema.ResourceImporter{
-			State: schema.ImportStatePassthrough,
+			State: resourceSumologicIngestBudgetV2Import,
 		},
 
 		Schema: map[string]*schema.Schema{
@@ -149,6 +139,28 @@ func resourceSumologicIngestBudgetV2Update(d *schema.ResourceData, meta interfac
 	return resourceSumologicIngestBudgetV2Read(d, meta)
 }
 
+func resourceSumologicIngestBudgetV2Import(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
+	c := meta.(*Client)
+
+	identifier := d.Id()
+
+	// Try to find by ID first
+	ingestBudgetV2, err1 := c.GetIngestBudgetV2(identifier)
+	if err1 == nil && ingestBudgetV2 != nil {
+		d.SetId(ingestBudgetV2.ID)
+		return []*schema.ResourceData{d}, nil
+	}
+
+	// If not found by ID, try to find by name
+	ingestBudgetV2, err2 := c.FindIngestBudgetV2(identifier)
+	if err2 == nil && ingestBudgetV2 != nil {
+		d.SetId(ingestBudgetV2.ID)
+		return []*schema.ResourceData{d}, nil
+	}
+
+	return nil, fmt.Errorf("failed to find IngestBudgetV2: by ID error: %v, by name error: %v", err1, err2)
+}
+
 func resourceToIngestBudgetV2(d *schema.ResourceData) IngestBudgetV2 {
 
 	return IngestBudgetV2{

sumologic/resource_sumologic_ingest_budget_v2_test.go

@@ -1,14 +1,3 @@
-// ----------------------------------------------------------------------------
-//
-//	***     AUTO GENERATED CODE    ***    AUTO GENERATED CODE     ***
-//
-// ----------------------------------------------------------------------------
-//
-//	This file is automatically generated by Sumo Logic and manual
-//	changes will be clobbered when the file is regenerated. Do not submit
-//	changes to this file.
-//
-// ----------------------------------------------------------------------------
 package sumologic
 
 import (
@@ -46,6 +35,12 @@ func TestAccSumologicIngestBudgetV2_basic(t *testing.T) {
 				ImportState:       true,
 				ImportStateVerify: true,
 			},
+			{
+				ResourceName:      "sumologic_ingest_budget_v2.foo",
+				ImportState:       true,
+				ImportStateVerify: true,
+				ImportStateId:     testName,
+			},
 		},
 	})
 }

sumologic/sumologic_ingest_budget_v2.go

@@ -1,14 +1,3 @@
-// ----------------------------------------------------------------------------
-//
-//	***     AUTO GENERATED CODE    ***    AUTO GENERATED CODE     ***
-//
-// ----------------------------------------------------------------------------
-//
-//	This file is automatically generated by Sumo Logic and manual
-//	changes will be clobbered when the file is regenerated. Do not submit
-//	changes to this file.
-//
-// ----------------------------------------------------------------------------
 package sumologic
 
 import (
@@ -91,6 +80,47 @@ func (s *Client) UpdateIngestBudgetV2(ingestBudgetV2 IngestBudgetV2) error {
 
 }
 
+func (s *Client) FindIngestBudgetV2(name string) (*IngestBudgetV2, error) {
+	type IngestBudgetV2List struct {
+		Next string           `json:"next"`
+		Data []IngestBudgetV2 `json:"data"`
+	}
+
+	next := ""
+
+	for {
+		body, err := s.Get(fmt.Sprintf("v2/ingestBudgets?token=%s", next))
+		if err != nil {
+			return nil, err
+		}
+
+		if body == nil {
+			return nil, fmt.Errorf("received empty response while searching for ingest budget '%s'", name)
+		}
+
+		var response IngestBudgetV2List
+
+		err = json.Unmarshal(body, &response)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, budgetV2 := range response.Data {
+			if budgetV2.Name == name {
+				return &budgetV2, nil
+			}
+		}
+
+		if response.Next == "" {
+			break
+		}
+
+		next = response.Next
+	}
+
+	return nil, fmt.Errorf("unable to find ingest budget '%s'", name)
+}
+
 type IngestBudgetV2 struct {
 	AuditThreshold int    `json:"auditThreshold,omitempty"`
 	Action         string `json:"action"`

website/docs/r/ingest_budget_v2.html.markdown

@@ -42,10 +42,11 @@ The following attributes are exported:
   * `id` - The internal ID of the ingest budget. 
 
 ## Import
-Ingest budgets can be imported using the budget ID, e.g.:
+Ingest budgets can be imported using the budget ID or name, e.g.:
 
 ```hcl
 terraform import sumologic_ingest_budget_v2.budget 00000000000123AB
+terraform import sumologic_ingest_budget_v2.dev_budget "DevelopersBudget"
 ```
 
 [1]: https://help.sumologic.com/Beta/Metadata_Ingest_Budgets