64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var logger *logrus.Logger
|
|
|
|
func InitLogger() {
|
|
logger = logrus.New()
|
|
disableLogColor := os.Getenv("DISABLE_LOG_COLOR")
|
|
logColor := true
|
|
if disableLogColor == "true" {
|
|
logColor = false
|
|
}
|
|
logger.SetFormatter(&logrus.TextFormatter{
|
|
FullTimestamp: true,
|
|
ForceColors: logColor,
|
|
})
|
|
logLevel := os.Getenv("LOG_LEVEL")
|
|
switch logLevel {
|
|
case "debug":
|
|
logger.SetLevel(logrus.DebugLevel)
|
|
logger.SetReportCaller(true)
|
|
case "info":
|
|
logger.SetLevel(logrus.InfoLevel)
|
|
case "warn", "warning":
|
|
logger.SetLevel(logrus.WarnLevel)
|
|
case "error":
|
|
logger.SetLevel(logrus.ErrorLevel)
|
|
case "fatal":
|
|
logger.SetLevel(logrus.FatalLevel)
|
|
case "panic":
|
|
logger.SetLevel(logrus.PanicLevel)
|
|
default:
|
|
// 默认使用 InfoLevel
|
|
logger.SetLevel(logrus.InfoLevel)
|
|
}
|
|
logOutput := os.Getenv("LOG_OUTPUT")
|
|
switch logOutput {
|
|
case "file":
|
|
// 设置日志输出到文件
|
|
logFile, err := os.OpenFile("/var/log/minimcd.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
|
|
if err != nil {
|
|
// 如果文件打开失败，回退到标准输出
|
|
logger.SetOutput(os.Stdout)
|
|
logger.Warn("Failed to log to file, using default stdout")
|
|
} else {
|
|
logger.SetOutput(logFile)
|
|
}
|
|
case "stdout", "":
|
|
// 默认输出到控制台
|
|
logger.SetOutput(os.Stdout)
|
|
default:
|
|
// 默认输出到控制台
|
|
logger.SetOutput(os.Stdout)
|
|
}
|
|
}
|
|
func GetLogger() *logrus.Logger {
|
|
return logger
|
|
}
|