message-server/logger/logger.go

package logger

import (
	"os"
	"time"

	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
)

// InitLogger 初始化日志
func InitLogger(logMode string, logLevel zerolog.Level) {
	if logMode == "color" { // 控制台模式 + 彩色输出
		consoleWriter := zerolog.ConsoleWriter{
			Out:        os.Stdout,
			TimeFormat: time.RFC3339,
		}
		// 默认 Info 级别
		log.Logger = log.Output(consoleWriter).Level(logLevel)
	} else if logMode == "standard" { // 生产模式：标准 JSON 日志
		// 生产模式：标准 JSON 日志
		zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
		log.Logger = log.Output(os.Stdout).Level(logLevel)
	} else {
		// 默认模式：控制台模式 + 彩色输出
		consoleWriter := zerolog.ConsoleWriter{
			Out:        os.Stdout,
			TimeFormat: time.RFC3339,
		}
		// 默认 Info 级别
		log.Logger = log.Output(consoleWriter).Level(logLevel)
	}
}