3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-08-06 10:21:05 +02:00
Code Issues Projects Releases Packages Wiki Activity

storage/postgres: pgx client tracing

Browse source
This commit is contained in:
Joe Kralicky 2025-01-21 21:41:58 +00:00
parent 926197dbae
commit 2f24aab7f0
No known key found for this signature in database
GPG key ID: 75C4875F34A9FB79
7 changed files with 97 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pkg/storage/postgres/backend.go
View file

@ -9,6 +9,7 @@ import (
"time"
"github.com/cenkalti/backoff/v4"
"github.com/exaring/otelpgx"
"github.com/jackc/pgx/v5/pgxpool"
"google.golang.org/protobuf/types/known/fieldmaskpb"
"google.golang.org/protobuf/types/known/timestamppb"
@ -353,6 +354,11 @@ func (backend *Backend) init(ctx context.Context) (serverVersion uint64, pool *p
return serverVersion, nil, err
}
if backend.cfg.tracerProvider != nil {
config.ConnConfig.Tracer = otelpgx.NewTracer(
otelpgx.WithTracerProvider(backend.cfg.tracerProvider))
}
pool, err = pgxpool.NewWithConfig(context.Background(), config)
if err != nil {
return serverVersion, nil, err