jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(backend): pg-connector use Temeout context

Browse source
This commit is contained in:
ShiKhu 2022-02-04 23:12:33 +01:00
parent 7f14269671
commit ca870106e2
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
backend/pkg/db/postgres/connector.go
View file

@ -8,6 +8,11 @@ import (
"github.com/jackc/pgx/v4/pgxpool"
)
func getTimeoutContext() context.Context {
ctx, _ := context.WithTimeout(context.Background(), time.Duration(time.Second*10))
return ctx
}
type Conn struct {
c *pgxpool.Pool // TODO: conditional usage of Pool/Conn (use interface?)
}
@ -26,15 +31,15 @@ func (conn *Conn) Close() error {
}
func (conn *Conn) query(sql string, args ...interface{}) (pgx.Rows, error) {
return conn.c.Query(context.Background(), sql, args...)
return conn.c.Query(getTimeoutContext(), sql, args...)
}
func (conn *Conn) queryRow(sql string, args ...interface{}) pgx.Row {
return conn.c.QueryRow(context.Background(), sql, args...)
return conn.c.QueryRow(getTimeoutContext(), sql, args...)
}
func (conn *Conn) exec(sql string, args ...interface{}) error {
_, err := conn.c.Exec(context.Background(), sql, args...)
_, err := conn.c.Exec(getTimeoutContext(), sql, args...)
return err
}