Merge pull request #3992 from hashicorp/recaser

`tools/generator-go-sdk` - add init() in id generator
hashicorp · Apr 12, 2024 · dfcf935 · dfcf935
2 parents 64934f1 + f3a5245
commit dfcf935
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/tools/generator-go-sdk/internal/generator/templater_id_parser.go b/tools/generator-go-sdk/internal/generator/templater_id_parser.go
@@ -41,12 +41,14 @@ import (
     "strings"
 
     "github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids"
+    "github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids/recaser"
 )
 
 %[2]s
 %[3]s
 %[4]s
-`, data.packageName, *copyrightLines, *structBody, *methods)
+%[5]s
+`, data.packageName, *copyrightLines, r.registerId(), *structBody, *methods)
 	return &out, nil
 }
 
@@ -83,6 +85,15 @@ type %[1]s struct {
 	return &out, nil
 }
 
+func (r resourceIdTemplater) registerId() string {
+	wordifiedName := wordifyString(r.name)
+	return fmt.Sprintf(`
+	func init() {
+		recaser.RegisterResourceId(&%[2]s{})
+	}
+`, wordifiedName, r.name)
+}
+
 func (r resourceIdTemplater) methods() (*string, error) {
 	nameWithoutSuffix := strings.TrimSuffix(r.name, "Id")
 

diff --git a/tools/generator-go-sdk/internal/generator/templater_id_parser_test.go b/tools/generator-go-sdk/internal/generator/templater_id_parser_test.go
@@ -39,10 +39,15 @@ import (
     "strings"
 
     "github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids"
+	"github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids/recaser"
 )
 
 // acctests licence placeholder
 
+func init() {
+    recaser.RegisterResourceId(&BasicTestId{})
+}
+
 var _ resourceids.ResourceId = &BasicTestId{}
 
 	// BasicTestId is a struct representing the Resource ID for a Basic Test
@@ -183,10 +188,15 @@ import (
 	"strings"
 
 	"github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids"
+    "github.com/hashicorp/go-azure-helpers/resourcemanager/resourceids/recaser"
 )
 
 // acctests licence placeholder
 
+func init() {
+    recaser.RegisterResourceId(&ConstantOnlyId{})
+}
+
 var _ resourceids.ResourceId = &ConstantOnlyId{}
 
 // ConstantOnlyId is a struct representing the Resource ID for a Constant Only