Mix parameters2

internal/compiler/parse.go

@@ -6,6 +6,7 @@ import (
 	"sort"
 	"strings"
 
+	"github.com/kyleconroy/sqlc/internal/config"
 	"github.com/kyleconroy/sqlc/internal/debug"
 	"github.com/kyleconroy/sqlc/internal/metadata"
 	"github.com/kyleconroy/sqlc/internal/opts"
@@ -37,7 +38,8 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 	if err := validate.ParamStyle(stmt); err != nil {
 		return nil, err
 	}
-	if err := validate.ParamRef(stmt); err != nil {
+	numbers, err := validate.ParamRef(stmt)
+	if err != nil {
 		return nil, err
 	}
 	raw, ok := stmt.(*ast.RawStmt)
@@ -75,7 +77,7 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 		return nil, err
 	}
 
-	raw, namedParams, edits := rewrite.NamedParameters(c.conf.Engine, raw)
+	raw, namedParams, edits := rewrite.NamedParameters(c.conf.Engine, raw, numbers)
 	rvs := rangeVars(raw.Stmt)
 	refs := findParameters(raw.Stmt)
 	if o.UsePositionalParameters {
@@ -85,7 +87,11 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 		}
 	} else {
 		refs = uniqueParamRefs(refs)
-		sort.Slice(refs, func(i, j int) bool { return refs[i].ref.Number < refs[j].ref.Number })
+		if c.conf.Engine == config.EngineMySQL {
+			sort.Slice(refs, func(i, j int) bool { return refs[i].ref.Location < refs[j].ref.Location })
+		} else {
+			sort.Slice(refs, func(i, j int) bool { return refs[i].ref.Number < refs[j].ref.Number })
+		}
 	}
 	qc, err := buildQueryCatalog(c.catalog, raw.Stmt)
 	if err != nil {
@@ -122,7 +128,6 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 	if err != nil {
 		return nil, err
 	}
-
 	return &Query{
 		Cmd:      cmd,
 		Comments: comments,

internal/endtoend/testdata/invalid_params/pgx/stderr.txt

@@ -2,4 +2,4 @@
 query.sql:4:1: could not determine data type of parameter $1
 query.sql:7:1: could not determine data type of parameter $2
 query.sql:10:8: column "foo" does not exist
-query.sql:13:1: query mixes positional parameters ($1) and named parameters (sqlc.arg or @arg)
+query.sql:13:1: could not determine data type of parameter $2

internal/endtoend/testdata/invalid_params/stdlib/stderr.txt

@@ -2,4 +2,4 @@
 query.sql:4:1: could not determine data type of parameter $1
 query.sql:7:1: could not determine data type of parameter $2
 query.sql:10:8: column "foo" does not exist
-query.sql:13:1: query mixes positional parameters ($1) and named parameters (sqlc.arg or @arg)
+query.sql:13:1: could not determine data type of parameter $2

internal/endtoend/testdata/mix_param_types/mysql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "name": "querytest",
+      "schema": "test.sql",
+      "queries": "test.sql",
+      "engine": "mysql"
+    }
+  ]
+}

internal/endtoend/testdata/mix_param_types/mysql/test.sql

@@ -0,0 +1,14 @@
+CREATE TABLE bar (
+       id serial not null,
+       name text not null,
+       phone text not null
+);
+
+-- name: CountOne :one
+SELECT count(1) FROM bar WHERE id = sqlc.arg(id) AND name <> ?; 
+
+-- name: CountTwo :one
+SELECT count(1) FROM bar WHERE id = ? AND name <> sqlc.arg(name);
+
+-- name: CountThree :one
+SELECT count(1) FROM bar WHERE id > ? AND phone <> sqlc.arg(phone) AND name <> ?;

internal/endtoend/testdata/mix_param_types/postgresql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "name": "querytest",
+      "schema": "test.sql",
+      "queries": "test.sql",
+      "engine": "postgresql"
+    }
+  ]
+}

internal/endtoend/testdata/mix_param_types/postgresql/test.sql

