feat(mysql): Support json.RawMessage for LOAD DATA INFILE (#3099)

sqlc-dev · Mar 28, 2024 · d227d5f · d227d5f
1 parent ea9f7ed
commit d227d5f
Show file tree

Hide file tree

Showing 5 changed files with 66 additions and 1 deletion.
diff --git a/internal/codegen/golang/templates/go-sql-driver-mysql/copyfromCopy.tmpl b/internal/codegen/golang/templates/go-sql-driver-mysql/copyfromCopy.tmpl
@@ -10,7 +10,7 @@ func convertRowsFor{{.MethodName}}(w *io.PipeWriter, {{.Arg.SlicePair}}) {
 {{- range $arg.CopyFromMySQLFields}}
 {{- if eq .Type "string"}}
 	e.AppendString({{if eq (len $arg.CopyFromMySQLFields) 1}}row{{else}}row.{{.Name}}{{end}})
-{{- else if eq .Type "[]byte"}}
+{{- else if or (eq .Type "[]byte") (eq .Type "json.RawMessage")}}
 	e.AppendBytes({{if eq (len $arg.CopyFromMySQLFields) 1}}row{{else}}row.{{.Name}}{{end}})
 {{- else}}
 	e.AppendValue({{if eq (len $arg.CopyFromMySQLFields) 1}}row{{else}}row.{{.Name}}{{end}})

diff --git a/internal/endtoend/testdata/json/mysql/go/copyfrom.go b/internal/endtoend/testdata/json/mysql/go/copyfrom.go
diff --git a/internal/endtoend/testdata/json/mysql/go/query.sql.go b/internal/endtoend/testdata/json/mysql/go/query.sql.go
diff --git a/internal/endtoend/testdata/json/mysql/query.sql b/internal/endtoend/testdata/json/mysql/query.sql
@@ -1,2 +1,5 @@
 -- name: SelectFoo :exec
 SELECT * FROM foo;
+
+-- name: BulkInsert :copyfrom
+INSERT INTO foo (a, b) VALUES (?, ?);
diff --git a/internal/endtoend/testdata/json/mysql/sqlc.json b/internal/endtoend/testdata/json/mysql/sqlc.json
@@ -4,6 +4,7 @@
     {
       "path": "go",
       "engine": "mysql",
+      "sql_driver": "github.com/go-sql-driver/mysql",
       "name": "querytest",
       "schema": "schema.sql",
       "queries": "query.sql"