feat(golang): Add initialisms configuration (#3308)

* feat(golang): Add intialisms configuration

* docs: Add docs for initialisms

* Fix test names

Author: kyleconroy

docs/reference/config.md

@@ -167,6 +167,8 @@ The `gen` mapping supports the following keys:
   - If true, emits the SQL statement as a code-block comment above the generated function, appending to any existing comments. Defaults to `false`.
 - `build_tags`:
   - If set, add a `//go:build <build_tags>` directive at the beginning of each generated Go file.
+- `initialisms`:
+  - An array of [initialisms](https://google.github.io/styleguide/go/decisions.html#initialisms) to upper-case. For example, `app_id` becomes `AppID`. Defaults to `["id"]`.
 - `json_tags_id_uppercase`:
   - If true, "Id" in json tags will be uppercase. If false, will be camelcase. Defaults to `false`
 - `json_tags_case_style`:

internal/codegen/golang/opts/options.go

@@ -43,6 +43,9 @@ type Options struct {
 	OmitSqlcVersion             bool              `json:"omit_sqlc_version,omitempty" yaml:"omit_sqlc_version"`
 	OmitUnusedStructs           bool              `json:"omit_unused_structs,omitempty" yaml:"omit_unused_structs"`
 	BuildTags                   string            `json:"build_tags,omitempty" yaml:"build_tags"`
+	Initialisms                 *[]string         `json:"initialisms,omitempty" yaml:"initialisms"`
+
+	InitialismsMap map[string]struct{} `json:"-" yaml:"-"`
 }
 
 type GlobalOptions struct {
@@ -111,6 +114,16 @@ func parseOpts(req *plugin.GenerateRequest) (*Options, error) {
 		*options.QueryParameterLimit = 1
 	}
 
+	if options.Initialisms == nil {
+		options.Initialisms = new([]string)
+		*options.Initialisms = []string{"id"}
+	}
+
+	options.InitialismsMap = map[string]struct{}{}
+	for _, initial := range *options.Initialisms {
+		options.InitialismsMap[initial] = struct{}{}
+	}
+
 	return &options, nil
 }
 

internal/codegen/golang/struct.go

@@ -32,8 +32,8 @@ func StructName(name string, options *opts.Options) string {
 	}, name)
 
 	for _, p := range strings.Split(name, "_") {
-		if p == "id" {
-			out += "ID"
+		if _, found := options.InitialismsMap[p]; found {
+			out += strings.ToUpper(p)
 		} else {
 			out += strings.Title(p)
 		}

internal/endtoend/testdata/golang_initialisms_empty/db/db.go

@@ -0,0 +1,2 @@
+-- name: SelectFoo :many
+SELECT * FROM foo;

internal/endtoend/testdata/golang_initialisms_empty/db/models.go

@@ -0,0 +1,3 @@
+CREATE TABLE foo(
+        bar_id text
+);

internal/endtoend/testdata/golang_initialisms_empty/db/query.sql.go

@@ -0,0 +1,16 @@
+{
+    "version": "2",
+    "sql": [
+        {
+            "schema": "schema.sql",
+            "queries": "query.sql",
+            "engine": "postgresql",
+            "gen": {
+                "go": {
+                    "out": "db",
+                    "initialisms": []
+                }
+            }
+        }
+    ]
+}

internal/endtoend/testdata/golang_initialisms_empty/query.sql

@@ -0,0 +1,2 @@
+-- name: SelectFoo :many
+SELECT * FROM foo;

internal/endtoend/testdata/golang_initialisms_empty/schema.sql

@@ -0,0 +1,4 @@
+CREATE TABLE foo(
+        bar_id text,
+        site_url text
+);

internal/endtoend/testdata/golang_initialisms_empty/sqlc.json

@@ -0,0 +1,16 @@
+{
+    "version": "2",
+    "sql": [
+        {
+            "schema": "schema.sql",
+            "queries": "query.sql",
+            "engine": "postgresql",
+            "gen": {
+                "go": {
+                    "out": "db",
+                    "initialisms": ["id", "url"]
+                }
+            }
+        }
+    ]
+}