forked from baron/baron-sso
논리 검사 계속. 스케폴딩 일부 진행
This commit is contained in:
@@ -2,15 +2,44 @@ package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
|
||||
"baron-sso-backend/internal/handler"
|
||||
"baron-sso-backend/internal/repository"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/gofiber/fiber/v2/middleware/cors"
|
||||
"github.com/gofiber/fiber/v2/middleware/encryptcookie"
|
||||
"github.com/gofiber/fiber/v2/middleware/logger"
|
||||
"github.com/gofiber/fiber/v2/middleware/recover"
|
||||
)
|
||||
|
||||
func getEnv(key, fallback string) string {
|
||||
if value, ok := os.LookupEnv(key); ok {
|
||||
return value
|
||||
}
|
||||
return fallback
|
||||
}
|
||||
|
||||
func main() {
|
||||
// Initialize Fiber
|
||||
// 1. Initialize DB Connections
|
||||
chHost := getEnv("CLICKHOUSE_HOST", "localhost")
|
||||
chPort, _ := strconv.Atoi(getEnv("CLICKHOUSE_PORT_NATIVE", "9000"))
|
||||
chUser := getEnv("CLICKHOUSE_USER", "default")
|
||||
chPass := getEnv("CLICKHOUSE_PASSWORD", "")
|
||||
chDB := getEnv("CLICKHOUSE_DB", "default")
|
||||
|
||||
auditRepo, err := repository.NewClickHouseRepository(chHost, chPort, chUser, chPass, chDB)
|
||||
if err != nil {
|
||||
log.Printf("Warning: Failed to connect to ClickHouse: %v. Audit logs will fail.", err)
|
||||
// Proceeding mostly for Dev purposes, but in Prod should generally fail or fallback.
|
||||
}
|
||||
|
||||
// 2. Initialize Handlers
|
||||
auditHandler := handler.NewAuditHandler(auditRepo)
|
||||
|
||||
// 3. Initialize Fiber
|
||||
app := fiber.New(fiber.Config{
|
||||
AppName: "Baron SSO Backend",
|
||||
})
|
||||
@@ -18,8 +47,9 @@ func main() {
|
||||
// Middleware
|
||||
app.Use(logger.New())
|
||||
app.Use(recover.New())
|
||||
app.Use(cors.New()) // Allow Frontend Access
|
||||
app.Use(encryptcookie.New(encryptcookie.Config{
|
||||
Key: "secret-key-must-be-32-bytes-long!", // TODO: Externalize to env
|
||||
Key: getEnv("COOKIE_SECRET", "secret-key-must-be-32-bytes-long!"),
|
||||
}))
|
||||
|
||||
// Routes
|
||||
@@ -27,13 +57,15 @@ func main() {
|
||||
return c.SendString("Baron SSO Audit Backend Online")
|
||||
})
|
||||
|
||||
// Health Check
|
||||
app.Get("/health", func(c *fiber.Ctx) error {
|
||||
return c.JSON(fiber.Map{
|
||||
"status": "ok",
|
||||
})
|
||||
return c.JSON(fiber.Map{"status": "ok"})
|
||||
})
|
||||
|
||||
// API Group
|
||||
api := app.Group("/api/v1")
|
||||
api.Post("/audit", auditHandler.CreateLog)
|
||||
|
||||
// Start Server
|
||||
log.Fatal(app.Listen(":3000"))
|
||||
port := getEnv("PORT", "3000")
|
||||
log.Fatal(app.Listen(":" + port))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user