package main import ( "context" "fmt" "log" "net/http" "os/signal" "syscall" "time" "api-service/internal/server" "github.com/joho/godotenv" // Import the godotenv package _ "api-service/docs" ) // @title API Service // @version 1.0.0 // @description A comprehensive Go API service with Swagger documentation // @termsOfService http://swagger.io/terms/ // @contact.name API Support // @contact.url http://www.swagger.io/support // @contact.email support@swagger.io // @license.name Apache 2.0 // @license.url http://www.apache.org/licenses/LICENSE-2.0.html // @host localhost:8080 // @BasePath /api/v1 // @schemes http https func gracefulShutdown(apiServer *http.Server, done chan bool) { // Create context that listens for the interrupt signal from the OS. ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM) defer stop() // Listen for the interrupt signal. <-ctx.Done() log.Println("shutting down gracefully, press Ctrl+C again to force") stop() // Allow Ctrl+C to force shutdown // The context is used to inform the server it has 5 seconds to finish // the request it is currently handling ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() if err := apiServer.Shutdown(ctx); err != nil { log.Printf("Server forced to shutdown with error: %v", err) } log.Println("Server exiting") // Notify the main goroutine that the shutdown is complete done <- true } func main() { log.Println("Starting API Service...") // Load environment variables from .env file if err := godotenv.Load(); err != nil { log.Printf("Warning: .env file not found or could not be loaded: %v", err) log.Println("Continuing with system environment variables...") } server := server.NewServer() // Create a done channel to signal when the shutdown is complete done := make(chan bool, 1) // Run graceful shutdown in a separate goroutine go gracefulShutdown(server, done) log.Printf("Server starting on port %s", server.Addr) err := server.ListenAndServe() if err != nil && err != http.ErrServerClosed { panic(fmt.Sprintf("http server error: %s", err)) } // Wait for the graceful shutdown to complete <-done log.Println("Graceful shutdown complete.") }