You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
562 B
23 lines
562 B
package callbacks |
|
|
|
import ( |
|
"gorm.io/gorm" |
|
) |
|
|
|
func RowQuery(db *gorm.DB) { |
|
if db.Error == nil { |
|
BuildQuerySQL(db) |
|
if db.DryRun { |
|
return |
|
} |
|
|
|
if isRows, ok := db.Get("rows"); ok && isRows.(bool) { |
|
db.Statement.Settings.Delete("rows") |
|
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...) |
|
} |
|
|
|
db.RowsAffected = -1 |
|
} |
|
}
|
|
|