GOMODULES
/
pgx
mirror of https://github.com/jackc/pgx.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NamedArgs allows underscore

non-blocking
Jack Christensen 2022-04-25 10:16:47 -05:00
parent c093c4af21
commit d13bdbbd35
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
named_args.go
View File

@ -128,7 +128,7 @@ func namedArgState(l *sqlLexer) stateFn {
l.start = l.pos
}
return nil
} else if !(isLetter(r) || (r >= '0' && r <= '9')) {
} else if !(isLetter(r) || (r >= '0' && r <= '9') || r == '_') {
l.pos -= width
na := namedArg(l.src[l.start:l.pos])
if _, found := l.nameToOrdinal[na]; !found {

6
named_args_test.go
View File

@ -36,6 +36,12 @@ func TestNamedArgsRewriteQuery(t *testing.T) {
expectedSQL: "select $1::int, $2::text",
expectedArgs: []any{int32(42), "foo"},
},
{
sql: "select @Abc::int, @b_4::text",
namedArgs: pgx.NamedArgs{"Abc": int32(42), "b_4": "foo"},
expectedSQL: "select $1::int, $2::text",
expectedArgs: []any{int32(42), "foo"},
},
{
sql: "at end @",
namedArgs: pgx.NamedArgs{"a": int32(42), "b": "foo"},