Projeto bot pontos

backend/internal/adapters/postgres/pool.go

@@ -0,0 +1,58 @@
+// Package postgres é o adapter de saída para o PostgreSQL.
+package postgres
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"time"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+
+	"botpontos/internal/config"
+)
+
+const (
+	connectAttempts = 10
+	connectDelay    = 3 * time.Second
+)
+
+// OpenPool abre o pool e espera o Postgres responder, com algumas tentativas
+// para tolerar a ordem de subida dos containers.
+func OpenPool(ctx context.Context, cfg config.DB, logger *slog.Logger) (*pgxpool.Pool, error) {
+	poolCfg, err := pgxpool.ParseConfig(cfg.DSN())
+	if err != nil {
+		return nil, fmt.Errorf("configuração do Postgres: %w", err)
+	}
+	poolCfg.MaxConns = cfg.MaxConns
+
+	pool, err := pgxpool.NewWithConfig(ctx, poolCfg)
+	if err != nil {
+		return nil, fmt.Errorf("criar pool do Postgres: %w", err)
+	}
+	if err := waitFor(ctx, logger, pool.Ping); err != nil {
+		pool.Close()
+		return nil, err
+	}
+	return pool, nil
+}
+
+func waitFor(ctx context.Context, logger *slog.Logger, ping func(context.Context) error) error {
+	var err error
+	for attempt := 1; attempt <= connectAttempts; attempt++ {
+		pingCtx, cancel := context.WithTimeout(ctx, 5*time.Second)
+		err = ping(pingCtx)
+		cancel()
+		if err == nil {
+			return nil
+		}
+
+		logger.Warn("Postgres indisponível; nova tentativa", "attempt", attempt, "error", err)
+		select {
+		case <-ctx.Done():
+			return ctx.Err()
+		case <-time.After(connectDelay):
+		}
+	}
+	return fmt.Errorf("conectar ao Postgres após %d tentativas: %w", connectAttempts, err)
+}