sower/pkg/deferlog/init.go

package deferlog

import (
	"os"
	"strconv"
	"strings"
	"time"

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

var StructLogger = zerolog.New(os.Stdout).
	With().Timestamp().Logger()

var ConsoleLogger = zerolog.New(zerolog.ConsoleWriter{
	Out:        os.Stdout,
	TimeFormat: time.StampMilli,
	FormatCaller: func(i interface{}) string {
		caller := i.(string)
		if idx := strings.Index(caller, "/pkg/mod/"); idx > 0 {
			return caller[idx+9:]
		}
		if idx := strings.LastIndexByte(caller, '/'); idx > 0 {
			return caller[idx+1:]
		}
		return caller
	},
}).With().Timestamp().Logger()

func init() {
	zerolog.ErrorStackMarshaler = func(err error) interface{} {
		return pkgerrors.MarshalStack(err)
	}

	if ok, _ := strconv.ParseBool(os.Getenv("DEBUG")); ok {
		SetDefaultLogger(ConsoleLogger, 1, zerolog.DebugLevel)

	} else if fi, _ := os.Stdout.Stat(); (fi.Mode() & os.ModeCharDevice) == 0 {
		SetDefaultLogger(StructLogger, 1, zerolog.InfoLevel)

	} else {
		SetDefaultLogger(ConsoleLogger, 1, zerolog.InfoLevel)
	}
}

func SetDefaultLogger(logger zerolog.Logger, deferSkip int, logLevel zerolog.Level) {
	log.Logger = logger.With().Caller().Logger().Level(logLevel)
	Logger = logger.With().CallerWithSkipFrameCount(deferSkip + 2).Logger().Level(logLevel)
}