diff --git a/internal/compiler/resolve.go b/internal/compiler/resolve.go index 57e3fbb8df..803cfcb0e3 100644 --- a/internal/compiler/resolve.go +++ b/internal/compiler/resolve.go @@ -193,6 +193,14 @@ func resolveCatalogRefs(c *catalog.Catalog, rvs []*pg.RangeVar, args []paramRef, if inode.Number != ref.ref.Number { continue } + case *pg.TypeCast: + pr, ok := inode.Arg.(*pg.ParamRef) + if !ok { + continue + } + if pr.Number != ref.ref.Number { + continue + } case *pg.NamedArgExpr: pr, ok := inode.Arg.(*pg.ParamRef) if !ok { diff --git a/internal/endtoend/testdata/func_args_typecast/endtoend.json b/internal/endtoend/testdata/func_args_typecast/endtoend.json new file mode 100644 index 0000000000..e9665ca4a3 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/endtoend.json @@ -0,0 +1,3 @@ +{ + "experimental_parser_only": true +} diff --git a/internal/endtoend/testdata/func_args_typecast/go/db.go b/internal/endtoend/testdata/func_args_typecast/go/db.go new file mode 100644 index 0000000000..6a99519302 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/go/db.go @@ -0,0 +1,29 @@ +// Code generated by sqlc. DO NOT EDIT. + +package querytest + +import ( + "context" + "database/sql" +) + +type DBTX interface { + ExecContext(context.Context, string, ...interface{}) (sql.Result, error) + PrepareContext(context.Context, string) (*sql.Stmt, error) + QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error) + QueryRowContext(context.Context, string, ...interface{}) *sql.Row +} + +func New(db DBTX) *Queries { + return &Queries{db: db} +} + +type Queries struct { + db DBTX +} + +func (q *Queries) WithTx(tx *sql.Tx) *Queries { + return &Queries{ + db: tx, + } +} diff --git a/internal/endtoend/testdata/func_args_typecast/go/models.go b/internal/endtoend/testdata/func_args_typecast/go/models.go new file mode 100644 index 0000000000..4e2b892600 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/go/models.go @@ -0,0 +1,5 @@ +// Code generated by sqlc. DO NOT EDIT. + +package querytest + +import () diff --git a/internal/endtoend/testdata/func_args_typecast/go/query.sql.go b/internal/endtoend/testdata/func_args_typecast/go/query.sql.go new file mode 100644 index 0000000000..8abb20f8e5 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/go/query.sql.go @@ -0,0 +1,24 @@ +// Code generated by sqlc. DO NOT EDIT. +// source: query.sql + +package querytest + +import ( + "context" +) + +const plusPositionalCast = `-- name: PlusPositionalCast :one +SELECT plus($1, $2::INTEGER) +` + +type PlusPositionalCastParams struct { + A int32 + Column2 int32 +} + +func (q *Queries) PlusPositionalCast(ctx context.Context, arg PlusPositionalCastParams) (int32, error) { + row := q.db.QueryRowContext(ctx, plusPositionalCast, arg.A, arg.Column2) + var plus int32 + err := row.Scan(&plus) + return plus, err +} diff --git a/internal/endtoend/testdata/func_args_typecast/query.sql b/internal/endtoend/testdata/func_args_typecast/query.sql new file mode 100644 index 0000000000..399a5de922 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/query.sql @@ -0,0 +1,8 @@ +CREATE FUNCTION plus(a integer, b integer) RETURNS integer AS $$ + BEGIN + RETURN a + b; + END; +$$ LANGUAGE plpgsql; + +-- name: PlusPositionalCast :one +SELECT plus($1, $2::INTEGER); diff --git a/internal/endtoend/testdata/func_args_typecast/sqlc.json b/internal/endtoend/testdata/func_args_typecast/sqlc.json new file mode 100644 index 0000000000..ac7c2ed829 --- /dev/null +++ b/internal/endtoend/testdata/func_args_typecast/sqlc.json @@ -0,0 +1,11 @@ +{ + "version": "1", + "packages": [ + { + "path": "go", + "name": "querytest", + "schema": "query.sql", + "queries": "query.sql" + } + ] +}