56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/sirupsen/logrus"
|
|
"os"
|
|
"os/signal"
|
|
"penahub.gitlab.yandexcloud.net/backend/penahub_common/privilege/client"
|
|
"penahub.gitlab.yandexcloud.net/backend/penahub_common/privilege/config"
|
|
"penahub.gitlab.yandexcloud.net/backend/penahub_common/privilege/controller"
|
|
)
|
|
|
|
func main() {
|
|
logger := logrus.New()
|
|
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil {
|
|
logger.Fatal("Error loading configuration:", err)
|
|
}
|
|
|
|
hubAdminClient := client.NewHubAdminClient(&client.HubAdminClientDeps{
|
|
Logger: logger,
|
|
URLs: &config.AdminMicroServiceURL{
|
|
GetPrivilege: cfg.AdminGetPrivilege,
|
|
PostPrivilege: cfg.AdminPostPrivilege,
|
|
},
|
|
})
|
|
|
|
privilegeController := controller.NewPrivilegeController(logger, hubAdminClient)
|
|
|
|
app := fiber.New()
|
|
|
|
app.Post("/check", privilegeController.Check)
|
|
|
|
go gracefulShutdown(app, logger)
|
|
|
|
err = app.Listen(cfg.HTTPHost + ":" + cfg.HTTPPort)
|
|
if err != nil {
|
|
logger.Fatal("Error starting Fiber app:", err)
|
|
}
|
|
}
|
|
|
|
func gracefulShutdown(app *fiber.App, logger *logrus.Logger) {
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt, os.Kill)
|
|
|
|
<-c
|
|
logger.Info("Shutting down...")
|
|
|
|
if err := app.Shutdown(); err != nil {
|
|
logger.Error("Graceful shutdown failed:", err)
|
|
} else {
|
|
logger.Info("Graceful shutdown completed")
|
|
}
|
|
}
|