geck-go/logging/init.go

package logging

import (
	"os"
	"time"

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

// Initialize configures ZeroLog with defaults.
func Initialize(console bool) {
	if console {
		cw := zerolog.NewConsoleWriter()
		cw.Out = os.Stderr
		cw.TimeFormat = "15:04:05"
		log.Logger = zerolog.New(cw).With().Timestamp().Logger()
	}

	zerolog.DurationFieldUnit = time.Millisecond
}

// SetGlobalLevel sets the global log level according to the passed debug flag.
func SetGlobalLevel(debug bool) {
	if debug {
		zerolog.SetGlobalLevel(zerolog.DebugLevel)
	} else {
		zerolog.SetGlobalLevel(zerolog.InfoLevel)
	}
}