golib/logger/init.go

//
// init.go
// Copyright (C) 2023 tiglog <me@tiglog.com>
//
// Distributed under terms of the MIT license.
//

package logger

import (
	"path/filepath"
	"strconv"
	"time"

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

var loggerLevel = zerolog.InfoLevel

func init() {
	// zerolog.ErrorStackMarshaler = func(err error) interface{} {
	// 	return pkgerrors.MarshalStack(errors.Wrap(err, "wrap"))
	// }
	zerolog.ErrorStackMarshaler = pkgerrors.MarshalStack

	zerolog.TimeFieldFormat = time.RFC3339
	zerolog.CallerMarshalFunc = func(pc uintptr, file string, line int) string {
		return filepath.Base(file) + ":" + strconv.Itoa(line)
	}
}

func SetLevel(level zerolog.Level) {
	loggerLevel = level
}