Support REPLACE and IGNORE in MySQL :copyfrom Queries

internal/cmd/shim.go

@@ -161,6 +161,8 @@ func pluginQueries(r *compiler.Result) []*plugin.Query {
 			Params:          params,
 			Filename:        q.Metadata.Filename,
 			InsertIntoTable: iit,
+			IsReplace:       q.IsReplace,
+			IgnoreErr:       q.IgnoreErr,
 		})
 	}
 	return out

internal/codegen/golang/query.go

@@ -266,7 +266,9 @@ type Query struct {
 	Ret          QueryValue
 	Arg          QueryValue
 	// Used for :copyfrom
-	Table *plugin.Identifier
+	Table     *plugin.Identifier
+	IsReplace bool
+	IgnoreErr bool
 }
 
 func (q Query) hasRetType() bool {

internal/codegen/golang/result.go

@@ -223,6 +223,8 @@ func buildQueries(req *plugin.GenerateRequest, options *opts.Options, structs []
 			SQL:          query.Text,
 			Comments:     comments,
 			Table:        query.InsertIntoTable,
+			IsReplace:    query.IsReplace,
+			IgnoreErr:    query.IgnoreErr,
 		}
 		sqlpkg := parseDriver(options.SqlPackage)
 

internal/codegen/golang/templates/go-sql-driver-mysql/copyfromCopy.tmpl

@@ -40,7 +40,7 @@ func (q *Queries) {{.MethodName}}(ctx context.Context{{if $.EmitMethodsWithDBArg
 	go convertRowsFor{{.MethodName}}(pw, {{.Arg.Name}})
 	// The string interpolation is necessary because LOAD DATA INFILE requires
 	// the file name to be given as a literal string.
-	result, err := {{if (not $.EmitMethodsWithDBArgument)}}q.{{end}}db.ExecContext(ctx, fmt.Sprintf("LOAD DATA LOCAL INFILE '%s' INTO TABLE {{.TableIdentifierForMySQL}} %s ({{range $index, $name := .Arg.ColumnNames}}{{if gt $index 0}}, {{end}}{{$name}}{{end}})", "Reader::" + rh, mysqltsv.Escaping))
+	result, err := {{if (not $.EmitMethodsWithDBArgument)}}q.{{end}}db.ExecContext(ctx, fmt.Sprintf("LOAD DATA LOCAL INFILE '%s' {{if .IsReplace}}REPLACE {{else if .IgnoreErr}}IGNORE {{end}}INTO TABLE {{.TableIdentifierForMySQL}} %s ({{range $index, $name := .Arg.ColumnNames}}{{if gt $index 0}}, {{end}}{{$name}}{{end}})", "Reader::" + rh, mysqltsv.Escaping))
 	if err != nil {
 		return 0, err
 	}

internal/compiler/analyze.go

@@ -14,6 +14,8 @@ import (
 
 type analysis struct {
 	Table      *ast.TableName
+	IsReplace  bool
+	IgnoreErr  bool
 	Columns    []*Column
 	Parameters []Parameter
 	Named      *named.ParamSet
@@ -142,6 +144,8 @@ func (c *Compiler) _analyzeQuery(raw *ast.RawStmt, query string, failfast bool)
 	raw, namedParams, edits := rewrite.NamedParameters(c.conf.Engine, raw, numbers, dollar)
 
 	var table *ast.TableName
+	var isReplace bool
+	var ignoreErr bool
 	switch n := raw.Stmt.(type) {
 	case *ast.InsertStmt:
 		if err := check(validate.InsertStmt(n)); err != nil {
@@ -152,6 +156,8 @@ func (c *Compiler) _analyzeQuery(raw *ast.RawStmt, query string, failfast bool)
 		if err := check(err); err != nil {
 			return nil, err
 		}
+		isReplace = n.IsReplace
+		ignoreErr = n.IgnoreErr
 	}
 
 	if err := check(validate.FuncCall(c.catalog, c.combo, raw)); err != nil {
@@ -207,6 +213,8 @@ func (c *Compiler) _analyzeQuery(raw *ast.RawStmt, query string, failfast bool)
 
 	return &analysis{
 		Table:      table,
+		IsReplace:  isReplace,
+		IgnoreErr:  ignoreErr,
 		Columns:    cols,
 		Parameters: params,
 		Query:      expanded,

internal/compiler/parse.go

@@ -178,6 +178,8 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 		Columns:         anlys.Columns,
 		SQL:             trimmed,
 		InsertIntoTable: anlys.Table,
+		IsReplace:       anlys.IsReplace,
+		IgnoreErr:       anlys.IgnoreErr,
 	}, nil
 }
 

internal/compiler/query.go

@@ -50,6 +50,8 @@ type Query struct {
 
 	// Needed for CopyFrom
 	InsertIntoTable *ast.TableName
+	IsReplace       bool
+	IgnoreErr       bool
 
 	// Needed for vet
 	RawStmt *ast.RawStmt

internal/endtoend/testdata/mysql_copyfrom_replace/query.sql

@@ -0,0 +1,7 @@
+-- name: UpsertLocations :copyfrom
+REPLACE INTO locations (id, name, address, latitude, longitude)
+VALUES (?, ?, ?, ?, ?);
+
+-- name: IgnoreLocations :copyfrom
+INSERT IGNORE INTO locations (id, name, address, latitude, longitude)
+VALUES (?, ?, ?, ?, ?);

internal/endtoend/testdata/mysql_copyfrom_replace/schema.sql

@@ -0,0 +1,7 @@
+CREATE TABLE locations (
+  id        VARCHAR(512) PRIMARY KEY,
+  name      TEXT NOT NULL,
+  address   TEXT NOT NULL,
+  latitude  FLOAT,
+  longitude FLOAT
+);

internal/endtoend/testdata/mysql_copyfrom_replace/sqlc.yaml

@@ -0,0 +1,11 @@
+version: "2"
+sql:
+  - engine: "mysql"
+    queries: "query.sql"
+    schema: "schema.sql"
+    gen:
+      go:
+        package: "db"
+        out: "db"
+        sql_package: "database/sql"
+        sql_driver: "github.com/go-sql-driver/mysql"

internal/engine/dolphin/convert.go

@@ -493,6 +493,8 @@ func (c *cc) convertInsertStmt(n *pcast.InsertStmt) *ast.InsertStmt {
 		Relation:      rangeVar,
 		Cols:          c.convertColumnNames(n.Columns),
 		ReturningList: &ast.List{},
+		IsReplace:     n.IsReplace,
+		IgnoreErr:     n.IgnoreErr,
 	}
 	if ss, ok := c.convert(n.Select).(*ast.SelectStmt); ok {
 		ss.ValuesLists = c.convertLists(n.Lists)

internal/sql/ast/insert_stmt.go

@@ -12,6 +12,8 @@ type InsertStmt struct {
 	WithClause           *WithClause
 	Override             OverridingKind
 	DefaultValues        bool // SQLite-specific: INSERT INTO ... DEFAULT VALUES
+	IsReplace            bool // MySQL-specific
+	IgnoreErr            bool // MySQL-specific
 }
 
 func (n *InsertStmt) Pos() int {