Base Project

internal/initialization/postgresql.go

@@ -0,0 +1,47 @@
+package initialization
+
+import (
+	"context"
+	"fmt"
+	"time"
+	"wm-backend/internal/models"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+	"github.com/rs/zerolog/log"
+)
+
+func ConnectPostgreSQL(config *models.Config) (*pgxpool.Pool, error) {
+	connStr := fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=disable",
+		config.Database.Username, config.Database.Password, config.Database.Host, config.Database.Port, config.Database.Name)
+
+	var pool *pgxpool.Pool
+	var err error
+
+	maxRetries := 5
+	for i := range maxRetries {
+		poolConfig, parseErr := pgxpool.ParseConfig(connStr)
+		if parseErr != nil {
+			log.Error().Err(parseErr).Msgf("Failed to parse DB config (attempt %d/%d)", i+1, maxRetries)
+			time.Sleep(5 * time.Second)
+			continue
+		}
+
+		pool, err = pgxpool.NewWithConfig(context.Background(), poolConfig)
+		if err != nil {
+			log.Error().Err(err).Msgf("Failed to connect to PostgreSQL (attempt %d/%d)", i+1, maxRetries)
+			time.Sleep(5 * time.Second)
+			continue
+		}
+
+		err = pool.Ping(context.Background())
+		if err != nil {
+			log.Error().Err(err).Msgf("Failed to ping PostgreSQL (attempt %d/%d)", i+1, maxRetries)
+			time.Sleep(5 * time.Second)
+			continue
+		}
+
+		log.Info().Msg("Successfully connected to PostgreSQL")
+		return pool, nil
+	}
+	return nil, fmt.Errorf("failed to connect to PostgreSQL after %d attempts: %v", maxRetries, err)
+}