mirror of
https://github.com/nprasad2077/NBA_Go.git
synced 2026-09-22 14:05:13 +00:00
tests running
This commit is contained in:
+11
-3
@@ -2,19 +2,27 @@ package config
|
||||
|
||||
import (
|
||||
"log"
|
||||
"gorm.io/driver/sqlite" // Or Postgres
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
"github.com/nprasad2077/NBA_Go/models"
|
||||
"github.com/nprasad2077/NBA_Go/utils/metrics"
|
||||
)
|
||||
|
||||
func InitDB() *gorm.DB {
|
||||
db, err := gorm.Open(sqlite.Open("/app/data/nba.db"), &gorm.Config{})
|
||||
if err != nil {
|
||||
log.Printf("/app/config/database.go:11\n[error] failed to initialize database, got error %v", err)
|
||||
log.Fatal("Failed to connect database")
|
||||
}
|
||||
|
||||
|
||||
// Track database operations
|
||||
metrics.DBOperationsTotal.WithLabelValues("connect", "database").Inc()
|
||||
|
||||
// Auto migrate models
|
||||
db.AutoMigrate(&models.PlayerAdvancedStat{})
|
||||
db.AutoMigrate(&models.PlayerTotalStat{})
|
||||
|
||||
|
||||
metrics.DBOperationsTotal.WithLabelValues("migrate", "database").Inc()
|
||||
|
||||
return db
|
||||
}
|
||||
Reference in New Issue
Block a user