db.go

package requiem

import (
	"fmt"

	"github.com/caarlos0/env"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

type dbConfig struct {
	Host         string `env:"DB_HOST"`
	Port         string `env:"DB_PORT"`
	Username     string `env:"DB_USERNAME"`
	Password     string `env:"DB_PASSWORD"`
	DatabaseName string `env:"DB_NAME"`
	SSLMode      string `env:"DB_SSL_MODE" envDefault:"disable"`
}

func loadDBConfig() dbConfig {
	cfg := dbConfig{}
	err := env.Parse(&cfg)
	if err != nil {
		Logger.Fatal("Could not load DB config: %s", err.Error())
	}

	return cfg
}

func buildGormConfig(debugMode bool) *gorm.Config {
	c := &gorm.Config{}
	if debugMode {
		c.Logger = logger.Default.LogMode(logger.Info)
	} else {
		c.Logger = logger.Default.LogMode(logger.Silent)
	}

	return c
}

// newPostgresDBConnection obtains DB connection details from environment variables
// and initializes a Postgres DB connection.
func newPostgresDBConnection(debugMode bool) *gorm.DB {
	cfg := loadDBConfig()
	dsn := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=%s",
		cfg.Host, cfg.Port, cfg.Username, cfg.Password, cfg.DatabaseName, cfg.SSLMode)

	db, err := gorm.Open(postgres.Open(dsn), buildGormConfig(debugMode))
	if err != nil {
		Logger.Fatal("Could not connect to DB %s", err)
	}

	return db
}

// newInMemoryDBConnection initializes a SQLite in-memory DB connection
func newInMemoryDBConnection(debugMode bool) *gorm.DB {
	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), buildGormConfig(debugMode))
	if err != nil {
		Logger.Fatal("Could not connect to DB %s", err)
	}

	return db
}