gorm/callbacks/row.go

22 lines
546 B
Go
Raw Normal View History

2020-02-22 14:41:01 +03:00
package callbacks
import (
2020-06-02 04:16:07 +03:00
"gorm.io/gorm"
2020-02-22 14:41:01 +03:00
)
func RowQuery(db *gorm.DB) {
2020-05-31 18:55:56 +03:00
if db.Error == nil {
if db.Statement.SQL.String() == "" {
BuildQuerySQL(db)
}
2020-02-22 14:41:01 +03:00
2020-07-04 03:33:10 +03:00
if !db.DryRun {
if isRows, ok := db.InstanceGet("rows"); ok && isRows.(bool) {
2020-07-04 03:33:10 +03:00
db.Statement.Dest, db.Error = db.Statement.ConnPool.QueryContext(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...)
}
2020-05-31 18:55:56 +03:00
}
2020-02-22 14:41:01 +03:00
}
}