postgres.go

package repository

import (
	"fmt"
	"log"

	"github.com/Arjuna-Ragil/Localbase/Internal/config"
	"github.com/Arjuna-Ragil/Localbase/Internal/core/domain"
	"github.com/jmoiron/sqlx"
	_ "github.com/lib/pq"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

type DBContainer struct {
	Gorm *gorm.DB
	Sqlx *sqlx.DB
}

func ConnectDB() (*DBContainer, error) {
	cfg := config.LoadConfig()

	gormDB, err := gorm.Open(postgres.Open(cfg.DBURL), &gorm.Config{})
	if err != nil {
		log.Fatalf("Failed to connect to gorm database: %v", err)
	}
	fmt.Println("Connected to gorm database")

	sqlxDB, err := sqlx.Connect("postgres", cfg.DBURL)
	if err != nil {
		log.Fatalf("Failed to connect to sqlx database: %v", err)
	}
	fmt.Println("Connected to sqlx database")

	return &DBContainer{
		Gorm: gormDB,
		Sqlx: sqlxDB,
	}, nil
}
func (db *DBContainer) Migrate() error {
	err := db.Gorm.AutoMigrate(
		&domain.User{},
		&domain.Invitation{},
	)
	if err != nil {
		log.Fatalf("Failed to migrate users: %v", err)
	}
	fmt.Println("Migrated users")
	return nil
}