memnix / memnix-rest

package database

import (

	"fmt"

	"log"

	"os"

	"strconv"

	"time"

	"github.com/joho/godotenv"

	"gorm.io/driver/postgres"

	"gorm.io/gorm"

)

var (

	// DBConn is a pointer to gorm.DB

	DBConn   *gorm.DB

	user     string

	password string

	host     string

	db       string

	port     string

)

// LoadVar

func LoadVar() {

	// Load the .env file

	err := godotenv.Load()

	if err != nil {

		log.Fatal("Error loading .env file")

	}

	user = os.Getenv("DB_USER")         // Get DB_USER from env

	password = os.Getenv("DB_PASSWORD") // Get DB_PASSWORD from env

	host = os.Getenv("DB_HOST")         // Get DB_HOST from env

	db = os.Getenv("DB_DB")             // Get DB_DB (db name) from env

	port = os.Getenv("DB_PORT")         // Get DB_PORT from env

}

// Connect creates a connection to database

func Connect() (err error) {

	// Load var from .env file

	LoadVar()

	// Convert port

	port, err := strconv.Atoi(port)

	if err != nil {

		return err

	}

	// Create postgres connection string

	dsn := fmt.Sprintf("user=%s password=%s host=%s dbname=%s port=%d sslmode=disable", user, password, host, db, port)

	// Open connection

	DBConn, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})

	if err != nil {

		return err

	}

	sqlDB, err := DBConn.DB()

	if err != nil {

		return err

	}

	sqlDB.SetMaxIdleConns(10)

	sqlDB.SetMaxOpenConns(50)

	sqlDB.SetConnMaxLifetime(time.Hour)

	return nil

}