tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix (clause/expression): Allow sql stmt terminator

Browse Source 
Allow the sql stmt terminator ";" at the end of a named parameter.

Example: select * from table_name where name == @name;
This commit is contained in:
Jim 2021-09-10 07:28:40 -04:00
parent 04f049c1da
commit ab75c5557d
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clause/expression.go
View File

@ -121,7 +121,7 @@ func (expr NamedExpr) Build(builder Builder) {
if v == '@' && !inName { if v == '@' && !inName {
inName = true inName = true
name = []byte{} name = []byte{}
} else if v == ' ' || v == ',' || v == ')' || v == '"' || v == '\'' || v == '`' || v == '\n' { } else if v == ' ' || v == ',' || v == ')' || v == '"' || v == '\'' || v == '`' || v == '\n' || v == ';' {
if inName { if inName {
if nv, ok := namedMap[string(name)]; ok { if nv, ok := namedMap[string(name)]; ok {
builder.AddVar(builder, nv) builder.AddVar(builder, nv)

5
clause/expression_test.go
View File

@ -89,6 +89,11 @@ func TestNamedExpr(t *testing.T) {
SQL: "create table ? (? ?, ? ?)", SQL: "create table ? (? ?, ? ?)",
Vars: []interface{}{}, Vars: []interface{}{},
Result: "create table ? (? ?, ? ?)", Result: "create table ? (? ?, ? ?)",
}, {
SQL: "name1 = @name AND name2 = @name;",
Vars: []interface{}{sql.Named("name", "jinzhu")},
Result: "name1 = ? AND name2 = ?;",
ExpectedVars: []interface{}{"jinzhu", "jinzhu"},
}} }}
for idx, result := range results { for idx, result := range results {