diff options
Diffstat (limited to 'cmd/gotosocial/action/testrig/testrig.go')
| -rw-r--r-- | cmd/gotosocial/action/testrig/testrig.go | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/cmd/gotosocial/action/testrig/testrig.go b/cmd/gotosocial/action/testrig/testrig.go index 452df805c..f08bec609 100644 --- a/cmd/gotosocial/action/testrig/testrig.go +++ b/cmd/gotosocial/action/testrig/testrig.go @@ -38,6 +38,7 @@ import ( "github.com/superseriousbusiness/gotosocial/internal/gtserror" "github.com/superseriousbusiness/gotosocial/internal/language" "github.com/superseriousbusiness/gotosocial/internal/log" + "github.com/superseriousbusiness/gotosocial/internal/metrics" "github.com/superseriousbusiness/gotosocial/internal/middleware" "github.com/superseriousbusiness/gotosocial/internal/oidc" tlprocessor "github.com/superseriousbusiness/gotosocial/internal/processing/timeline" @@ -68,6 +69,10 @@ var Start action.GTSAction = func(ctx context.Context) error { return fmt.Errorf("error initializing tracing: %w", err) } + if err := metrics.Initialize(); err != nil { + return fmt.Errorf("error initializing metrics: %w", err) + } + // Initialize caches and database state.DB = testrig.NewTestDB(&state) @@ -142,6 +147,11 @@ var Start action.GTSAction = func(ctx context.Context) error { if config.GetTracingEnabled() { middlewares = append(middlewares, tracing.InstrumentGin()) } + + if config.GetMetricsEnabled() { + middlewares = append(middlewares, metrics.InstrumentGin()) + } + middlewares = append(middlewares, []gin.HandlerFunc{ middleware.Logger(config.GetLogClientIP()), middleware.UserAgent(), |