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

DryRun for RowQuery, Exec, close #3106

Browse Source
This commit is contained in:
Jinzhu 2020-07-04 08:33:10 +08:00
parent 90a40361ed
commit 89ea62077d
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
callbacks/raw.go
View File

@ -5,7 +5,7 @@ import (
) )
func RawExec(db *gorm.DB) { func RawExec(db *gorm.DB) {
if db.Error == nil { if db.Error == nil && !db.DryRun {
result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...) result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
if err != nil { if err != nil {
db.AddError(err) db.AddError(err)

10
callbacks/row.go
View File

@ -10,10 +10,12 @@ func RowQuery(db *gorm.DB) {
BuildQuerySQL(db) BuildQuerySQL(db)
} }
if _, ok := db.Get("rows"); ok { if !db.DryRun {
db.Statement.Dest, db.Error = db.Statement.ConnPool.QueryContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...) if _, ok := db.Get("rows"); ok {
} else { db.Statement.Dest, db.Error = db.Statement.ConnPool.QueryContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
db.Statement.Dest = db.Statement.ConnPool.QueryRowContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...) } else {
db.Statement.Dest = db.Statement.ConnPool.QueryRowContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)
}
} }
} }
} }