diff --git a/main.go b/main.go index 442e8a5..bfe0cf7 100755 --- a/main.go +++ b/main.go @@ -24,8 +24,13 @@ var PublicFS embed.FS func main() { // Load environment variables - if err := godotenv.Load(".env"); err != nil { - log.Fatal("Error loading .env file") + err := godotenv.Load(".env") + if err != nil { + // If loading .env fails, try loading stack.env + err = godotenv.Load("stack.env") + if err != nil { + log.Fatal("Error loading environment file") + } } // Initialize the database connection pool