60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package goblocks
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"sync"
|
|
|
|
_ "github.com/jackc/pgconn"
|
|
_ "github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
_ "github.com/jackc/pgx/v5/stdlib"
|
|
)
|
|
|
|
var (
|
|
pgxPools = make(map[string]*pgxpool.Pool)
|
|
pgxMutex sync.RWMutex
|
|
)
|
|
|
|
func (a *App) NewPGXPool(name string) *pgxpool.Pool {
|
|
pgxMutex.Lock()
|
|
defer pgxMutex.Unlock()
|
|
|
|
if pool, exists := pgxPools[name]; exists {
|
|
return pool
|
|
}
|
|
|
|
dbPool, err := pgxpool.New(context.Background(), a.Datasource(name))
|
|
if err != nil {
|
|
slog.Error("error connecting to database", "error", err, "name", name)
|
|
panic(err)
|
|
}
|
|
|
|
if err := dbPool.Ping(context.Background()); err != nil {
|
|
slog.Error("error pinging database, maybe incorrect datasource", "error", err, "name", name)
|
|
panic(err)
|
|
}
|
|
|
|
pgxPools[name] = dbPool
|
|
slog.Info("connected to database", "name", name)
|
|
return dbPool
|
|
}
|
|
|
|
func (a *App) GetPGXPool(name string) (*pgxpool.Pool, bool) {
|
|
pgxMutex.RLock()
|
|
defer pgxMutex.RUnlock()
|
|
pool, exists := pgxPools[name]
|
|
return pool, exists
|
|
}
|
|
|
|
func (a *App) ClosePGXPools() {
|
|
pgxMutex.Lock()
|
|
defer pgxMutex.Unlock()
|
|
|
|
for name, pool := range pgxPools {
|
|
pool.Close()
|
|
delete(pgxPools, name)
|
|
slog.Info("closed database connection", "name", name)
|
|
}
|
|
}
|