diff --git a/logs/database_logs.go b/logs/database_logs.go index 3edece0ccb41b957c1c3f69e31a46700d9f4a4ad..72f9c18c9dfdaa68b20f3b8740befb75d0420077 100644 --- a/logs/database_logs.go +++ b/logs/database_logs.go @@ -30,6 +30,10 @@ var ignoredTableInserts = []string{ "revenue_rollup", } +type ContextWithSuppressDBLog interface { + DoSuppressDBLog() bool +} + type QueryHook struct { IgnoredTableInserts []string Debug bool @@ -39,7 +43,14 @@ func (d QueryHook) BeforeQuery(ctx context.Context, _ *bun.QueryEvent) context.C return ctx } -func (d QueryHook) AfterQuery(_ context.Context, event *bun.QueryEvent) { +func (d QueryHook) AfterQuery(ctx context.Context, event *bun.QueryEvent) { + contextWithSuppressDBLog, ok := ctx.(ContextWithSuppressDBLog) + if ok { + if contextWithSuppressDBLog.DoSuppressDBLog() { + return + } + } + sqlQuery := event.Query queryDuration := time.Now().Sub(event.StartTime)