48 lines
960 B
Go
48 lines
960 B
Go
package logx
|
||
|
||
import (
|
||
"log/slog"
|
||
"os"
|
||
"strings"
|
||
)
|
||
|
||
// Config 定义日志初始化参数。
|
||
type Config struct {
|
||
Level string `yaml:"level"`
|
||
Format string `yaml:"format"`
|
||
AddSource bool `yaml:"addSource"`
|
||
}
|
||
|
||
// Init 初始化全局 slog logger，并设置为默认 logger。
|
||
func Init(cfg Config) *slog.Logger {
|
||
level := parseLevel(cfg.Level)
|
||
opts := &slog.HandlerOptions{
|
||
Level: level,
|
||
AddSource: cfg.AddSource,
|
||
}
|
||
|
||
var handler slog.Handler
|
||
if strings.EqualFold(cfg.Format, "json") {
|
||
handler = slog.NewJSONHandler(os.Stdout, opts)
|
||
} else {
|
||
handler = slog.NewTextHandler(os.Stdout, opts)
|
||
}
|
||
|
||
logger := slog.New(handler)
|
||
slog.SetDefault(logger)
|
||
return logger
|
||
}
|
||
|
||
func parseLevel(level string) slog.Level {
|
||
switch strings.ToLower(strings.TrimSpace(level)) {
|
||
case "debug":
|
||
return slog.LevelDebug
|
||
case "warn", "warning":
|
||
return slog.LevelWarn
|
||
case "error":
|
||
return slog.LevelError
|
||
default:
|
||
return slog.LevelInfo
|
||
}
|
||
}
|