Export generated query for telemetry usage (#1054)

* add emit_exported_queries to config

* change capitalization of query var name based on EmitExportedQueries

* create tests for emit exported queries on pgx and stdlib drivers

Author: Leanndrosouza

internal/codegen/golang/result.go

@@ -145,9 +145,16 @@ func buildQueries(r *compiler.Result, settings config.CombinedSettings, structs
 			continue
 		}
 
+		var constantName string
+		if settings.Go.EmitExportedQueries {
+			constantName = codegen.Title(query.Name)
+		} else {
+			constantName = codegen.LowerTitle(query.Name)
+		}
+
 		gq := Query{
 			Cmd:          query.Cmd,
-			ConstantName: codegen.LowerTitle(query.Name),
+			ConstantName: constantName,
 			FieldName:    codegen.LowerTitle(query.Name) + "Stmt",
 			MethodName:   query.Name,
 			SourceName:   query.Filename,

internal/codegen/utils.go

@@ -11,6 +11,10 @@ func LowerTitle(s string) string {
 	return string(a)
 }
 
+func Title(s string) string {
+	return strings.Title(s)
+}
+
 // Go string literals cannot contain backtick. If a string contains
 // a backtick, replace it the following way:
 //

internal/config/config.go

@@ -116,6 +116,7 @@ type SQLGo struct {
 	EmitPreparedQueries   bool              `json:"emit_prepared_queries" yaml:"emit_prepared_queries"`
 	EmitExactTableNames   bool              `json:"emit_exact_table_names,omitempty" yaml:"emit_exact_table_names"`
 	EmitEmptySlices       bool              `json:"emit_empty_slices,omitempty" yaml:"emit_empty_slices"`
+	EmitExportedQueries   bool              `json:"emit_exported_queries" yaml:"emit_exported_queries"`
 	JSONTagsCaseStyle     string            `json:"json_tags_case_style,omitempty" yaml:"json_tags_case_style"`
 	Package               string            `json:"package" yaml:"package"`
 	Out                   string            `json:"out" yaml:"out"`

internal/config/v_one.go

@@ -27,6 +27,7 @@ type v1PackageSettings struct {
 	EmitPreparedQueries   bool       `json:"emit_prepared_queries" yaml:"emit_prepared_queries"`
 	EmitExactTableNames   bool       `json:"emit_exact_table_names,omitempty" yaml:"emit_exact_table_names"`
 	EmitEmptySlices       bool       `json:"emit_empty_slices,omitempty" yaml:"emit_empty_slices"`
+	EmitExportedQueries   bool       `json:"emit_exported_queries,omitempty" yaml:"emit_exported_queries"`
 	JSONTagsCaseStyle     string     `json:"json_tags_case_style,omitempty" yaml:"json_tags_case_style"`
 	Driver                string     `json:"driver" yaml:"driver"`
 	Overrides             []Override `json:"overrides" yaml:"overrides"`
@@ -115,6 +116,7 @@ func (c *V1GenerateSettings) Translate() Config {
 					EmitPreparedQueries:   pkg.EmitPreparedQueries,
 					EmitExactTableNames:   pkg.EmitExactTableNames,
 					EmitEmptySlices:       pkg.EmitEmptySlices,
+					EmitExportedQueries:   pkg.EmitExportedQueries,
 					Package:               pkg.Name,
 					Out:                   pkg.Path,
 					Driver:                pkg.Driver,

internal/endtoend/testdata/emit_exported_queries/pgx/go/db.go

@@ -0,0 +1,4 @@
+CREATE TABLE bar (id serial not null);
+
+-- name: UpdateBarID :exec
+UPDATE bar SET id = $1 WHERE id = $2;

internal/endtoend/testdata/emit_exported_queries/pgx/go/models.go

@@ -0,0 +1,14 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "driver": "pgx/v4",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql",
+      "emit_exported_queries": true
+    }
+  ]
+}