go/testing-go-code-with-postgres
1
0
Fork 0
mirror of https://github.com/xorcare/testing-go-code-with-postgres.git synced 2025-06-30 23:23:40 +02:00
Code Issues Releases Activity

Make a lazy loading for opening a connection

Browse Source 
In the current approach, the connection logic is only triggered at the
moment of immediate need for connection.

We also get a single connection opening point.
This commit is contained in:
Vasiliy Vasilyuk
2023-07-12 20:55:14 +03:00
parent c2e0810c2c
commit 4bb736475b
1 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
testingpg/testingpg.go
View File

@ -28,6 +28,8 @@ func New(t TestingT) *Postgres {
}
type Postgres struct {
t TestingT
url string
ref string
@ -51,6 +53,8 @@ func newPostgres(t TestingT) *Postgres {
require.NoError(t, err)
return &Postgres{
t: t,
url: urlStr,
ref: refDatabase,
@ -63,26 +67,28 @@ func (p *Postgres) URL() string {
}
func (p *Postgres) PgxPool() *pgxpool.Pool {
return p.conn
pool, err := pgxpool.New(context.Background(), p.URL())
require.NoError(p.t, err)
// Automatically close connection after the test is completed.
p.t.Cleanup(func() {
pool.Close()
})
return pool
}
func (p *Postgres) cloneFromReference(t TestingT) *Postgres {
cfg, err := pgxpool.ParseConfig(p.url)
require.NoError(t, err)
pool, err := pgxpool.New(context.Background(), p.url)
require.NoError(t, err)
newDatabaseName := uuid.New().String()
const sqlTemplate = `CREATE DATABASE %q WITH TEMPLATE %s OWNER %s;`
const sqlTemplate = `CREATE DATABASE %q WITH TEMPLATE %s;`
sql := fmt.Sprintf(
sqlTemplate,
newDatabaseName,
p.ref,
cfg.ConnConfig.User,
)
_, err = pool.Exec(context.Background(), sql)
_, err := p.PgxPool().Exec(context.Background(), sql)
require.NoError(t, err)
// Automatically drop database copy after the test is completed.
@ -96,20 +102,17 @@ func (p *Postgres) cloneFromReference(t TestingT) *Postgres {
require.NoError(t, err)
})
urlString := replaceDBName(t, cfg, newDatabaseName)
newPool, err := pgxpool.New(context.Background(), urlString)
require.NoError(t, err)
urlString := replaceDBName(t, p.URL(), newDatabaseName)
return &Postgres{
t: p.t,
url: urlString,
ref: newDatabaseName,
conn: newPool,
}
}
func replaceDBName(t TestingT, cfg *pgxpool.Config, dbname string) string {
r, err := url.Parse(cfg.ConnString())
func replaceDBName(t TestingT, dataSourceURL, dbname string) string {
r, err := url.Parse(dataSourceURL)
require.NoError(t, err)
r.Path = dbname
return r.String()