query-orchestration/internal/database/connection.go

package database

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

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

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

	return connStr
}

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

	pool, err := pgxpool.New(ctx, connStr)
	if err != nil {
		log.Fatalf("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.Fatalf("Unable to connect to database: %v\n", err)
	}

	return conn
}