engigu
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package logger
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var Log *logrus.Logger
|
|
|
|
func init() {
|
|
Log = logrus.New()
|
|
|
|
// 设置日志格式
|
|
Log.SetFormatter(&logrus.TextFormatter{
|
|
FullTimestamp: true,
|
|
TimestampFormat: "2006-01-02 15:04:05",
|
|
ForceColors: true,
|
|
})
|
|
|
|
// 设置日志级别
|
|
Log.SetLevel(logrus.InfoLevel)
|
|
|
|
// 输出到标准输出
|
|
Log.SetOutput(os.Stdout)
|
|
}
|
|
|
|
// SetupFileOutput 设置文件输出
|
|
func SetupFileOutput(logDir string) error {
|
|
if err := os.MkdirAll(logDir, 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
logFile := filepath.Join(logDir, time.Now().Format("2006-01-02")+".log")
|
|
file, err := os.OpenFile(logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
Log.SetOutput(file)
|
|
return nil
|
|
}
|
|
|
|
// SetLevel 设置日志级别
|
|
func SetLevel(level string) {
|
|
switch level {
|
|
case "debug":
|
|
Log.SetLevel(logrus.DebugLevel)
|
|
case "info":
|
|
Log.SetLevel(logrus.InfoLevel)
|
|
case "warn":
|
|
Log.SetLevel(logrus.WarnLevel)
|
|
case "error":
|
|
Log.SetLevel(logrus.ErrorLevel)
|
|
default:
|
|
Log.SetLevel(logrus.InfoLevel)
|
|
}
|
|
}
|
|
|
|
// 便捷方法
|
|
func Debug(args ...interface{}) { Log.Debug(args...) }
|
|
func Info(args ...interface{}) { Log.Info(args...) }
|
|
func Warn(args ...interface{}) { Log.Warn(args...) }
|
|
func Error(args ...interface{}) { Log.Error(args...) }
|
|
func Fatal(args ...interface{}) { Log.Fatal(args...) }
|
|
|
|
func Debugf(format string, args ...interface{}) { Log.Debugf(format, args...) }
|
|
func Infof(format string, args ...interface{}) { Log.Infof(format, args...) }
|
|
func Warnf(format string, args ...interface{}) { Log.Warnf(format, args...) }
|
|
func Errorf(format string, args ...interface{}) { Log.Errorf(format, args...) }
|
|
func Fatalf(format string, args ...interface{}) { Log.Fatalf(format, args...) }
|
|
|
|
// WithField 带字段的日志
|
|
func WithField(key string, value interface{}) *logrus.Entry {
|
|
return Log.WithField(key, value)
|
|
}
|
|
|
|
// WithFields 带多个字段的日志
|
|
func WithFields(fields logrus.Fields) *logrus.Entry {
|
|
return Log.WithFields(fields)
|
|
}
|