added an init to the binaries

This commit is contained in:
2024-10-14 14:38:12 +03:00
parent b625b3cf5e
commit 1a6877c350
7 changed files with 119 additions and 0 deletions

View File

@@ -0,0 +1,35 @@
package initializers
import (
"os"
"github.com/joho/godotenv"
"gorm.io/driver/postgres"
"gorm.io/gorm"
)
var DB *gorm.DB
func ConnectToDb() {
var err error
// Load the .env file
err = godotenv.Load()
if err != nil {
panic("Error loading .env file")
}
dsn := "host=" + os.Getenv("DB_HOST") +
" user=" + os.Getenv("DB_USER") +
" password=" + os.Getenv("DB_PASSWORD") +
" dbname=" + os.Getenv("DB_NAME") +
" port=" + os.Getenv("DB_PORT") +
" sslmode=" + os.Getenv("DB_SSLMODE") +
" TimeZone=" + os.Getenv("DB_TIMEZONE")
DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
panic("Failed to connect to DB")
}
}

View File

@@ -0,0 +1,14 @@
package initializers
import (
"log"
"github.com/joho/godotenv"
)
func LoadEnvVariables() {
err := godotenv.Load()
if err != nil {
log.Fatal("Error loading .env file")
}
}

View File

@@ -0,0 +1,10 @@
package initializers
import "git.grassecon.net/urdt/ussd/internal/models"
func SyncDatabase() {
DB.AutoMigrate(&models.User{})
DB.AutoMigrate(&models.ActiveVoucher{})
DB.AutoMigrate(&models.Voucher{})
DB.AutoMigrate(&models.Transaction{})
}