1
0
forked from baron/baron-sso

로그 포맷 통일, slog 적용

This commit is contained in:
2026-01-15 14:16:34 +09:00
parent 2aff11bc5d
commit 5dd2c94555
9 changed files with 286 additions and 32 deletions

View File

@@ -1,17 +1,19 @@
package main
import (
"log"
"log/slog"
"os"
"strconv"
"baron-sso-backend/internal/handler"
"strconv"
"time"
"github.com/bwmarrin/snowflake"
"baron-sso-backend/internal/handler"
"baron-sso-backend/internal/logger"
"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"
"github.com/gofiber/fiber/v2/middleware/requestid"
)
@@ -24,13 +26,25 @@ func getEnv(key, fallback string) string {
}
func main() {
// 0. Initialize Logger
logger.Init(logger.Config{
ServiceName: "baron-sso",
Environment: getEnv("GO_ENV", "dev"),
})
// Initialize Snowflake Node (Node 2 for Baron)
node, err := snowflake.NewNode(2)
if err != nil {
slog.Error("Failed to initialize snowflake node", "error", err)
os.Exit(1)
}
// 1. Log Config on Startup
log.Println("==========================================")
log.Println("Starting Baron SSO Backend...")
log.Printf("FRONTEND_URL: %s", getEnv("FRONTEND_URL", "http://ssologin.hmac.kr"))
log.Printf("REDIS_ADDR: %s", getEnv("REDIS_ADDR", "redis:6379"))
log.Printf("DESCOPE_ID: %s", getEnv("DESCOPE_PROJECT_ID", "not-set"))
log.Println("==========================================")
slog.Info("Starting Baron SSO Backend",
"frontend_url", getEnv("FRONTEND_URL", "http://ssologin.hmac.kr"),
"redis_addr", getEnv("REDIS_ADDR", "redis:6379"),
"descope_id", getEnv("DESCOPE_PROJECT_ID", "not-set"),
)
// 2. Initialize DB Connections
chHost := getEnv("CLICKHOUSE_HOST", "localhost")
@@ -41,8 +55,7 @@ func main() {
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.
slog.Warn("Failed to connect to ClickHouse. Audit logs will fail.", "error", err)
}
// 2. Initialize Handlers
@@ -52,17 +65,42 @@ func main() {
// 3. Initialize Fiber
app := fiber.New(fiber.Config{
AppName: "Baron SSO Backend",
DisableStartupMessage: true, // Clean logs
})
// Middleware
app.Use(requestid.New(requestid.Config{
Generator: func() string {
return handler.GenerateSecureToken(4) // 8 chars hex
return node.Generate().String()
},
}))
app.Use(logger.New(logger.Config{
Format: "[${time}] ${status} - ${method} ${path}\n",
}))
// [Standardized] HTTP Request Logger Middleware using slog
app.Use(func(c *fiber.Ctx) error {
start := time.Now()
// Handle request
err := c.Next()
// Log after request
latency := time.Since(start)
msg := "http_request"
if err != nil {
msg = "http_request_error"
}
slog.Info(msg,
"status", c.Response().StatusCode(),
"method", c.Method(),
"path", c.Path(),
"latency", latency.String(),
"ip", c.IP(),
"req_id", c.GetRespHeader(fiber.HeaderXRequestID),
)
return err
})
app.Use(recover.New())
app.Use(cors.New(cors.Config{
AllowOrigins: "*", // Adjust in production
@@ -100,21 +138,52 @@ func main() {
// Webhook for Descope Generic Email Gateway (Fake Email Strategy)
auth.Post("/webhooks/descope-email", authHandler.HandleDescopeEmailRelay)
// Client Logging Route (For Debugging)
// Client Logging Route (Standardized & Flattened)
api.Post("/client-log", func(c *fiber.Ctx) error {
type LogReq struct {
Level string `json:"level"`
Message string `json:"message"`
Level string `json:"level"`
Message string `json:"message"`
Data map[string]interface{} `json:"data,omitempty"`
}
var req LogReq
if err := c.BodyParser(&req); err != nil {
return c.SendStatus(fiber.StatusBadRequest)
}
log.Printf("[CLIENT-LOG] [%s] %s", req.Level, req.Message)
// Prepare attributes for flattening
attrs := []any{
slog.String("source", "client"),
}
for k, v := range req.Data {
// Skip svc if it's already set by the global logger to avoid confusion,
// or keep it as client_svc
if k == "svc" {
attrs = append(attrs, slog.Any("client_svc", v))
} else {
attrs = append(attrs, slog.Any(k, v))
}
}
// Map and log with correct level
var level slog.Level
switch req.Level {
case "SEVERE", "ERROR":
level = slog.LevelError
case "WARNING", "WARN":
level = slog.LevelWarn
default:
level = slog.LevelInfo
}
slog.Log(c.Context(), level, req.Message, attrs...)
return c.SendStatus(fiber.StatusOK)
})
// Start Server
port := getEnv("PORT", "3000")
log.Fatal(app.Listen(":" + port))
slog.Info("Server listening", "port", port)
if err := app.Listen(":" + port); err != nil {
slog.Error("Server failed to start", "error", err)
os.Exit(1)
}
}