zlog

zlog

基于slog，主要扩展了TextHandler
实现了一个用于流式日志的SafeWriter，适用于长日志（比如gorm）

✅ 1. 支持 attrsReplace 拦截所有字段

• 级别显示（比如用 emoji 🟢/🟡/🔴）
• 消息内容（比如脱敏）
• 源码路径（比如只显示包名）

✅ 2. 支持智能颜色控制, 终端有颜色，文件无颜色
✅ 3. 支持 TimeFormat 配置时间格式
✅ 4. 精确的调用栈跳过设置
✅ 5. 完全兼容 slog 的 Group 语义
✅ 6. 流式日志的SafeWriter

基于slog

go get github.com/zohu/zlog

type Options struct {
	Level       slog.Leveler // 日志等级
	Color       Color        // 配置颜色
	TimeFormat  string       // 时间格式
	AttrReplace func(groups []string, attr slog.Attr) slog.Attr // 属性替换
	Stack       Stack       // 显示堆栈
	StackSkip   int         // 堆栈跳过
	Writer      io.Writer   // 日志写入 默认os.Stdout
}



// 修改配置
WithOptions(options *Options)

// 使用
Debug("hello world", "name", "zhang")
Info("hello world", "name", "zhang")
Warn("hello world", "name", "zhang")
Error("hello world", "name", "zhang")

log1 := NewZLogger(&Options{
    Level:      slog.LevelDebug,
    Color:      ColorAlways,
    TimeFormat: time.RFC3339,
})

log1.Debugf("hello world name=%s", "zhang")
log1.Infof("hello world name=%s", "zhang")
log1.Warnf("hello world name=%s", "zhang")
log1.Errorf("hello world name=%s", "zhang")

log2 := NewZLogger(&Options{
    Level:      slog.LevelDebug,
    Color:      ColorAlways,
    TimeFormat: time.RFC3339,
    Stack:      StackOff,
    Writer:     SafeWriter(),
})
log2.Info("hello ... world") // >= 64k
time.Sleep(time.Second)

// 写入日志文件
writer, err := NewTimeRotatingWriter("log", WithConsole(true))
if err != nil {
    t.Fatal(err)
    return
}
log3 := NewZLogger(&Options{
    Writer: writer,
})

// 流式
log3 := NewZLogger(&Options{
    Writer: SafeWriter(writer),
})

// 流式日志、长日志、gorm、或者遇到日志 token too long 的场景
log2 := NewZLogger(&Options{
    Writer: SafeWriter(), // 默认是os.Stdout，替换为 SafeWriter(/*os.Stdout*/)
})

// 比如 gorm 框架的日志
type Logger struct {
    *zlog.Logger
    ignoreRecordNotFound bool
    logSlow              time.Duration
}

func NewLogger(o *Options) *Logger {
    options := &zlog.Options{
        StackSkip: -1,
		Writer: SafeWriter(), // 默认是os.Stdout，替换为 SafeWriter(/*os.Stdout*/)
	}
    if o.Debug != nil && *o.Debug {
        options.Level = slog.LevelDebug
    }
    l := &Logger{
        Logger:               zlog.NewZLogger(options),
        ignoreRecordNotFound: o.LogIgnoreNotFound == "yes",
        logSlow:              o.LogSlow,
    }
    return l
}

func (l Logger) LogMode(level logger.LogLevel) logger.Interface {
    return l
}
func (l Logger) Info(ctx context.Context, s string, i ...interface{}) {
    l.Infof(s, i...)
}
func (l Logger) Warn(ctx context.Context, s string, i ...interface{}) {
    l.Warnf(s, i...)
}
func (l Logger) Error(ctx context.Context, s string, i ...interface{}) {
    l.Errorf(s, i...)
}
func (l Logger) Trace(ctx context.Context, begin time.Time, fc func() (string, int64), err error) {
    elapsed := time.Since(begin)
    switch {
    case err != nil && (!l.ignoreRecordNotFound || !errors.Is(err, gorm.ErrRecordNotFound)):
        sql, rows := fc()
        l.Errorf("rows=%d elapsed=%.3fs err=%s sql=%s", rows, elapsed.Seconds(), err.Error(), sql)
    case l.logSlow != 0 && elapsed > l.logSlow:
        sql, rows := fc()
        var e string
        if err != nil {
            e = fmt.Sprintf("err=%s ", err.Error())
        }
        l.Warnf("rows=%d elapsed=%.3fs %ssql=%s", rows, elapsed.Seconds(), e, sql)
    default:
        sql, rows := fc()
        var e string
        if err != nil {
            e = fmt.Sprintf("err=%s ", err.Error())
        }
        l.Debugf("rows=%d elapsed=%.3fs %ssql=%s", rows, elapsed.Seconds(), e, sql)
    }
}

2025-10-17 01:22:06 INF zlog/zlog_test.go:11 hello world name=zhang
2025-10-17 01:22:06 WRN zlog/zlog_test.go:12 hello world name=zhang
2025-10-17 01:22:06 ERR zlog/zlog_test.go:13 hello world name=zhang
2025-10-17T01:22:06+08:00 DBG zlog/zlog_test.go:21 hello world name=zhang
2025-10-17T01:22:06+08:00 INF zlog/zlog_test.go:22 hello world name=zhang
2025-10-17T01:22:06+08:00 WRN zlog/zlog_test.go:23 hello world name=zhang
2025-10-17T01:22:06+08:00 ERR zlog/zlog_test.go:24 hello world name=zhang
2025-10-17T01:22:06+08:00 INF hello world