Update main.go

main.go

@@ -12,92 +12,82 @@ import (
 
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
-	"github.com/joho/godotenv" // Optional: for loading .env file
+	"github.com/joho/godotenv" 
 )
 
 func main() {
-	// Load .env file if present (optional, good for local dev)
 	_ = godotenv.Load()
 
-	// Load configuration
 	cfg := LoadConfig()
 
-	// Set Gin mode (release or debug)
 	if cfg.GinMode == "release" {
 		gin.SetMode(gin.ReleaseMode)
 	} else {
 		gin.SetMode(gin.DebugMode)
 	}
+	// Giữ lại: Log khởi động quan trọng
 	log.Printf("Starting Go Proxy Server in %s mode...", gin.Mode())
 
-	// Initialize Gin router
 	router := gin.New()
 
-	// Middleware
-	router.Use(gin.Logger()) // Standard Gin logger
-	router.Use(gin.Recovery()) // Recover from panics
-	// CORS middleware (allow all for simplicity, adjust as needed)
+	// Giữ lại: Gin Logger là tiêu chuẩn, nhưng có thể xem xét loại bỏ nếu muốn log *cực kỳ* ít.
+	// Hiện tại giữ lại để có thông tin request cơ bản.
+	router.Use(gin.Logger()) 
+	router.Use(gin.Recovery()) 
 	router.Use(cors.New(cors.Config{
 		AllowOrigins:     []string{"*"},
 		AllowMethods:     []string{"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"},
-		AllowHeaders:     []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-API-Key"}, // Include X-API-Key
+		AllowHeaders:     []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-API-Key"}, 
 		ExposeHeaders:    []string{"Content-Length"},
 		AllowCredentials: true,
 		MaxAge:           12 * time.Hour,
 	}))
 
-	// --- Routes ---
-	// Health check
 	router.GET("/health", HealthCheckHandler)
 
-	// Main proxy endpoint group
 	v1 := router.Group("/v1")
 	{
-		// Apply API Key Authentication middleware if keys are configured
 		if len(cfg.ValidAPIKeys) > 0 {
+			// Giữ lại: Log khởi động quan trọng
 			log.Printf("API Key authentication enabled (%d keys configured).", len(cfg.ValidAPIKeys))
 			v1.Use(APIKeyAuthMiddleware(cfg.ValidAPIKeys))
 		} else {
+			// Giữ lại: Log cảnh báo quan trọng (cũng có trong config.go, nhưng ở đây cũng tốt)
 			log.Println("WARN: No PROXY_API_KEYS configured. Proxy is open (no authentication).")
 		}
 		v1.POST("/messages", MessagesHandler)
 	}
 
-	// --- Server Setup ---
 	server := &http.Server{
 		Addr:    fmt.Sprintf(":%s", cfg.Port),
 		Handler: router,
-		// Add timeouts for production hardening
 		ReadTimeout:  10 * time.Second,
-		WriteTimeout: cfg.ReadTimeout + 30*time.Second, // Ensure write timeout is longer than read timeout for streaming
+		WriteTimeout: cfg.ReadTimeout + 30*time.Second, 
 		IdleTimeout:  120 * time.Second,
 	}
 
-	// --- Graceful Shutdown ---
-	// Run server in a goroutine so it doesn't block
 	go func() {
+		// Giữ lại: Log khởi động quan trọng
 		log.Printf("Server listening on port %s", cfg.Port)
 		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			// Giữ lại: Log lỗi nghiêm trọng
 			log.Fatalf("listen: %s\n", err)
 		}
 	}()
 
-	// Wait for interrupt signal to gracefully shut down the server
 	quit := make(chan os.Signal, 1)
-	// kill (no param) default send syscall.SIGTERM
-	// kill -2 is syscall.SIGINT
-	// kill -9 is syscall.SIGKILL but can't be caught, so don't need to add it
 	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
 	<-quit
+	// Giữ lại: Log tắt server
 	log.Println("Shutting down server...")
 
-	// The context is used to inform the server it has 5 seconds to finish
-	// the requests it is currently handling
 	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 	if err := server.Shutdown(ctx); err != nil {
+		// Giữ lại: Log lỗi nghiêm trọng
 		log.Fatal("Server forced to shutdown:", err)
 	}
 
+	// Giữ lại: Log tắt server
 	log.Println("Server exiting")
-}
+}