69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"legalgo-BE-go/config"
|
|
"legalgo-BE-go/database"
|
|
repository "legalgo-BE-go/internal/accessor"
|
|
internalhttp "legalgo-BE-go/internal/api/http"
|
|
pkgconfig "legalgo-BE-go/internal/config"
|
|
"legalgo-BE-go/internal/services"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/joho/godotenv"
|
|
"go.uber.org/fx"
|
|
)
|
|
|
|
func init() {
|
|
// Check if the .env file exists
|
|
if _, err := os.Stat(".env"); err == nil {
|
|
// Load environment variables from .env file if it exists
|
|
if err := godotenv.Load(); err != nil {
|
|
log.Println("Error loading .env file, continuing without it")
|
|
}
|
|
} else {
|
|
log.Println(".env file not found, skipping load")
|
|
}
|
|
|
|
config.InitEnv()
|
|
}
|
|
|
|
func run(lc fx.Lifecycle, db *database.DB, apiRouter chi.Router) {
|
|
lc.Append(fx.Hook{
|
|
OnStart: func(ctx context.Context) error {
|
|
fmt.Println("Application has started...")
|
|
pkgconfig.Router(apiRouter)
|
|
|
|
return nil
|
|
},
|
|
OnStop: func(ctx context.Context) error {
|
|
fmt.Println("Shutting down...")
|
|
return nil
|
|
},
|
|
})
|
|
}
|
|
|
|
func main() {
|
|
app := fx.New(
|
|
config.Module,
|
|
database.Module,
|
|
repository.Module,
|
|
services.Module,
|
|
internalhttp.Module,
|
|
fx.Invoke(run),
|
|
)
|
|
|
|
if err := app.Start(context.Background()); err != nil {
|
|
log.Fatal("Error starting app:", err)
|
|
}
|
|
|
|
// Wait for the app to stop gracefully
|
|
if err := app.Stop(context.Background()); err != nil {
|
|
log.Fatal("Error stopping app:", err)
|
|
}
|
|
}
|