diff --git a/pkg/sql/connection.go b/pkg/sql/connection.go index 5ae2d6f3c0..fe9dc52aaa 100644 --- a/pkg/sql/connection.go +++ b/pkg/sql/connection.go @@ -67,30 +67,28 @@ func NewDbConnection(cfg *Config, logger *zap.SugaredLogger) (*pg.DB, error) { } //-------------- - if cfg.LogQuery || cfg.LogAllQuery { + dbConnection.OnQueryProcessed(func(event *pg.QueryProcessedEvent) { + queryDuration := time.Since(event.StartTime) - dbConnection.OnQueryProcessed(func(event *pg.QueryProcessedEvent) { - query, err := event.FormattedQuery() - queryDuration := time.Since(event.StartTime) - - // Expose prom metrics - if cfg.ExportPromMetrics { - middleware.PgQueryDuration.WithLabelValues("value").Observe(queryDuration.Seconds()) - } + // Expose prom metrics + if cfg.ExportPromMetrics { + middleware.PgQueryDuration.WithLabelValues("value").Observe(queryDuration.Seconds()) + } - if err != nil { - logger.Errorw("Error formatting query", - "err", err) - return - } + query, err := event.FormattedQuery() + if err != nil { + logger.Errorw("Error formatting query", + "err", err) + return + } - if cfg.LogAllQuery || queryDuration.Milliseconds() > cfg.QueryDurationThreshold { - logger.Debugw("query time", - "duration", queryDuration.Seconds(), - "query", query) - } - }) - } + // Log pg query if enabled + if cfg.LogAllQuery || (cfg.LogQuery && queryDuration.Milliseconds() > cfg.QueryDurationThreshold) { + logger.Debugw("query time", + "duration", queryDuration.Seconds(), + "query", query) + } + }) return dbConnection, err }