feat(log): 新建 logger 包,基于 log/slog 提供结构化日志
- 新建 internal/logger/logger.go,提供 Init/FromCtx/WithRequestID 方法 - Config 新增 LogConfig(level/format),支持 GEN2D_LOG_LEVEL/GEN2D_LOG_FORMAT 环境变量 - config.yml 和 .env.example 添加日志配置示例
This commit is contained in:
@@ -0,0 +1,71 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log/slog"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type ctxKey struct{}
|
||||
|
||||
// L 全局 logger,Init 之前使用默认 text 输出。
|
||||
var L *slog.Logger
|
||||
|
||||
func init() {
|
||||
L = slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug}))
|
||||
}
|
||||
|
||||
// Init 初始化全局 logger。
|
||||
// level: debug / info / warn / error
|
||||
// format: text / json
|
||||
func Init(level, format string) {
|
||||
var lvl slog.Level
|
||||
switch strings.ToLower(level) {
|
||||
case "debug":
|
||||
lvl = slog.LevelDebug
|
||||
case "warn":
|
||||
lvl = slog.LevelWarn
|
||||
case "error":
|
||||
lvl = slog.LevelError
|
||||
default:
|
||||
lvl = slog.LevelInfo
|
||||
}
|
||||
|
||||
opts := &slog.HandlerOptions{Level: lvl}
|
||||
|
||||
var h slog.Handler
|
||||
switch strings.ToLower(format) {
|
||||
case "json":
|
||||
h = slog.NewJSONHandler(os.Stdout, opts)
|
||||
default:
|
||||
h = slog.NewTextHandler(os.Stdout, opts)
|
||||
}
|
||||
|
||||
L = slog.New(h)
|
||||
slog.SetDefault(L)
|
||||
}
|
||||
|
||||
// With 创建带附加字段的 logger。
|
||||
func With(args ...any) *slog.Logger {
|
||||
return L.With(args...)
|
||||
}
|
||||
|
||||
// IntoCtx 将 logger 存入 context。
|
||||
func IntoCtx(ctx context.Context, l *slog.Logger) context.Context {
|
||||
return context.WithValue(ctx, ctxKey{}, l)
|
||||
}
|
||||
|
||||
// FromCtx 从 context 取出 logger,不存在时返回全局 L。
|
||||
func FromCtx(ctx context.Context) *slog.Logger {
|
||||
if l, ok := ctx.Value(ctxKey{}).(*slog.Logger); ok {
|
||||
return l
|
||||
}
|
||||
return L
|
||||
}
|
||||
|
||||
// WithRequestID 创建带 request_id 的 logger 并存入 context。
|
||||
func WithRequestID(ctx context.Context, requestID string) (context.Context, *slog.Logger) {
|
||||
l := L.With("request_id", requestID)
|
||||
return IntoCtx(ctx, l), l
|
||||
}
|
||||
Reference in New Issue
Block a user