go-common/logger/logger.go

package logger

import (
	"fmt"
	"io"
	"log"
	"os"
	"path/filepath"
	"sync"

	"git.toowon.com/jimmy/go-common/config"
)

// logMessage 异步日志消息结构
type logMessage struct {
	level  string // debug, info, warn, error
	format string
	args   []interface{}
	fields map[string]interface{} // 用于带字段的日志
}

// Logger 日志记录器
type Logger struct {
	infoLog  *log.Logger
	errorLog *log.Logger
	warnLog  *log.Logger
	debugLog *log.Logger
	config   *config.LoggerConfig

	// 异步相关字段
	async    bool             // 是否异步模式
	logChan  chan *logMessage // 异步日志channel
	done     chan struct{}    // 用于优雅关闭
	wg       sync.WaitGroup   // 等待所有日志写入完成
	closed   bool             // 是否已关闭
	closeMux sync.RWMutex     // 保护closed字段
}

// NewLogger 创建日志记录器
func NewLogger(cfg *config.LoggerConfig) (*Logger, error) {
	if cfg == nil {
		// 使用默认配置
		cfg = &config.LoggerConfig{
			Level:      "info",
			Output:     "stdout",
			FilePath:   "",
			Async:      false, // 默认同步
			BufferSize: 1000,  // 默认缓冲区大小
		}
	}

	// 设置默认值
	if cfg.Level == "" {
		cfg.Level = "info"
	}
	if cfg.Output == "" {
		cfg.Output = "stdout"
	}
	if cfg.BufferSize <= 0 {
		cfg.BufferSize = 1000 // 默认缓冲区大小
	}

	// 创建输出目标
	var writers []io.Writer
	switch cfg.Output {
	case "stdout":
		writers = append(writers, os.Stdout)
	case "stderr":
		writers = append(writers, os.Stderr)
	case "file":
		if cfg.FilePath == "" {
			return nil, fmt.Errorf("file path is required when output is file")
		}
		// 确保目录存在
		dir := filepath.Dir(cfg.FilePath)
		if err := os.MkdirAll(dir, 0755); err != nil {
			return nil, fmt.Errorf("failed to create log directory: %w", err)
		}
		file, err := os.OpenFile(cfg.FilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
		if err != nil {
			return nil, fmt.Errorf("failed to open log file: %w", err)
		}
		writers = append(writers, file)
	case "both":
		writers = append(writers, os.Stdout)
		if cfg.FilePath == "" {
			return nil, fmt.Errorf("file path is required when output is both")
		}
		dir := filepath.Dir(cfg.FilePath)
		if err := os.MkdirAll(dir, 0755); err != nil {
			return nil, fmt.Errorf("failed to create log directory: %w", err)
		}
		file, err := os.OpenFile(cfg.FilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
		if err != nil {
			return nil, fmt.Errorf("failed to open log file: %w", err)
		}
		writers = append(writers, file)
	default:
		return nil, fmt.Errorf("invalid output type: %s", cfg.Output)
	}

	// 创建多写入器
	multiWriter := io.MultiWriter(writers...)

	// 创建日志前缀
	prefix := ""
	if cfg.Prefix != "" {
		prefix = cfg.Prefix + " "
	}

	// 创建日志记录器
	logger := &Logger{
		config: cfg,
		async:  cfg.Async,
	}

	// 根据日志级别创建不同的logger
	flags := log.LstdFlags
	if cfg.DisableTimestamp {
		flags = 0
	}

	if cfg.Level == "debug" || cfg.Level == "info" || cfg.Level == "warn" || cfg.Level == "error" {
		logger.infoLog = log.New(multiWriter, prefix+"[INFO] ", flags)
		logger.warnLog = log.New(multiWriter, prefix+"[WARN] ", flags)
		logger.errorLog = log.New(multiWriter, prefix+"[ERROR] ", flags)
		if cfg.Level == "debug" {
			logger.debugLog = log.New(multiWriter, prefix+"[DEBUG] ", flags)
		}
	}

	// 如果启用异步模式，启动goroutine处理日志
	if cfg.Async {
		logger.logChan = make(chan *logMessage, cfg.BufferSize)
		logger.done = make(chan struct{})
		logger.wg.Add(1)
		go logger.processLogs()
	}

	return logger, nil
}

// processLogs 异步处理日志（goroutine）
func (l *Logger) processLogs() {
	defer l.wg.Done()

	for {
		select {
		case msg := <-l.logChan:
			if msg == nil {
				// channel已关闭，退出
				return
			}
			l.writeLog(msg)
		case <-l.done:
			// 收到关闭信号，处理完剩余日志后退出
			for {
				select {
				case msg := <-l.logChan:
					if msg == nil {
						return
					}
					l.writeLog(msg)
				default:
					return
				}
			}
		}
	}
}

// writeLog 实际写入日志（内部方法）
func (l *Logger) writeLog(msg *logMessage) {
	var logger *log.Logger
	switch msg.level {
	case "debug":
		logger = l.debugLog
	case "info":
		logger = l.infoLog
	case "warn":
		logger = l.warnLog
	case "error":
		logger = l.errorLog
	default:
		return
	}

	if logger == nil {
		return
	}

	// 如果有字段，先格式化字段
	format := msg.format
	if len(msg.fields) > 0 {
		fieldStr := formatFields(msg.fields)
		format = fieldStr + format
	}

	// 写入日志
	logger.Printf(format, msg.args...)
}

// isClosed 检查logger是否已关闭
func (l *Logger) isClosed() bool {
	l.closeMux.RLock()
	defer l.closeMux.RUnlock()
	return l.closed
}

// setClosed 设置logger为已关闭状态
func (l *Logger) setClosed() {
	l.closeMux.Lock()
	defer l.closeMux.Unlock()
	l.closed = true
}

// log 内部日志方法，根据模式选择同步或异步
func (l *Logger) log(level string, format string, args []interface{}, fields map[string]interface{}) {
	// 如果已关闭，直接返回
	if l.isClosed() {
		return
	}

	// 如果是异步模式，发送到channel
	if l.async {
		// 检查channel是否已关闭
		select {
		case l.logChan <- &logMessage{
			level:  level,
			format: format,
			args:   args,
			fields: fields,
		}:
			// 成功发送
		default:
			// channel已满或已关闭，同步写入（降级处理）
			l.writeLog(&logMessage{
				level:  level,
				format: format,
				args:   args,
				fields: fields,
			})
		}
	} else {
		// 同步模式，直接写入
		l.writeLog(&logMessage{
			level:  level,
			format: format,
			args:   args,
			fields: fields,
		})
	}
}

// Debug 记录调试日志
func (l *Logger) Debug(format string, v ...interface{}) {
	l.log("debug", format, v, nil)
}

// Info 记录信息日志
func (l *Logger) Info(format string, v ...interface{}) {
	l.log("info", format, v, nil)
}

// Warn 记录警告日志
func (l *Logger) Warn(format string, v ...interface{}) {
	l.log("warn", format, v, nil)
}

// Error 记录错误日志
func (l *Logger) Error(format string, v ...interface{}) {
	l.log("error", format, v, nil)
}

// Fatal 记录致命错误日志并退出程序（始终同步）
func (l *Logger) Fatal(format string, v ...interface{}) {
	// Fatal必须同步执行，确保日志写入后再退出
	if l.errorLog != nil {
		l.errorLog.Fatalf(format, v...)
	} else {
		log.Fatalf(format, v...)
	}
}

// Panic 记录恐慌日志并触发panic（始终同步）
func (l *Logger) Panic(format string, v ...interface{}) {
	// Panic必须同步执行，确保日志写入后再panic
	if l.errorLog != nil {
		l.errorLog.Panicf(format, v...)
	} else {
		log.Panicf(format, v...)
	}
}

// WithFields 创建带字段的日志记录器（简化版，返回格式化字符串）
func (l *Logger) WithFields(fields map[string]interface{}) *Logger {
	// 返回自身，实际使用时可以在format中包含fields
	return l
}

// formatFields 格式化字段
func formatFields(fields map[string]interface{}) string {
	if len(fields) == 0 {
		return ""
	}
	result := ""
	for k, v := range fields {
		result += fmt.Sprintf("%s=%v ", k, v)
	}
	return result
}

// Debugf 记录调试日志（带字段）
func (l *Logger) Debugf(fields map[string]interface{}, format string, v ...interface{}) {
	l.log("debug", format, v, fields)
}

// Infof 记录信息日志（带字段）
func (l *Logger) Infof(fields map[string]interface{}, format string, v ...interface{}) {
	l.log("info", format, v, fields)
}

// Warnf 记录警告日志（带字段）
func (l *Logger) Warnf(fields map[string]interface{}, format string, v ...interface{}) {
	l.log("warn", format, v, fields)
}

// Errorf 记录错误日志（带字段）
func (l *Logger) Errorf(fields map[string]interface{}, format string, v ...interface{}) {
	l.log("error", format, v, fields)
}

// Close 优雅关闭logger（仅异步模式需要）
// 等待所有日志写入完成后再返回
func (l *Logger) Close() error {
	if !l.async {
		// 同步模式不需要关闭
		return nil
	}

	// 检查是否已关闭
	if l.isClosed() {
		return nil
	}

	// 设置关闭状态
	l.setClosed()

	// 发送关闭信号
	close(l.done)

	// 关闭channel（会触发processLogs退出）
	close(l.logChan)

	// 等待所有日志写入完成
	l.wg.Wait()

	return nil
}