diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/db.go')
| -rw-r--r-- | vendor/github.com/uptrace/bun/db.go | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/vendor/github.com/uptrace/bun/db.go b/vendor/github.com/uptrace/bun/db.go index 97c3f2dfa..72fe118a9 100644 --- a/vendor/github.com/uptrace/bun/db.go +++ b/vendor/github.com/uptrace/bun/db.go @@ -163,7 +163,14 @@ func (db *DB) ScanRow(ctx context.Context, rows *sql.Rows, dest ...interface{}) return rs.ScanRow(ctx, rows) } +type queryHookIniter interface { + Init(db *DB) +} + func (db *DB) AddQueryHook(hook QueryHook) { + if initer, ok := hook.(queryHookIniter); ok { + initer.Init(db) + } db.queryHooks = append(db.queryHooks, hook) } |