Go语言中实现分级日志的策略与实践

go get github.com/sirupsen/logrus

package main

import (
    "flag"
    "fmt"
    "io"
    "os"
    "strings"

    "github.com/sirupsen/logrus"
)

// 定义命令行参数
var logLevel string
var logToFile bool
var logFilePath string

func init() {
    // 初始化命令行参数
    flag.StringVar(&logLevel, "log-level", "info", "Set the logging level (trace, debug, info, warn, error, fatal, panic)")
    flag.BoolVar(&logToFile, "log-to-file", false, "Enable logging to a file")
    flag.StringVar(&logFilePath, "log-file-path", "application.log", "Path to the log file")
    flag.Parse() // 解析命令行参数
}

func main() {
    // 1. 设置日志级别
    level, err := logrus.ParseLevel(logLevel)
    if err != nil {
        logrus.SetLevel(logrus.InfoLevel) // 如果解析失败，默认设置为 Info 级别
        logrus.Warnf("Invalid log level '%s' specified, defaulting to 'info'. Error: %v", logLevel, err)
    } else {
        logrus.SetLevel(level)
    }

    // 2. 配置日志输出目标
    var writers []io.Writer
    writers = append(writers, os.Stdout) // 总是输出到标准输出

    if logToFile {
        // 尝试打开日志文件，如果文件不存在则创建，如果存在则追加写入
        logFile, err := os.OpenFile(logFilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
        if err != nil {
            logrus.Errorf("Failed to open log file '%s': %v. Logging only to stdout.", logFilePath, err)
        } else {
            defer logFile.Close() // 确保程序退出时关闭文件
            writers = append(writers, logFile)
        }
    }

    // 使用 io.MultiWriter 将日志同时输出到多个目标
    mw := io.MultiWriter(writers...)
    logrus.SetOutput(mw)

    // 3. 设置日志格式
    logrus.SetFormatter(&logrus.TextFormatter{
        FullTimestamp:   true,              // 显示完整时间戳
        TimestampFormat: "2006-01-02 15:04:05", // 自定义时间戳格式
        ForceColors:     true,              // 尝试在终端输出中强制使用颜色
        DisableColors:   false,             // 禁用颜色（如果输出到文件通常需要禁用）
    })

    // 如果输出到文件，通常禁用颜色以避免文件内容中出现ANSI转义码
    if logToFile {
        // 创建一个不带颜色的Formatter用于文件输出
        fileFormatter := &logrus.TextFormatter{
            FullTimestamp:   true,
            TimestampFormat: "2006-01-02 15:04:05",
            DisableColors:   true, // 文件输出禁用颜色
        }
        // 针对文件输出，可以创建一个新的Logger实例或使用Hook
        // 简单起见，这里演示一个更直接但可能不够灵活的方式，
        // 实际项目中可以考虑使用logrus.New()创建独立Logger实例或使用Hook
        // 为了演示方便，我们暂时简化处理，实际生产环境建议更精细的控制。
        // 例如，可以为文件和stdout分别创建Logger实例。
        // 或者使用logrus的Hook功能来根据输出目标调整格式。
        // 这里为了保持示例的简洁性，我们假设stdout和文件使用相同的formatter，但文件禁用颜色。
        // 更严谨的做法：
        // fileHook := NewFileHook(logFile, fileFormatter)
        // logrus.AddHook(fileHook)
        // logrus.SetOutput(os.Stdout) // 主输出只到stdout
        // 简化处理：
        if len(writers) > 1 { // 如果有文件输出
            logrus.SetFormatter(&logrus.TextFormatter{
                FullTimestamp:   true,
                TimestampFormat: "2006-01-02 15:04:05",
                ForceColors:     false, // 如果有文件输出，默认关闭颜色，避免文件污染
                DisableColors:   true,
            })
        }
    }


    // 4. 记录不同级别的日志
    logrus.Trace("这是一个追踪消息。")
    logrus.Debug("这是一个调试消息。变量x=%d", 10)
    logrus.Info("应用程序已成功启动。")
    logrus.Warn("配置文件未找到，正在使用默认设置。")
    logrus.Error("数据库连接失败，请检查配置。")
    logrus.Fatal("发生严重错误，应用程序即将退出。") // Fatal 会调用 os.Exit(1)
    // logrus.Panic("这是一个导致程序恐慌的消息。") // Panic 会触发 panic
}