diff options
Diffstat (limited to 'vendor/github.com/uptrace/bun/hook.go')
-rw-r--r-- | vendor/github.com/uptrace/bun/hook.go | 116 |
1 files changed, 0 insertions, 116 deletions
diff --git a/vendor/github.com/uptrace/bun/hook.go b/vendor/github.com/uptrace/bun/hook.go deleted file mode 100644 index 016f06a1a..000000000 --- a/vendor/github.com/uptrace/bun/hook.go +++ /dev/null @@ -1,116 +0,0 @@ -package bun - -import ( - "context" - "database/sql" - "strings" - "sync/atomic" - "time" - "unicode" - - "github.com/uptrace/bun/schema" -) - -type QueryEvent struct { - DB *DB - - QueryAppender schema.QueryAppender // DEPRECATED: use IQuery instead - IQuery Query - Query string - QueryTemplate string - QueryArgs []interface{} - Model Model - - StartTime time.Time - Result sql.Result - Err error - - Stash map[interface{}]interface{} -} - -func (e *QueryEvent) Operation() string { - if e.IQuery != nil { - return e.IQuery.Operation() - } - return queryOperation(e.Query) -} - -func queryOperation(query string) string { - queryOp := strings.TrimLeftFunc(query, unicode.IsSpace) - - if idx := strings.IndexByte(queryOp, ' '); idx > 0 { - queryOp = queryOp[:idx] - } - if len(queryOp) > 16 { - queryOp = queryOp[:16] - } - return queryOp -} - -type QueryHook interface { - BeforeQuery(context.Context, *QueryEvent) context.Context - AfterQuery(context.Context, *QueryEvent) -} - -func (db *DB) beforeQuery( - ctx context.Context, - iquery Query, - queryTemplate string, - queryArgs []interface{}, - query string, - model Model, -) (context.Context, *QueryEvent) { - atomic.AddUint32(&db.stats.Queries, 1) - - if len(db.queryHooks) == 0 { - return ctx, nil - } - - event := &QueryEvent{ - DB: db, - - Model: model, - QueryAppender: iquery, - IQuery: iquery, - Query: query, - QueryTemplate: queryTemplate, - QueryArgs: queryArgs, - - StartTime: time.Now(), - } - - for _, hook := range db.queryHooks { - ctx = hook.BeforeQuery(ctx, event) - } - - return ctx, event -} - -func (db *DB) afterQuery( - ctx context.Context, - event *QueryEvent, - res sql.Result, - err error, -) { - switch err { - case nil, sql.ErrNoRows: - // nothing - default: - atomic.AddUint32(&db.stats.Errors, 1) - } - - if event == nil { - return - } - - event.Result = res - event.Err = err - - db.afterQueryFromIndex(ctx, event, len(db.queryHooks)-1) -} - -func (db *DB) afterQueryFromIndex(ctx context.Context, event *QueryEvent, hookIndex int) { - for ; hookIndex >= 0; hookIndex-- { - db.queryHooks[hookIndex].AfterQuery(ctx, event) - } -} |