go/testing-go-code-with-postgres
1
0
Fork 0
mirror of https://github.com/xorcare/testing-go-code-with-postgres.git synced 2024-12-24 16:28:34 +02:00
Code Issues Releases Activity

Make a singleton for *pgxpool.Pool

Browse Source 
This should reduce the number of parasitic connection reopenings.
This commit is contained in:
Vasiliy Vasilyuk 2023-07-12 21:08:04 +03:00
parent 4bb736475b
commit 59c4d2e3e1
No known key found for this signature in database
GPG Key ID: CD966F83D6FAFF48
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
testingpg/testingpg.go
View File

@ -9,6 +9,7 @@ import (
"fmt"
"net/url"
"os"
"sync"
"time"
"github.com/google/uuid"
@ -34,6 +35,7 @@ type Postgres struct {
ref string
conn *pgxpool.Pool
once sync.Once
}
func newPostgres(t TestingT) *Postgres {
@ -67,15 +69,11 @@ func (p *Postgres) URL() string {
}
func (p *Postgres) PgxPool() *pgxpool.Pool {
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()
p.once.Do(func() {
p.conn = newPGxPool(p.t, p.URL())
})
return pool
return p.conn
}
func (p *Postgres) cloneFromReference(t TestingT) *Postgres {
@ -117,3 +115,18 @@ func replaceDBName(t TestingT, dataSourceURL, dbname string) string {
r.Path = dbname
return r.String()
}
func newPGxPool(t TestingT, dataSourceURL string) *pgxpool.Pool {
ctx, done := context.WithTimeout(context.Background(), 1*time.Second)
defer done()
pool, err := pgxpool.New(ctx, dataSourceURL)
require.NoError(t, err)
// Automatically close connection after the test is completed.
t.Cleanup(func() {
pool.Close()
})
return pool
}