72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
|
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
|
|||
|
|
}
|