杭州明婳科技
381 lines
12 KiB
Go
381 lines
12 KiB
Go
package logger
|
||
|
||
import (
|
||
"errors"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"sync"
|
||
"sync/atomic"
|
||
|
||
"github.com/EthanCodeCraft/xlgo-core/config"
|
||
|
||
"go.uber.org/zap"
|
||
"go.uber.org/zap/zapcore"
|
||
"gopkg.in/natefinch/lumberjack.v2"
|
||
)
|
||
|
||
var (
|
||
// 内部 atomic 存储——四个 logger 的真实源（H7 修复）。
|
||
// 所有包级函数（Info/Debug/.../APILog/DBLog/Sync）经 atomic Load 读取，
|
||
// 使请求 goroutine 不与 Init/Close 重新装配全局变量竞争。
|
||
// 原实现用 m.mu（实例锁）保护包级全局变量，锁与被保护对象作用域错配，
|
||
// 读侧无锁裸读 → 热重载 re-Init/Close 与请求日志存在数据竞争。
|
||
loggerPtr atomic.Pointer[zap.Logger]
|
||
sugarPtr atomic.Pointer[zap.SugaredLogger]
|
||
apiLogPtr atomic.Pointer[zap.Logger]
|
||
dbLogPtr atomic.Pointer[zap.Logger]
|
||
|
||
// Logger 全局通用日志实例（兼容别名）。Init 之前为 Nop，调用安全。
|
||
//
|
||
// 兼容性说明（H7）：此导出变量由 Init/Close 在 m.mu 下同步维护，但直接读它
|
||
// 在 re-Init/Close 期间非并发安全；框架内部读路径已改走 atomic（currentLogger）。
|
||
// 并发安全访问请用包级 Info/Debug/Warn/Error/... 函数，不要直接读此变量。
|
||
Logger = zap.NewNop()
|
||
|
||
// fileWriters 持有所有 lumberjack 实例引用，
|
||
// Close() 调用时显式释放文件句柄。
|
||
// 必要性：lumberjack 不依赖 GC 关闭文件，进程长跑或测试场景下
|
||
// 不显式关闭会持有句柄导致 Windows 上无法删除日志目录。
|
||
// 仅在 m.mu 下访问（Init/Close/closeFileWriters），无读侧竞争。
|
||
fileWriters []*lumberjack.Logger
|
||
)
|
||
|
||
func init() {
|
||
// 初始化 atomic 存储为 Nop，保证包级函数在任何时刻 Load 均非 nil。
|
||
nop := zap.NewNop()
|
||
loggerPtr.Store(nop)
|
||
sugarPtr.Store(nop.Sugar())
|
||
apiLogPtr.Store(nop)
|
||
dbLogPtr.Store(nop)
|
||
}
|
||
|
||
// currentLogger 返回通用 logger 的 atomic 快照（永不 nil）。
|
||
func currentLogger() *zap.Logger {
|
||
if l := loggerPtr.Load(); l != nil {
|
||
return l
|
||
}
|
||
return zap.NewNop() // 防御：init 后不可达
|
||
}
|
||
|
||
// currentSugar 返回 sugared logger 的 atomic 快照（永不 nil）。
|
||
func currentSugar() *zap.SugaredLogger {
|
||
if s := sugarPtr.Load(); s != nil {
|
||
return s
|
||
}
|
||
return zap.NewNop().Sugar() // 防御：init 后不可达
|
||
}
|
||
|
||
// currentAPILog 返回 API logger 的 atomic 快照（永不 nil）。
|
||
func currentAPILog() *zap.Logger {
|
||
if l := apiLogPtr.Load(); l != nil {
|
||
return l
|
||
}
|
||
return zap.NewNop()
|
||
}
|
||
|
||
// currentDBLog 返回 DB logger 的 atomic 快照（永不 nil）。
|
||
func currentDBLog() *zap.Logger {
|
||
if l := dbLogPtr.Load(); l != nil {
|
||
return l
|
||
}
|
||
return zap.NewNop()
|
||
}
|
||
|
||
// LogManager 日志管理器（#10）。照 database.Manager 模式：
|
||
// 实例化 + DefaultLogger 全局默认 + 包级 facade 代理。
|
||
// 包级 Logger/sugar/apiLog/dbLog 由 Init 同步维护，下游 8 个包零改动。
|
||
type LogManager struct {
|
||
mu sync.Mutex
|
||
cfg *config.Config
|
||
// level 是所有 core 共享的 AtomicLevel，支持运行期 SetLevel 热切换（M19）。
|
||
// Init 前为 nil，SetLevel 守卫之。
|
||
level zap.AtomicLevel
|
||
}
|
||
|
||
// DefaultLogger 默认日志管理器，包级 facade 代理到它。
|
||
var DefaultLogger = NewLogManager()
|
||
|
||
// NewLogManager 创建日志管理器实例。
|
||
func NewLogManager() *LogManager { return &LogManager{} }
|
||
|
||
// SetLevel 运行期热切换日志级别（M19）。需先 Init；未 Init 时返回 false。
|
||
// 影响所有 core（app/api/db/console）——它们共享同一 AtomicLevel。
|
||
func (m *LogManager) SetLevel(l zapcore.Level) bool {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
if m.level == (zap.AtomicLevel{}) {
|
||
return false
|
||
}
|
||
m.level.SetLevel(l)
|
||
return true
|
||
}
|
||
|
||
// GetLevel 返回当前日志级别（未 Init 返回 InfoLevel）。
|
||
func (m *LogManager) GetLevel() zapcore.Level {
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
if m.level == (zap.AtomicLevel{}) {
|
||
return zapcore.InfoLevel
|
||
}
|
||
return m.level.Level()
|
||
}
|
||
|
||
// SetLevel 包级 facade：运行期热切换默认日志级别（M19）。未 Init 时无操作返回 false。
|
||
func SetLevel(l zapcore.Level) bool { return DefaultLogger.SetLevel(l) }
|
||
|
||
// GetLevel 包级 facade：返回默认日志当前级别。
|
||
func GetLevel() zapcore.Level { return DefaultLogger.GetLevel() }
|
||
|
||
// SetDefaultLogManager 提升指定 LogManager 为全局默认。
|
||
func SetDefaultLogManager(m *LogManager) {
|
||
if m != nil {
|
||
DefaultLogger = m
|
||
}
|
||
}
|
||
|
||
// Init 初始化日志。
|
||
//
|
||
// 三个 logger 的分流策略：
|
||
// - Logger（通用）：写 console + logs/app.log
|
||
// - APILog() ：写 console + logs/api.log
|
||
// - DBLog() ：写 console + logs/database.log
|
||
//
|
||
// 关键修复（v1.0.3）：旧实现把 apiCore 和 dbCore 都 Tee 进通用 Logger，
|
||
// 导致每条 logger.Info(...) 都会同时落到 api.log + database.log + console
|
||
// 三份，磁盘占用翻倍且分流形同虚设。新实现通用 Logger 只走独立的 app.log。
|
||
func (m *LogManager) Init(cfg *config.Config) error {
|
||
if cfg == nil {
|
||
return errors.New("logger: 配置为空")
|
||
}
|
||
|
||
// 确保日志目录存在（0750：owner+group 可访问，与 storage 目录权限一致）
|
||
if err := os.MkdirAll(cfg.Log.Dir, 0o750); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 日志编码器配置
|
||
encoderConfig := zapcore.EncoderConfig{
|
||
TimeKey: "time",
|
||
LevelKey: "level",
|
||
NameKey: "logger",
|
||
CallerKey: "caller",
|
||
FunctionKey: zapcore.OmitKey,
|
||
MessageKey: "msg",
|
||
StacktraceKey: "stacktrace",
|
||
LineEnding: zapcore.DefaultLineEnding,
|
||
EncodeLevel: zapcore.LowercaseLevelEncoder,
|
||
EncodeTime: zapcore.ISO8601TimeEncoder,
|
||
EncodeDuration: zapcore.SecondsDurationEncoder,
|
||
EncodeCaller: zapcore.ShortCallerEncoder,
|
||
}
|
||
|
||
// 根据运行模式设置日志级别（M19：用 AtomicLevel 支持运行期 SetLevel 热切换）
|
||
level := zap.NewAtomicLevelAt(zapcore.DebugLevel)
|
||
if cfg.IsProduction() {
|
||
level = zap.NewAtomicLevelAt(zapcore.InfoLevel)
|
||
}
|
||
// 记录到 manager，供 SetLevel/GetLevel 热切换（H7：在 m.mu 下写入）。
|
||
m.level = level
|
||
|
||
jsonEncoder := zapcore.NewJSONEncoder(encoderConfig)
|
||
consoleEncoder := zapcore.NewConsoleEncoder(encoderConfig)
|
||
consoleCore := zapcore.NewCore(consoleEncoder, zapcore.AddSync(os.Stdout), level)
|
||
|
||
// 通用日志独立文件，避免与 api/db 日志重复写入
|
||
appWriter := newRotatingWriter(cfg.Log, "app.log")
|
||
apiWriter := newRotatingWriter(cfg.Log, "api.log")
|
||
dbWriter := newRotatingWriter(cfg.Log, "database.log")
|
||
|
||
appCore := zapcore.NewCore(jsonEncoder, zapcore.AddSync(appWriter), level)
|
||
apiCore := zapcore.NewCore(jsonEncoder, zapcore.AddSync(apiWriter), level)
|
||
dbCore := zapcore.NewCore(jsonEncoder, zapcore.AddSync(dbWriter), level)
|
||
|
||
// 三个 logger 各写自己的文件 + console，互不 Tee
|
||
newLogger := zap.New(
|
||
zapcore.NewTee(appCore, consoleCore),
|
||
zap.AddCaller(), zap.AddCallerSkip(1),
|
||
)
|
||
newAPILog := zap.New(
|
||
zapcore.NewTee(apiCore, consoleCore),
|
||
zap.AddCaller(), zap.AddCallerSkip(1),
|
||
)
|
||
newDBLog := zap.New(
|
||
zapcore.NewTee(dbCore, consoleCore),
|
||
zap.AddCaller(), zap.AddCallerSkip(1),
|
||
)
|
||
|
||
m.mu.Lock()
|
||
defer m.mu.Unlock()
|
||
// 全部构造成功后再原子替换全局变量，避免半初始化状态。
|
||
// 同时关闭旧 writer 释放句柄（重复 Init 场景，主要服务于测试）。
|
||
// H7：四个 logger 经 atomic.Pointer Store，读侧（请求 goroutine）无锁原子 load；
|
||
// fileWriters 仅在 m.mu 下访问。Logger 兼容别名同步维护。
|
||
closeFileWriters()
|
||
loggerPtr.Store(newLogger)
|
||
sugarPtr.Store(newLogger.Sugar())
|
||
apiLogPtr.Store(newAPILog)
|
||
dbLogPtr.Store(newDBLog)
|
||
Logger = newLogger
|
||
fileWriters = []*lumberjack.Logger{appWriter, apiWriter, dbWriter}
|
||
m.cfg = cfg
|
||
|
||
return nil
|
||
}
|
||
|
||
// Init 包级 facade：初始化日志（代理到 DefaultLogger）。
|
||
func Init(cfg *config.Config) error {
|
||
return DefaultLogger.Init(cfg)
|
||
}
|
||
|
||
// newRotatingWriter 创建带 lumberjack 滚动归档的 writer
|
||
func newRotatingWriter(cfg config.LogConfig, filename string) *lumberjack.Logger {
|
||
return &lumberjack.Logger{
|
||
Filename: filepath.Join(cfg.Dir, filename),
|
||
MaxSize: cfg.MaxSize,
|
||
MaxBackups: cfg.MaxBackups,
|
||
MaxAge: cfg.MaxAge,
|
||
Compress: cfg.Compress,
|
||
}
|
||
}
|
||
|
||
// closeFileWriters 关闭并清空当前持有的 lumberjack writer
|
||
func closeFileWriters() {
|
||
for _, w := range fileWriters {
|
||
if w != nil {
|
||
_ = w.Close()
|
||
}
|
||
}
|
||
fileWriters = nil
|
||
}
|
||
|
||
// Sync 同步全部 logger 缓冲到底层 writer。
|
||
//
|
||
// 注意：在 Windows / 部分 *nix 平台上对 stdout/stderr 调用 Sync 会返回
|
||
// "invalid argument" / "inappropriate ioctl for device"，属于 zap 已知行为，
|
||
// 这里把这类错误识别并忽略，只返回真实的写入失败。
|
||
func (m *LogManager) Sync() error {
|
||
var errs []error
|
||
for _, l := range []*zap.Logger{currentLogger(), currentAPILog(), currentDBLog()} {
|
||
if l == nil {
|
||
continue
|
||
}
|
||
if err := l.Sync(); err != nil && !isHarmlessSyncError(err) {
|
||
errs = append(errs, err)
|
||
}
|
||
}
|
||
return errors.Join(errs...)
|
||
}
|
||
|
||
// Sync 包级 facade：同步全部 logger 缓冲（代理到 DefaultLogger）。
|
||
func Sync() error {
|
||
return DefaultLogger.Sync()
|
||
}
|
||
|
||
// Close 关闭日志文件句柄，重置全局 logger 为 Nop。
|
||
// 通常由 App.Shutdown 在 Sync 之后调用；测试场景需要清理临时目录时也应调用。
|
||
//
|
||
// 调用后再次写日志不会 panic（fall back to nop logger），但不会写入文件。
|
||
// 如需重新启用，请再次调用 Init。
|
||
func (m *LogManager) Close() error {
|
||
syncErr := m.Sync()
|
||
|
||
m.mu.Lock()
|
||
closeFileWriters()
|
||
|
||
// H7：重置为 Nop 经 atomic Store，与读侧一致；Logger 兼容别名同步。
|
||
nop := zap.NewNop()
|
||
loggerPtr.Store(nop)
|
||
sugarPtr.Store(nop.Sugar())
|
||
apiLogPtr.Store(nop)
|
||
dbLogPtr.Store(nop)
|
||
Logger = nop
|
||
m.mu.Unlock()
|
||
|
||
return syncErr
|
||
}
|
||
|
||
// Close 包级 facade：关闭日志文件句柄，重置为 Nop（代理到 DefaultLogger）。
|
||
func Close() error {
|
||
return DefaultLogger.Close()
|
||
}
|
||
|
||
// isHarmlessSyncError 识别 stdout/stderr Sync 在不同平台返回的预期错误。
|
||
// 这些错误来自 console core，对真实 writer 无影响，可安全忽略。
|
||
func isHarmlessSyncError(err error) bool {
|
||
if err == nil {
|
||
return true
|
||
}
|
||
msg := err.Error()
|
||
for _, sub := range []string{
|
||
"invalid argument", // Linux stdout
|
||
"inappropriate ioctl for device", // macOS stdout
|
||
"bad file descriptor", // Windows stdout
|
||
} {
|
||
if strings.Contains(msg, sub) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// Debug 调试日志
|
||
func Debug(msg string, fields ...zap.Field) {
|
||
currentLogger().Debug(msg, fields...)
|
||
}
|
||
|
||
// Info 信息日志
|
||
func Info(msg string, fields ...zap.Field) {
|
||
currentLogger().Info(msg, fields...)
|
||
}
|
||
|
||
// Warn 警告日志
|
||
func Warn(msg string, fields ...zap.Field) {
|
||
currentLogger().Warn(msg, fields...)
|
||
}
|
||
|
||
// Error 错误日志
|
||
func Error(msg string, fields ...zap.Field) {
|
||
currentLogger().Error(msg, fields...)
|
||
}
|
||
|
||
// Fatal 致命错误日志（仅供应用层使用，框架内部禁止调用）
|
||
func Fatal(msg string, fields ...zap.Field) {
|
||
currentLogger().Fatal(msg, fields...)
|
||
}
|
||
|
||
// Debugf 格式化调试日志
|
||
func Debugf(template string, args ...any) {
|
||
currentSugar().Debugf(template, args...)
|
||
}
|
||
|
||
// Infof 格式化信息日志
|
||
func Infof(template string, args ...any) {
|
||
currentSugar().Infof(template, args...)
|
||
}
|
||
|
||
// Warnf 格式化警告日志
|
||
func Warnf(template string, args ...any) {
|
||
currentSugar().Warnf(template, args...)
|
||
}
|
||
|
||
// Errorf 格式化错误日志
|
||
func Errorf(template string, args ...any) {
|
||
currentSugar().Errorf(template, args...)
|
||
}
|
||
|
||
// Fatalf 格式化致命错误日志（仅供应用层使用，框架内部禁止调用）
|
||
func Fatalf(template string, args ...any) {
|
||
currentSugar().Fatalf(template, args...)
|
||
}
|
||
|
||
// APILog 返回 API 专用日志器（写 logs/api.log + console）
|
||
func APILog() *zap.Logger {
|
||
return currentAPILog()
|
||
}
|
||
|
||
// DBLog 返回数据库专用日志器（写 logs/database.log + console）
|
||
func DBLog() *zap.Logger {
|
||
return currentDBLog()
|
||
}
|