51 lines
922 B
Go
51 lines
922 B
Go
package tfa
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var log *logrus.Logger
|
|
|
|
func NewDefaultLogger() *logrus.Logger {
|
|
// Setup logger
|
|
log = logrus.StandardLogger()
|
|
logrus.SetOutput(os.Stdout)
|
|
|
|
// Set logger format
|
|
switch config.LogFormat {
|
|
case "pretty":
|
|
break
|
|
case "json":
|
|
logrus.SetFormatter(&logrus.JSONFormatter{})
|
|
// "text" is the default
|
|
default:
|
|
logrus.SetFormatter(&logrus.TextFormatter{
|
|
DisableColors: true,
|
|
FullTimestamp: true,
|
|
})
|
|
}
|
|
|
|
// Set logger level
|
|
switch config.LogLevel {
|
|
case "trace":
|
|
logrus.SetLevel(logrus.TraceLevel)
|
|
case "debug":
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
case "info":
|
|
logrus.SetLevel(logrus.InfoLevel)
|
|
case "error":
|
|
logrus.SetLevel(logrus.ErrorLevel)
|
|
case "fatal":
|
|
logrus.SetLevel(logrus.FatalLevel)
|
|
case "panic":
|
|
logrus.SetLevel(logrus.PanicLevel)
|
|
// warn is the default
|
|
default:
|
|
logrus.SetLevel(logrus.WarnLevel)
|
|
}
|
|
|
|
return log
|
|
}
|