Fix tests

This commit is contained in:
Lunny Xiao 2023-07-24 15:32:35 +08:00
parent 01b903f996
commit 6c9528c6c9
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
3 changed files with 6 additions and 5 deletions

View File

@ -244,7 +244,7 @@ func (statement *Statement) writeSelectColumns(w *builder.BytesWriter, columnStr
} }
func (statement *Statement) writeWhereCond(w *builder.BytesWriter, cond builder.Cond) error { func (statement *Statement) writeWhereCond(w *builder.BytesWriter, cond builder.Cond) error {
if !statement.cond.IsValid() { if !cond.IsValid() {
return nil return nil
} }

View File

@ -427,7 +427,7 @@ func (statement *Statement) writeUpdateLimit(updateWriter *builder.BytesWriter,
} }
} }
func (statement *Statement) WriteUpdate(updateWriter *builder.BytesWriter, cond builder.Cond, colNames []string) error { func (statement *Statement) WriteUpdate(updateWriter *builder.BytesWriter, cond builder.Cond, colNames []string, args []interface{}) error {
if _, err := fmt.Fprintf(updateWriter, "UPDATE"); err != nil { if _, err := fmt.Fprintf(updateWriter, "UPDATE"); err != nil {
return err return err
} }
@ -454,6 +454,7 @@ func (statement *Statement) WriteUpdate(updateWriter *builder.BytesWriter, cond
return err return err
} }
} }
updateWriter.Append(args...)
// write from // write from
if err := statement.writeUpdateFrom(updateWriter); err != nil { if err := statement.writeUpdateFrom(updateWriter); err != nil {

View File

@ -227,14 +227,14 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
} }
updateWriter := builder.NewWriter() updateWriter := builder.NewWriter()
if err := session.statement.WriteUpdate(updateWriter, cond, colNames); err != nil { if err := session.statement.WriteUpdate(updateWriter, cond, colNames, args); err != nil {
return 0, err return 0, err
} }
tableName := session.statement.TableName() // table name must been get before exec because statement will be reset tableName := session.statement.TableName() // table name must been get before exec because statement will be reset
useCache := session.statement.UseCache useCache := session.statement.UseCache
res, err := session.exec(updateWriter.String(), append(args, updateWriter.Args()...)...) res, err := session.exec(updateWriter.String(), updateWriter.Args()...)
if err != nil { if err != nil {
return 0, err return 0, err
} else if doIncVer { } else if doIncVer {