fix(codegen): Correct column names in :copyfrom (#2838)

* fix(codegen): Correct column names in :copyfrom

* Disable BuildKite reporting

Author: kyleconroy

.github/workflows/ci.yml

@@ -72,7 +72,7 @@ jobs:
       working-directory: internal/endtoend/testdata
 
     - name: report
-      if: ${{ github.ref == 'refs/heads/main' }}
+      if: false
       run: ./scripts/report.sh
       env:
         BUILDKITE_ANALYTICS_TOKEN: ${{ secrets.BUILDKITE_ANALYTICS_TOKEN }}
@@ -87,4 +87,4 @@ jobs:
       with:
         go-version: '1.21.3'
     - run: go install golang.org/x/vuln/cmd/govulncheck@latest
-    - run: govulncheck ./...
+    - run: govulncheck ./...

examples/batch/sqlc.json

@@ -13,6 +13,9 @@
       "database": {
         "managed": true
       },
+      "analyzer": {
+        "database": false
+      },
       "rules": [
         "sqlc/db-prepare"
       ],
@@ -22,4 +25,4 @@
       "emit_interface": true
     }
   ]
-}
+}

internal/codegen/golang/query.go

@@ -166,7 +166,11 @@ func (v QueryValue) ColumnNamesAsGoSlice() string {
 	}
 	escapedNames := make([]string, len(v.Struct.Fields))
 	for i, f := range v.Struct.Fields {
-		escapedNames[i] = fmt.Sprintf("%q", f.DBName)
+		if f.Column != nil && f.Column.OriginalName != "" {
+			escapedNames[i] = fmt.Sprintf("%q", f.Column.OriginalName)
+		} else {
+			escapedNames[i] = fmt.Sprintf("%q", f.DBName)
+		}
 	}
 	return "[]string{" + strings.Join(escapedNames, ", ") + "}"
 }

internal/config/v_one.go

@@ -21,7 +21,7 @@ type v1PackageSettings struct {
 	Name                      string     `json:"name" yaml:"name"`
 	Engine                    Engine     `json:"engine,omitempty" yaml:"engine"`
 	Database                  *Database  `json:"database,omitempty" yaml:"database"`
-  Analyzer                  Analyzer   `json:"analyzer" yaml:"analyzer"`
+	Analyzer                  Analyzer   `json:"analyzer" yaml:"analyzer"`
 	Path                      string     `json:"path" yaml:"path"`
 	Schema                    Paths      `json:"schema" yaml:"schema"`
 	Queries                   Paths      `json:"queries" yaml:"queries"`
@@ -148,6 +148,7 @@ func (c *V1GenerateSettings) Translate() Config {
 			Schema:   pkg.Schema,
 			Queries:  pkg.Queries,
 			Rules:    pkg.Rules,
+			Analyzer: pkg.Analyzer,
 			Gen: SQLGen{
 				Go: &SQLGo{
 					EmitInterface:             pkg.EmitInterface,

internal/endtoend/endtoend_test.go

@@ -21,7 +21,7 @@ import (
 )
 
 func TestExamples(t *testing.T) {
-	// t.Parallel()
+	t.Parallel()
 	ctx := context.Background()
 
 	examples, err := filepath.Abs(filepath.Join("..", "..", "examples"))
@@ -40,7 +40,7 @@ func TestExamples(t *testing.T) {
 		}
 		tc := replay.Name()
 		t.Run(tc, func(t *testing.T) {
-			// t.Parallel()
+			t.Parallel()
 			path := filepath.Join(examples, tc)
 			var stderr bytes.Buffer
 			opts := &cmd.Options{

internal/endtoend/testdata/copyfrom_named_params/issue.md

@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/2833

internal/endtoend/testdata/copyfrom_named_params/postgresql/pgx/go/copyfrom.go

@@ -0,0 +1,6 @@
+-- name: StageUserData :copyfrom
+insert into "user_data" ("id", "user")
+values (
+    sqlc.arg('id_param'),
+    sqlc.arg('user_param')
+);

internal/endtoend/testdata/copyfrom_named_params/postgresql/pgx/go/db.go

@@ -0,0 +1,5 @@
+create table "user_data" (
+	"id" varchar not null,
+	"user" varchar not null,
+	primary key ("id")
+);

internal/endtoend/testdata/copyfrom_named_params/postgresql/pgx/go/models.go

@@ -0,0 +1,10 @@
+version: "2"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"

internal/endtoend/testdata/copyfrom_named_params/postgresql/pgx/go/query.sql.go

@@ -39,7 +39,10 @@ func validateCopyfrom(n ast.Node) error {
 		return nil
 	}
 	for _, v := range sublist.Items {
-		if _, ok := v.(*ast.ParamRef); !ok {
+		_, ok := v.(*ast.ParamRef)
+		ok = ok || named.IsParamFunc(v)
+		ok = ok || named.IsParamSign(v)
+		if !ok {
 			return errors.New(":copyfrom doesn't support non-parameter values")
 		}
 	}