changes

pgx.go

@@ -0,0 +1,59 @@
+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)
+	}
+}