package context import ( // stdlib "fmt" "os" "runtime" "strings" "time" // other "github.com/rs/zerolog" ) // Puts memory usage into log lines. func (c *Context) getMemoryUsage(e *zerolog.Event, level zerolog.Level, message string) { var m runtime.MemStats runtime.ReadMemStats(&m) e.Str("memalloc", fmt.Sprintf("%dMB", m.Alloc/1024/1024)) e.Str("memsys", fmt.Sprintf("%dMB", m.Sys/1024/1024)) e.Str("numgc", fmt.Sprintf("%d", m.NumGC)) } // Initializes logger. func (c *Context) initializeLogger() { // Устанавливаем форматирование логгера. output := zerolog.ConsoleWriter{Out: os.Stdout, NoColor: false, TimeFormat: time.RFC3339} output.FormatLevel = func(i interface{}) string { var v string if ii, ok := i.(string); ok { ii = strings.ToUpper(ii) switch ii { case "DEBUG": v = fmt.Sprintf("\x1b[30m%-5s\x1b[0m", ii) case "ERROR": v = fmt.Sprintf("\x1b[31m%-5s\x1b[0m", ii) case "FATAL": v = fmt.Sprintf("\x1b[35m%-5s\x1b[0m", ii) case "INFO": v = fmt.Sprintf("\x1b[32m%-5s\x1b[0m", ii) case "PANIC": v = fmt.Sprintf("\x1b[36m%-5s\x1b[0m", ii) case "WARN": v = fmt.Sprintf("\x1b[33m%-5s\x1b[0m", ii) default: v = ii } } return fmt.Sprintf("| %s |", v) } c.Logger = zerolog.New(output).With().Timestamp().Logger() c.Logger = c.Logger.Hook(zerolog.HookFunc(c.getMemoryUsage)) } // Initialize logger after configuration parse. func (c *Context) initializeLoggerPost() { // Set log level. c.Logger.Info().Msgf("Setting logger level: %s", c.Config.Logging.LogLevel) switch c.Config.Logging.LogLevel { case "DEBUG": zerolog.SetGlobalLevel(zerolog.DebugLevel) case "INFO": zerolog.SetGlobalLevel(zerolog.InfoLevel) case "WARN": zerolog.SetGlobalLevel(zerolog.WarnLevel) case "ERROR": zerolog.SetGlobalLevel(zerolog.ErrorLevel) case "FATAL": zerolog.SetGlobalLevel(zerolog.FatalLevel) case "PANIC": zerolog.SetGlobalLevel(zerolog.PanicLevel) } }