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)
}
}

View File

@@ -12,6 +12,7 @@ require (
require (
github.com/ClickHouse/ch-go v0.69.0 // indirect
github.com/andybalholm/brotli v1.2.0 // indirect
github.com/bwmarrin/snowflake v0.3.0 // indirect
github.com/cespare/xxhash/v2 v2.3.0 // indirect
github.com/decred/dcrd/dcrec/secp256k1/v4 v4.4.0 // indirect
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect

View File

@@ -4,6 +4,8 @@ github.com/ClickHouse/clickhouse-go/v2 v2.42.0 h1:MdujEfIrpXesQUH0k0AnuVtJQXk6RZ
github.com/ClickHouse/clickhouse-go/v2 v2.42.0/go.mod h1:riWnuo4YMVdajYll0q6FzRBomdyCrXyFY3VXeXczA8s=
github.com/andybalholm/brotli v1.2.0 h1:ukwgCxwYrmACq68yiUqwIWnGY0cTPox/M94sVwToPjQ=
github.com/andybalholm/brotli v1.2.0/go.mod h1:rzTDkvFWvIrjDXZHkuS16NPggd91W3kUSvPlQ1pLaKY=
github.com/bwmarrin/snowflake v0.3.0 h1:xm67bEhkKh6ij1790JB83OujPR5CzNe8QuQqAgISZN0=
github.com/bwmarrin/snowflake v0.3.0/go.mod h1:NdZxfVWX+oR6y2K0o6qAYv6gIOP9rjG0/E9WsDpxqwE=
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=

View File

@@ -0,0 +1,49 @@
package logger
import (
"log/slog"
"os"
"strings"
)
// Config holds the logger configuration
type Config struct {
ServiceName string
Environment string // "dev", "local", "production"
}
// Init initializes the global logger with slog.
// It detects the environment to switch between TextHandler (dev) and JSONHandler (prod).
func Init(cfg Config) {
var handler slog.Handler
opts := &slog.HandlerOptions{
// Default level
Level: slog.LevelInfo,
// Customize attributes (Time format)
ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
if a.Key == slog.TimeKey {
return slog.String(a.Key, a.Value.Time().Format("2006-01-02 15:04:05"))
}
return a
},
}
// Adjust level and format based on environment
env := strings.ToLower(cfg.Environment)
if env == "dev" || env == "local" || env == "development" {
opts.Level = slog.LevelDebug
handler = slog.NewTextHandler(os.Stdout, opts)
} else {
// Production defaults to JSON
handler = slog.NewJSONHandler(os.Stdout, opts)
}
// Create logger with common attributes
logger := slog.New(handler).With(
slog.String("svc", cfg.ServiceName),
)
// Set as global default logger
slog.SetDefault(logger)
}