mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
48 lines
966 B
Go
48 lines
966 B
Go
package hrp
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog"
|
|
zlog "github.com/rs/zerolog/log"
|
|
)
|
|
|
|
var log = zlog.Logger
|
|
|
|
// SetLogger configures the log level and format.
|
|
func SetLogger(level string, logJSON bool) {
|
|
if !logJSON {
|
|
setLogPretty()
|
|
}
|
|
setLogLevel(level)
|
|
}
|
|
|
|
func setLogLevel(level string) {
|
|
level = strings.ToUpper(level)
|
|
log.Info().Msgf("Set log level to %s", level)
|
|
switch level {
|
|
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)
|
|
}
|
|
}
|
|
|
|
func setLogPretty() {
|
|
log = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
|
|
log.Info().Msg("Set log to color console other than JSON format.")
|
|
}
|
|
|
|
func GetLogger() zerolog.Logger {
|
|
return log
|
|
}
|