From 6392e00653d3b81062ef60d8ae2fa2621873533f Mon Sep 17 00:00:00 2001 From: Dominik Süß Date: Tue, 9 May 2023 19:19:48 +0200 Subject: feat: initial tracing support (#1623) --- internal/db/bundb/bundb.go | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'internal/db') diff --git a/internal/db/bundb/bundb.go b/internal/db/bundb/bundb.go index 71433dbc2..ecf8752ea 100644 --- a/internal/db/bundb/bundb.go +++ b/internal/db/bundb/bundb.go @@ -42,6 +42,7 @@ import ( "github.com/superseriousbusiness/gotosocial/internal/id" "github.com/superseriousbusiness/gotosocial/internal/log" "github.com/superseriousbusiness/gotosocial/internal/state" + "github.com/superseriousbusiness/gotosocial/internal/tracing" "github.com/uptrace/bun" "github.com/uptrace/bun/dialect/pgdialect" "github.com/uptrace/bun/dialect/sqlitedialect" @@ -134,6 +135,10 @@ func NewBunDBService(ctx context.Context, state *state.State) (db.DB, error) { // Add database query hook conn.DB.AddQueryHook(queryHook{}) + if config.GetTracingEnabled() { + conn.DB.AddQueryHook(tracing.InstrumentBun()) + } + // execute sqlite pragmas *after* adding database hook; // this allows the pragma queries to be logged if t == "sqlite" { -- cgit v1.2.3