fleet docker

main.go

@@ -2,56 +2,69 @@ package main
 
 import (
     "log"
+    "os"
     "time"
     "github.com/gofiber/fiber/v2/middleware/logger"
     "github.com/gofiber/fiber/v2"
-	"github.com/nprasad2077/NBA_Go/config"
-	"github.com/nprasad2077/NBA_Go/routes" 
+    "github.com/nprasad2077/NBA_Go/config"
+    "github.com/nprasad2077/NBA_Go/routes" 
     "github.com/nprasad2077/NBA_Go/services"
     _"github.com/nprasad2077/NBA_Go/docs" // swag docs
-	fiberswagger "github.com/swaggo/fiber-swagger"
+    fiberswagger "github.com/swaggo/fiber-swagger"
+    "gorm.io/gorm"
 )
 
-func main() {
-    app := fiber.New()
-
-    app.Use(logger.New())
-
-    db := config.InitDB()
-
-    // Automatically fetch player stats on startup
-    go func() {
-        for season := 1993; season <= 2025; season++ {
-            if err := services.FetchAndStorePlayerAdvancedStats(db, season); err != nil {
-                log.Printf("Fetch failed for player advanced  season %d: %v\n", season, err)
-            } else {
-                log.Printf("Fetch successful for player advanced season %d\n", season)
-            }
-            time.Sleep(1100 * time.Millisecond) // optional delay
-        }
-        log.Printf("player advanced Import Success")
-    }()
-
-    go func() {
-        for season := 1993; season <= 2025; season++ {
-            if err := services.FetchAndStorePlayerTotalStats(db, season); err != nil {
-                log.Printf("Fetch failed for player totals season %d: %v\n", season, err)
-            } else {
-                log.Printf("Fetch successful for player totals  season %d\n", season)
-            }
-            time.Sleep(1000 * time.Millisecond) // optional delay
-        }
-        log.Printf("player totals Import Success")
-    }()
-
+// importData handles fetching and storing all player stats
+func importData(db *gorm.DB) {
+    log.Println("Starting data import process...")
     
+    // Import advanced stats
+    for season := 2020; season <= 2025; season++ {
+        if err := services.FetchAndStorePlayerAdvancedStats(db, season); err != nil {
+            log.Printf("Fetch failed for player advanced season %d: %v\n", season, err)
+        } else {
+            log.Printf("Fetch successful for player advanced season %d\n", season)
+        }
+        time.Sleep(1100 * time.Millisecond) // optional delay
+    }
+    log.Printf("Player advanced import completed")
+    
+    // Import total stats
+    for season := 2020; season <= 2025; season++ {
+        if err := services.FetchAndStorePlayerTotalStats(db, season); err != nil {
+            log.Printf("Fetch failed for player totals season %d: %v\n", season, err)
+        } else {
+            log.Printf("Fetch successful for player totals season %d\n", season)
+        }
+        time.Sleep(1000 * time.Millisecond) // optional delay
+    }
+    log.Printf("Player totals import completed")
+    
+    log.Println("Data import process completed successfully!")
+}
 
-
+func main() {
+    // Check if we're running in import-data mode
+    if len(os.Args) > 1 && os.Args[1] == "import-data" {
+        db := config.InitDB()
+        importData(db)
+        return // Exit after import is complete
+    }
+    
+    // Regular API server mode
+    app := fiber.New()
+    app.Use(logger.New())
+    
+    db := config.InitDB()
+    
+    // No automatic data fetching in API server mode
+    // Only the db-init container will handle data imports
 
     routes.RegisterPlayerAdvancedRoutes(app, db)
     routes.RegisterPlayerTotalRoutes(app, db)
 
     app.Get("/swagger/*", fiberswagger.WrapHandler)
-
-    app.Listen(":3001")
+    
+    log.Println("Starting API server on port 5000...")
+    app.Listen(":5000")
 }