admin
1db56d6741
- Add error check for JSON response in panic handler - Remove duplicate defer CloseDB() call (handled in shutdown) - Add error checks for all WriteField() calls in test files - Add error checks for CreateFormFile() and Write() calls - Fix golangci-lint Go 1.25 compatibility by installing from source
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"jd-book-uploader/config"
|
|
"jd-book-uploader/handlers"
|
|
"jd-book-uploader/middleware"
|
|
"jd-book-uploader/services"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func main() {
|
|
// Load configuration
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil {
|
|
log.Fatalf("Failed to load configuration: %v", err)
|
|
}
|
|
|
|
// Initialize database connection
|
|
_, err = services.NewDBPool(cfg)
|
|
if err != nil {
|
|
log.Fatalf("Failed to connect to database: %v", err)
|
|
}
|
|
// Note: CloseDB is called explicitly in graceful shutdown, not in defer
|
|
log.Println("Database connected successfully")
|
|
|
|
// Initialize Firebase
|
|
_, err = services.InitFirebase(cfg)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize Firebase: %v", err)
|
|
}
|
|
log.Println("Firebase initialized successfully")
|
|
|
|
// Create Fiber app
|
|
app := fiber.New(fiber.Config{
|
|
AppName: "JD Book Uploader API",
|
|
})
|
|
|
|
// Register middleware
|
|
app.Use(middleware.RecoverHandler()) // Recover from panics
|
|
app.Use(middleware.ErrorHandler) // Global error handler
|
|
app.Use(middleware.SetupCORS(cfg)) // CORS
|
|
app.Use(middleware.SetupLogger()) // Request logging
|
|
|
|
// Register routes
|
|
api := app.Group("/api")
|
|
api.Get("/health", handlers.HealthCheck)
|
|
api.Post("/books", handlers.UploadBook)
|
|
api.Post("/stationery", handlers.UploadStationery)
|
|
|
|
// Start server
|
|
port := cfg.Port
|
|
if port == "" {
|
|
port = "8080"
|
|
}
|
|
|
|
// Graceful shutdown
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
log.Printf("Server starting on port %s", port)
|
|
if err := app.Listen(":" + port); err != nil {
|
|
log.Fatalf("Server failed to start: %v", err)
|
|
}
|
|
}()
|
|
|
|
<-c
|
|
log.Println("Gracefully shutting down...")
|
|
|
|
// Close database connection
|
|
if err := services.CloseDB(); err != nil {
|
|
log.Printf("Error closing database: %v", err)
|
|
}
|
|
|
|
// Shutdown Fiber app
|
|
if err := app.Shutdown(); err != nil {
|
|
log.Fatalf("Server shutdown failed: %v", err)
|
|
}
|
|
|
|
log.Println("Server stopped")
|
|
}
|