GOMODULES
/
pgx
mirror of https://github.com/jackc/pgx.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

stdlib AfterConnect takes context

pull/586/head
Jack Christensen 2019-06-29 13:53:43 -05:00
parent 08b412740f
commit a99cd7758a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
stdlib/opendb.go
View File

@ -14,7 +14,7 @@ import (
type OptionOpenDB func(*connector)
// OptionAfterConnect provide a callback for after connect.
func OptionAfterConnect(ac func(*pgx.Conn) error) OptionOpenDB {
func OptionAfterConnect(ac func(context.Context, *pgx.Conn) error) OptionOpenDB {
return func(dc *connector) {
dc.AfterConnect = ac
}
@ -23,7 +23,7 @@ func OptionAfterConnect(ac func(*pgx.Conn) error) OptionOpenDB {
func OpenDB(config pgx.ConnConfig, opts ...OptionOpenDB) *sql.DB {
c := connector{
ConnConfig: config,
AfterConnect: func(*pgx.Conn) error { return nil }, // noop after connect by default
AfterConnect: func(context.Context, *pgx.Conn) error { return nil }, // noop after connect by default
driver: pgxDriver,
}
@ -36,7 +36,7 @@ func OpenDB(config pgx.ConnConfig, opts ...OptionOpenDB) *sql.DB {
type connector struct {
pgx.ConnConfig
AfterConnect func(*pgx.Conn) error // function to call on every new connection
AfterConnect func(context.Context, *pgx.Conn) error // function to call on every new connection
driver *Driver
}
@ -51,7 +51,7 @@ func (c connector) Connect(ctx context.Context) (driver.Conn, error) {
return nil, err
}
if err = c.AfterConnect(conn); err != nil {
if err = c.AfterConnect(ctx, conn); err != nil {
return nil, err
}