db launch fixes

This commit is contained in:
Ravi Prasad
2025-06-03 21:08:17 -05:00
parent 5a027f6636
commit a3225b1803
2 changed files with 28 additions and 26 deletions
+16 -12
View File
@@ -8,24 +8,28 @@ import (
"github.com/nprasad2077/NBA_Go/utils/metrics" "github.com/nprasad2077/NBA_Go/utils/metrics"
) )
func InitDB() *gorm.DB { func InitDB(shouldMigrate bool) *gorm.DB {
db, err := gorm.Open(sqlite.Open("/app/data/nba.db"), &gorm.Config{}) db, err := gorm.Open(sqlite.Open("/app/data/nba.db"), &gorm.Config{})
if err != nil { if err != nil {
log.Printf("/app/config/database.go:11\n[error] failed to initialize database, got error %v", err) log.Fatalf("failed to connect database: %v", err)
log.Fatal("Failed to connect database")
} }
// Track database operations
metrics.DBOperationsTotal.WithLabelValues("connect", "database").Inc() metrics.DBOperationsTotal.WithLabelValues("connect", "database").Inc()
// Auto migrate models if shouldMigrate {
db.AutoMigrate(&models.PlayerAdvancedStat{}) if err := db.AutoMigrate(&models.PlayerAdvancedStat{}); err != nil {
db.AutoMigrate(&models.PlayerTotalStat{}) log.Fatalf("migrate PlayerAdvancedStat: %v", err)
db.AutoMigrate(&models.PlayerShotChart{}) }
if err := db.AutoMigrate(&models.PlayerTotalStat{}); err != nil {
db.AutoMigrate(&models.APIKey{}) 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() metrics.DBOperationsTotal.WithLabelValues("migrate", "database").Inc()
}
return db return db
} }
+6 -8
View File
@@ -25,6 +25,7 @@ import (
"github.com/gofiber/fiber/v2/middleware/logger" "github.com/gofiber/fiber/v2/middleware/logger"
"github.com/prometheus/client_golang/prometheus/promhttp" "github.com/prometheus/client_golang/prometheus/promhttp"
fiberswagger "github.com/swaggo/fiber-swagger" fiberswagger "github.com/swaggo/fiber-swagger"
// "gorm.io/gorm"
"github.com/nprasad2077/NBA_Go/config" "github.com/nprasad2077/NBA_Go/config"
"github.com/nprasad2077/NBA_Go/controllers" "github.com/nprasad2077/NBA_Go/controllers"
@@ -36,7 +37,8 @@ import (
func main() { func main() {
// ——— One-off import-data mode ——— // ——— One-off import-data mode ———
if len(os.Args) > 1 && os.Args[1] == "import-data" { if len(os.Args) > 1 && os.Args[1] == "import-data" {
db := config.InitDB() // Run all migrations + import steps exactly once
db := config.InitDB(true)
importPlayerAdvanced(db) importPlayerAdvanced(db)
log.Println("🎉 Player Advanced Import completed successfully") log.Println("🎉 Player Advanced Import completed successfully")
@@ -44,21 +46,17 @@ func main() {
importPlayerAdvancedPlayoffs(db) importPlayerAdvancedPlayoffs(db)
log.Println("🎉 Player Advanced Playoffs Import completed successfully") log.Println("🎉 Player Advanced Playoffs Import completed successfully")
importPlayerTotalsScrape(db) importPlayerTotalsScrape(db)
log.Println("🎉 Player Totals (scraped) Import completed successfully") log.Println("🎉 Player Totals (scraped) Import completed successfully")
importPlayerTotalsPlayoffsScrape(db) importPlayerTotalsPlayoffsScrape(db)
log.Println("🎉 Player Playoffs (scraped) Import completed successfully") log.Println("🎉 Player Playoffs (scraped) Import completed successfully")
// importPlayerShotChart(db)
// log.Println("🎉 Player Shot Chart Import Completed Successfully ")
log.Println("🏀 ALL Imports completed successfully ✅ 🙌") log.Println("🏀 ALL Imports completed successfully ✅ 🙌")
return return
} }
// gracefull shutdown context // ——— Normal API startup: skip AutoMigrate ———
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM) ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer stop() defer stop()
@@ -81,8 +79,8 @@ func main() {
app.Use(logger.New()) app.Use(logger.New())
app.Use(middleware.MetricsMiddleware()) app.Use(middleware.MetricsMiddleware())
// DB connection // DB connection (no migrations on API startup)
db := config.InitDB() db := config.InitDB(false)
/* ---------- PUBLIC ROUTES (no API key) ---------- */ /* ---------- PUBLIC ROUTES (no API key) ---------- */
app.Get("/metrics", adaptor.HTTPHandler(promhttp.Handler())) app.Get("/metrics", adaptor.HTTPHandler(promhttp.Handler()))