NBA_Go/config/database.go

package config

import (
	"github.com/nprasad2077/NBA_Go/models"
	"github.com/nprasad2077/NBA_Go/utils/metrics"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"log"
)

func InitDB(shouldMigrate bool) *gorm.DB {
	db, err := gorm.Open(sqlite.Open("/app/data/nba.db"), &gorm.Config{})
	if err != nil {
		log.Fatalf("failed to connect database: %v", err)
	}
	metrics.DBOperationsTotal.WithLabelValues("connect", "database").Inc()

	if shouldMigrate {
		if err := db.AutoMigrate(&models.PlayerAdvancedStat{}); err != nil {
			log.Fatalf("migrate PlayerAdvancedStat: %v", err)
		}
		if err := db.AutoMigrate(&models.PlayerTotalStat{}); err != nil {
			log.Fatalf("migrate PlayerTotalStat: %v", err)
		}
		if err := db.AutoMigrate(&models.PlayerShotChart{}); err != nil {
			log.Fatalf("migrate PlayerShotChart: %v", err)
		}
		if err := db.AutoMigrate(&models.APIKey{}); err != nil {
			log.Fatalf("migrate APIKey: %v", err)
		}
		metrics.DBOperationsTotal.WithLabelValues("migrate", "database").Inc()
	}

	return db
}