88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package log
|
||
|
||
import (
|
||
"io"
|
||
"os"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
type Level logrus.Level
|
||
|
||
const (
|
||
PanicLevel Level = iota
|
||
FatalLevel
|
||
ErrorLevel
|
||
WarnLevel
|
||
InfoLevel
|
||
DebugLevel
|
||
TraceLevel
|
||
)
|
||
|
||
func SetLevel(level Level) {
|
||
switch level {
|
||
case PanicLevel:
|
||
logrus.SetLevel(logrus.PanicLevel)
|
||
case FatalLevel:
|
||
logrus.SetLevel(logrus.FatalLevel)
|
||
case ErrorLevel:
|
||
logrus.SetLevel(logrus.ErrorLevel)
|
||
case WarnLevel:
|
||
logrus.SetLevel(logrus.WarnLevel)
|
||
case InfoLevel:
|
||
logrus.SetLevel(logrus.InfoLevel)
|
||
case DebugLevel:
|
||
logrus.SetLevel(logrus.DebugLevel)
|
||
case TraceLevel:
|
||
logrus.SetLevel(logrus.TraceLevel)
|
||
}
|
||
}
|
||
|
||
func Panic(msg ...interface{}) {
|
||
logrus.Panic(msg...)
|
||
}
|
||
|
||
func Fatal(msg ...interface{}) {
|
||
logrus.Fatal(msg)
|
||
}
|
||
|
||
func Error(msg ...interface{}) {
|
||
logrus.Error(msg)
|
||
}
|
||
|
||
func Warn(msg ...interface{}) {
|
||
logrus.Warn(msg)
|
||
}
|
||
|
||
func Info(msg ...interface{}) {
|
||
logrus.Info(msg)
|
||
}
|
||
|
||
func Debug(msg ...interface{}) {
|
||
logrus.Debug(msg)
|
||
}
|
||
|
||
func Trace(msg ...interface{}) {
|
||
logrus.Trace(msg)
|
||
}
|
||
|
||
func InitLogToFile() {
|
||
//设置输出样式，自带的只有两种样式logrus.JSONFormatter{}和logrus.TextFormatter{}
|
||
logrus.SetFormatter(&logrus.TextFormatter{})
|
||
logrus.SetOutput(os.Stdout)
|
||
//设置output,默认为stderr,可以为任何io.Writer，比如文件*os.File
|
||
file, err := os.OpenFile("logs/pluto-server.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||
writers := []io.Writer{
|
||
file,
|
||
os.Stdout}
|
||
//同时写文件和屏幕
|
||
fileAndStdoutWriter := io.MultiWriter(writers...)
|
||
if err == nil {
|
||
logrus.SetOutput(fileAndStdoutWriter)
|
||
} else {
|
||
logrus.Info("failed to log to file.")
|
||
}
|
||
//设置最低loglevel
|
||
logrus.SetLevel(logrus.InfoLevel)
|
||
}
|