tablet/diamond-orm
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add AfterConnect callback to pgx pool config

Browse Source
This commit is contained in:
☙◦ The Tablet ❀ GamerGirlandCo ◦❧ 2025-07-16 23:14:34 -04:00
parent 33c47b2aa8
commit bb5d6e6db3
Signed by: tablet
GPG Key ID: 924A5F6AF051E87C
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
diamond.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"fmt" "fmt"
"github.com/jackc/pgx/v5" "github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgconn"
"github.com/jackc/pgx/v5/pgxpool" "github.com/jackc/pgx/v5/pgxpool"
"io" "io"
"log/slog" "log/slog"
@ -177,6 +178,14 @@ func Open(connString string, cfg *Config) (*Engine, error) {
e.pgCfg.MinConns = 5 e.pgCfg.MinConns = 5
e.pgCfg.MaxConns = 10 e.pgCfg.MaxConns = 10
e.pgCfg.MaxConnIdleTime = time.Minute * 2 e.pgCfg.MaxConnIdleTime = time.Minute * 2
e.pgCfg.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
oldHandler := conn.Config().OnPgError
conn.Config().OnPgError = func(conn *pgconn.PgConn, pgError *pgconn.PgError) bool {
e.logger.Error("ERROR ->", "err", pgError.Error())
return oldHandler(conn, pgError)
}
return nil
}
if err != nil { if err != nil {
return nil, err return nil, err
} }