main.go

package logging

import (
	"os"

	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"go.elastic.co/ecszerolog"
)

// SetupLogging prepares the logger, defaulting to JSON logging in Elastic format
func SetupLogging() {
	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
	logger := ecszerolog.New(os.Stdout)
	log.Logger = logger

	if _, ok := os.LookupEnv("LOCAL_LOGGER"); ok {
		log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stdout}) // pretty print to stderr
		log.Logger = log.With().Caller().Logger()                      // shows filename and line number
	}

	if level, ok := os.LookupEnv("LOG_LEVEL"); ok {
		switch level {
		case "debug", "DEBUG":
			zerolog.SetGlobalLevel(zerolog.DebugLevel)
		case "info", "INFO":
			zerolog.SetGlobalLevel(zerolog.InfoLevel)
		case "warn", "WARN":
			zerolog.SetGlobalLevel(zerolog.WarnLevel)
		case "error", "ERROR":
			zerolog.SetGlobalLevel(zerolog.ErrorLevel)
		case "panic", "PANIC":
			zerolog.SetGlobalLevel(zerolog.PanicLevel)
		default:
			logger.Error().Msgf("unknown value for LOG_LEVEL: %s, using debug", level)
			zerolog.SetGlobalLevel(zerolog.DebugLevel)
		}
	} else {
		zerolog.SetGlobalLevel(zerolog.DebugLevel)
	}
}