config/database_connection_config.go

package config

import (
	"fmt"
	"godp.abdanhafidz.com/models"
	"log"
	"os"

	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	"github.com/joho/godotenv"
)

var DB *gorm.DB
var err error
var Salt string

func init() {
	godotenv.Load()
	if err != nil {
		fmt.Println("Gagal membaca file .env")
		return
	}
	os.Setenv("TZ", "Asia/Jakarta")
	dbHost := os.Getenv("DB_HOST")
	dbPort := os.Getenv("DB_PORT")
	dbUser := os.Getenv("DB_USER")
	dbPassword := os.Getenv("DB_PASSWORD")
	dbName := os.Getenv("DB_NAME")
	Salt := os.Getenv("SALT")
	dsn := "host=" + dbHost + " user=" + dbUser + " password=" + dbPassword + " dbname=" + dbName + " port=" + dbPort + " sslmode=disable TimeZone=Asia/Jakarta"
	DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{TranslateError: true})
	if err != nil {
		panic(err)
	}
	if Salt == "" {
		Salt = "D3f4u|t"
	}

	// Call AutoMigrateAll to perform auto-migration
	AutoMigrateAll(DB)
}

func AutoMigrateAll(db *gorm.DB) {
	// Enable logger to see SQL logs
	db.Logger.LogMode(logger.Info)

	// Auto-migrate all models
	db.Exec("CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\";")
	if err := db.AutoMigrate(&models.Account{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.AccountDetails{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.EmailVerification{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ExternalAuth{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.FCM{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ForgotPassword{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.Events{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.Announcement{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ProblemSet{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.Questions{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.EventAssign{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ProblemSetAssign{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ExamProgress{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.ExamProgress_Result{}); err != nil {
		log.Fatal(err)
	}
	if err := db.AutoMigrate(&models.Result{}); err != nil {
		log.Fatal(err)
	}

	fmt.Println("Migration completed successfully.")
}