add register driver

This commit is contained in:
namhq1989 2023-02-03 20:58:03 +07:00
parent 9b0fbe90c5
commit 01ae1ec191
1 changed files with 2 additions and 1 deletions

View File

@ -5,7 +5,7 @@ import (
"fmt" "fmt"
"time" "time"
_ "github.com/jackc/pgx/v4/stdlib" "github.com/jackc/pgx/v4/stdlib"
"github.com/volatiletech/sqlboiler/v4/boil" "github.com/volatiletech/sqlboiler/v4/boil"
"go.elastic.co/apm/module/apmsql/v2" "go.elastic.co/apm/module/apmsql/v2"
_ "go.elastic.co/apm/module/apmsql/v2/pgxv4" _ "go.elastic.co/apm/module/apmsql/v2/pgxv4"
@ -32,6 +32,7 @@ func Connect(cfg Config, server string) (db *sql.DB, err error) {
// connect // connect
if cfg.UseElasticAPM { if cfg.UseElasticAPM {
apmsql.Register("pgx", &stdlib.Driver{})
db, err = apmsql.Open("pgx", uri) db, err = apmsql.Open("pgx", uri)
} else { } else {
db, err = sql.Open("pgx", uri) db, err = sql.Open("pgx", uri)