Example of logging with pgxpool v5

pgxpool_logging_example.go

func main() {
  // postgres client
	ctx := context.Background()

	connStr := os.Getenv("DB_CONNECTION_STRING")
	cc, err := pgxpool.ParseConfig(connStr)
	if err != nil {
		log.Fatalf("Failed to parse config: %v", err)
	}
	cc.MinConns = 5
	cc.MaxConns = 10
	logger, err := zap.NewDevelopment()
	if err != nil {
		log.Fatalf("failed to create logger: %v", err)
	}
	cc.ConnConfig.Tracer = &myQueryTracer{
		log: logger.Sugar(),
	}

	// Create a connection pool using the configuration.
	conn, err := pgxpool.NewWithConfig(ctx, cc)
	if err != nil {
		log.Fatalf("failed opening connection to postgres: %v", err)
	}
	defer conn.Close()

	db := queries.New(conn)
}

type myQueryTracer struct {
	log *zap.SugaredLogger
}

func (tracer *myQueryTracer) TraceQueryStart(
	ctx context.Context,
	_ *pgx.Conn,
	data pgx.TraceQueryStartData) context.Context {
	tracer.log.Debugw("Executing command", "sql", data.SQL, "args", data.Args)

	return ctx
}

func (tracer *myQueryTracer) TraceQueryEnd(ctx context.Context, conn *pgx.Conn, data pgx.TraceQueryEndData) {
}