@@ -0,0 +1,14 @@
+CREATE TABLE bar (
+       id serial not null,
+       name text not null,
+       phone text not null
+);
+
+-- name: CountOne :one
+SELECT count(1) FROM bar WHERE id = sqlc.arg(id) AND name <> $1 LIMIT sqlc.arg('limit');
+
+-- name: CountTwo :one
+SELECT count(1) FROM bar WHERE id = $1 AND name <> sqlc.arg(name);
+
+-- name: CountThree :one
+SELECT count(1) FROM bar WHERE id > $2 AND phone <> sqlc.arg(phone) AND name <> $1;

internal/endtoend/testdata/sqlc_arg_invalid/mysql/stderr.txt

@@ -1,5 +1,5 @@
 # package querytest
 query.sql:7:1: function "sqlc.argh" does not exist
 query.sql:10:45: expected 1 parameter to sqlc.arg; got 2
-query.sql:13:45: expected parameter to sqlc.arg to be string or reference; got *ast.FuncCall
-query.sql:16:1: query mixes positional parameters ($1) and named parameters (sqlc.arg or @arg)
+query.sql:13:54: Invalid argument to sqlc.arg()
+query.sql:16:54: Invalid argument to sqlc.arg()

internal/endtoend/testdata/sqlc_arg_invalid/postgresql/stderr.txt

@@ -1,5 +1,5 @@
 # package querytest
 query.sql:7:1: function "sqlc.argh" does not exist
 query.sql:10:45: expected 1 parameter to sqlc.arg; got 2
-query.sql:13:45: expected parameter to sqlc.arg to be string or reference; got *ast.FuncCall
-query.sql:16:1: query mixes positional parameters ($1) and named parameters (sqlc.arg or @arg)
+query.sql:13:54: Invalid argument to sqlc.arg()
+query.sql:16:54: Invalid argument to sqlc.arg()

internal/sql/rewrite/parameters.go

@@ -41,7 +41,7 @@ func isNamedParamSignCast(node ast.Node) bool {
 	return astutils.Join(expr.Name, ".") == "@" && cast
 }
 
-func NamedParameters(engine config.Engine, raw *ast.RawStmt) (*ast.RawStmt, map[int]string, []source.Edit) {
+func NamedParameters(engine config.Engine, raw *ast.RawStmt, numbs map[int]bool) (*ast.RawStmt, map[int]string, []source.Edit) {
 	foundFunc := astutils.Search(raw, named.IsParamFunc)
 	foundSign := astutils.Search(raw, named.IsParamSign)
 	if len(foundFunc.Items)+len(foundSign.Items) == 0 {
@@ -56,7 +56,6 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt) (*ast.RawStmt, map[
 	node := astutils.Apply(raw, func(cr *astutils.Cursor) bool {
 		node := cr.Node()
 		switch {
-
 		case named.IsParamFunc(node):
 			fun := node.(*ast.FuncCall)
 			param, isConst := flatten(fun.Args)
@@ -66,7 +65,10 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt) (*ast.RawStmt, map[
 					Location: fun.Location,
 				})
 			} else {
-				argn += 1
+				argn++
+				for numbs[argn] {
+					argn++
+				}
 				args[param] = argn
 				cr.Replace(&ast.ParamRef{
 					Number:   argn,
@@ -103,7 +105,10 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt) (*ast.RawStmt, map[
 				}
 				cr.Replace(cast)
 			} else {
-				argn += 1
+				argn++
+				for numbs[argn] {
+					argn++
+				}
 				args[param] = argn
 				cast.Arg = &ast.ParamRef{
 					Number:   argn,
@@ -128,7 +133,10 @@ func NamedParameters(engine config.Engine, raw *ast.RawStmt) (*ast.RawStmt, map[
 					Location: expr.Location,
 				})
 			} else {
-				argn += 1
+				argn++
+				for numbs[argn] {
+					argn++
+				}
 				args[param] = argn
 				cr.Replace(&ast.ParamRef{
 					Number:   argn,