150 lines
3.4 KiB
Go
150 lines
3.4 KiB
Go
// Package logx 分级日志（debug/info/warn/error），输出 stderr 与可选文件。
|
||
package logx
|
||
|
||
import (
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// Level 日志级别。
|
||
type Level int
|
||
|
||
const (
|
||
// LevelDebug 调试。
|
||
LevelDebug Level = iota
|
||
// LevelInfo 常规信息。
|
||
LevelInfo
|
||
// LevelWarn 警告。
|
||
LevelWarn
|
||
// LevelError 错误。
|
||
LevelError
|
||
)
|
||
|
||
const defaultRing = 8
|
||
|
||
// Logger 线程安全分级日志器。
|
||
type Logger struct {
|
||
mu sync.Mutex
|
||
level Level
|
||
w io.Writer
|
||
closer io.Closer
|
||
ring []string
|
||
ringCap int
|
||
}
|
||
|
||
// New 创建日志器；file 为空仅输出 stderr，否则 stderr + 文件。
|
||
func New(file string, level Level) (*Logger, error) {
|
||
return newLogger(file, level, true, defaultRing)
|
||
}
|
||
|
||
// NewFile 仅写文件（GUI 客户端：无 CMD，构建与发布.md §1）。file 为空则丢弃。
|
||
func NewFile(file string, level Level) (*Logger, error) {
|
||
return newLogger(file, level, false, defaultRing)
|
||
}
|
||
|
||
func newLogger(file string, level Level, stderr bool, ringCap int) (*Logger, error) {
|
||
var writers []io.Writer
|
||
if stderr {
|
||
writers = append(writers, os.Stderr)
|
||
}
|
||
if file != "" {
|
||
if err := os.MkdirAll(pathDir(file), 0o755); err != nil {
|
||
return nil, fmt.Errorf("logx.New mkdir: %w", err)
|
||
}
|
||
f, err := os.OpenFile(file, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o644)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("logx.New %s: %w", file, err)
|
||
}
|
||
writers = append(writers, f)
|
||
}
|
||
w := io.Writer(io.Discard)
|
||
switch len(writers) {
|
||
case 1:
|
||
w = writers[0]
|
||
case 2:
|
||
w = io.MultiWriter(writers[0], writers[1])
|
||
}
|
||
if ringCap < 1 {
|
||
ringCap = defaultRing
|
||
}
|
||
var closer io.Closer
|
||
if file != "" {
|
||
if f, ok := writers[len(writers)-1].(*os.File); ok {
|
||
closer = f
|
||
}
|
||
}
|
||
return &Logger{level: level, w: w, closer: closer, ringCap: ringCap}, nil
|
||
}
|
||
|
||
// Close 关闭日志文件（若有）。
|
||
func (l *Logger) Close() error {
|
||
if l == nil || l.closer == nil {
|
||
return nil
|
||
}
|
||
return l.closer.Close()
|
||
}
|
||
|
||
// Recent 返回最近若干行（旧→新，供 Loading 左下角）。
|
||
func (l *Logger) Recent() []string {
|
||
if l == nil {
|
||
return nil
|
||
}
|
||
l.mu.Lock()
|
||
defer l.mu.Unlock()
|
||
out := make([]string, len(l.ring))
|
||
copy(out, l.ring)
|
||
return out
|
||
}
|
||
|
||
// pathDir 返回路径的目录部分（"a.log" → "."）。
|
||
func pathDir(p string) string {
|
||
for i := len(p) - 1; i >= 0; i-- {
|
||
if p[i] == '/' || p[i] == '\\' {
|
||
if i == 0 {
|
||
return p[:1]
|
||
}
|
||
return p[:i]
|
||
}
|
||
}
|
||
return "."
|
||
}
|
||
|
||
func (l *Logger) logf(lv Level, tag, format string, args ...any) {
|
||
if l == nil || lv < l.level {
|
||
return
|
||
}
|
||
msg := fmt.Sprintf(format, args...)
|
||
line := fmt.Sprintf("[%s] [%s] %s", time.Now().Format("15:04:05.000"), tag, msg)
|
||
l.mu.Lock()
|
||
defer l.mu.Unlock()
|
||
fmt.Fprintln(l.w, line)
|
||
l.pushRingLocked(msg)
|
||
}
|
||
|
||
func (l *Logger) pushRingLocked(s string) {
|
||
if l.ringCap <= 0 {
|
||
return
|
||
}
|
||
if len(l.ring) < l.ringCap {
|
||
l.ring = append(l.ring, s)
|
||
return
|
||
}
|
||
copy(l.ring, l.ring[1:])
|
||
l.ring[l.ringCap-1] = s
|
||
}
|
||
|
||
// Debugf 调试输出。
|
||
func (l *Logger) Debugf(format string, args ...any) { l.logf(LevelDebug, "DEBUG", format, args...) }
|
||
|
||
// Infof 常规输出。
|
||
func (l *Logger) Infof(format string, args ...any) { l.logf(LevelInfo, "INFO", format, args...) }
|
||
|
||
// Warnf 警告输出。
|
||
func (l *Logger) Warnf(format string, args ...any) { l.logf(LevelWarn, "WARN", format, args...) }
|
||
|
||
// Errorf 错误输出。
|
||
func (l *Logger) Errorf(format string, args ...any) { l.logf(LevelError, "ERROR", format, args...) }
|