add register driver

2023-02-03 20:58:03 +07:00
1 changed files with 2 additions and 1 deletions
--- a/postgresql.go
+++ b/postgresql.go
@ -5,7 +5,7 @@ import (
 	"fmt"
 	"time"

-	_ "github.com/jackc/pgx/v4/stdlib"
+	"github.com/jackc/pgx/v4/stdlib"
 	"github.com/volatiletech/sqlboiler/v4/boil"
 	"go.elastic.co/apm/module/apmsql/v2"
 	_ "go.elastic.co/apm/module/apmsql/v2/pgxv4"
@ -32,6 +32,7 @@ func Connect(cfg Config, server string) (db *sql.DB, err error) {

 	// connect
 	if cfg.UseElasticAPM {
+		apmsql.Register("pgx", &stdlib.Driver{})
 		db, err = apmsql.Open("pgx", uri)
 	} else {
 		db, err = sql.Open("pgx", uri)