49 lines
877 B
Go
49 lines
877 B
Go
package log
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var Logger = logrus.New()
|
|
|
|
type CustomTextFormatter struct {
|
|
logrus.TextFormatter
|
|
}
|
|
|
|
func (f *CustomTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {
|
|
if entry.Level != logrus.InfoLevel {
|
|
return f.TextFormatter.Format(entry)
|
|
}
|
|
|
|
var b bytes.Buffer
|
|
b.WriteString(entry.Message)
|
|
b.WriteByte('\n')
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
|
|
func InitLogger(debug bool) {
|
|
if debug {
|
|
Logger.SetFormatter(&logrus.TextFormatter{
|
|
DisableLevelTruncation: true,
|
|
PadLevelText: true,
|
|
DisableTimestamp: true,
|
|
})
|
|
Logger.SetLevel(logrus.DebugLevel)
|
|
} else {
|
|
Logger.SetFormatter(&CustomTextFormatter{
|
|
logrus.TextFormatter{
|
|
DisableLevelTruncation: true,
|
|
PadLevelText: true,
|
|
DisableTimestamp: true,
|
|
},
|
|
})
|
|
Logger.SetLevel(logrus.InfoLevel)
|
|
}
|
|
|
|
Logger.SetOutput(os.Stdout)
|
|
}
|