query-orchestration/internal/database/connection.go

package database

import (
	"context"
	"fmt"
	"log"
	"queryorchestration/internal/env"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgxpool"
)

func createConnectionString() string {
	driver := "postgres"
	dbUser := env.GetPanic("DB_USER")
	dbPass := env.GetPanic("DB_PASS")
	dbHost := env.GetPanic("DB_HOST")
	dbPort := env.GetPanic("DB_PORT")
	dbName := env.GetPanic("DB_NAME")
	opts := "sslmode=disable"
	connStr := fmt.Sprintf("%s://%s:%s@%s:%s/%s?%s", driver, dbUser, dbPass, dbHost, dbPort, dbName, opts)

	return connStr
}

func GetDBPool(ctx context.Context) *pgxpool.Pool {
	connStr := createConnectionString()

	pool, err := pgxpool.New(ctx, connStr)
	if err != nil {
		log.Panicf("Unable to create database pool: %v\n", err)
	}

	return pool
}

func GetDBConn(ctx context.Context) *pgx.Conn {
	connStr := createConnectionString()

	conn, err := pgx.Connect(ctx, connStr)
	if err != nil {
		log.Panicf("Unable to connect to database: %v\n", err)
	}

	return conn
}