| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- package fslog
- import (
- "fmt"
- "go.uber.org/zap"
- "go.uber.org/zap/zapcore"
- "gopkg.in/natefinch/lumberjack.v2"
- "os"
- "path/filepath"
- "sync"
- )
- const (
- LoggerLevelInfo = "info"
- LoggerLevelError = "error"
- LoggerLevelDebug = "debug"
- LoggerLevelWarn = "Warn"
- )
- var LoggerLever = LoggerLevelDebug
- func chooseLoggerLever() zapcore.Level {
- switch LoggerLever {
- case LoggerLevelInfo:
- return zapcore.InfoLevel
- case LoggerLevelError:
- return zapcore.ErrorLevel
- case LoggerLevelDebug:
- return zapcore.DebugLevel
- case LoggerLevelWarn:
- return zapcore.WarnLevel
- default:
- return zapcore.DebugLevel
- }
- }
- var logger *zap.Logger
- var initLoggerOnce sync.Once
- const logDir = "logs"
- func lazyInitialize() {
- initLoggerOnce.Do(func() {
- logFilePath := filepath.Join(logDir, "log")
- lumberJackLogger := &lumberjack.Logger{
- Filename: logFilePath,
- MaxAge: 10,
- }
- developmentEncoderConfig := zap.NewDevelopmentEncoderConfig()
- encoder := zapcore.NewConsoleEncoder(developmentEncoderConfig)
- core := zapcore.NewTee(
- zapcore.NewCore(encoder, zapcore.AddSync(lumberJackLogger), chooseLoggerLever()),
- zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), chooseLoggerLever()),
- )
- logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
- })
- }
- func GetLogger() *zap.Logger {
- return logger
- }
- func Info(message string) {
- lazyInitialize()
- logger.Info(message)
- }
- func Error(err error) {
- lazyInitialize()
- message := fmt.Sprintf("\n%+v\n", err)
- logger.Error(message)
- }
- func Debug(message string) {
- lazyInitialize()
- logger.Debug(message)
- }
- func Warn(message string) {
- lazyInitialize()
- logger.Warn(message)
- }
- func pathExists(path string) bool {
- _, err := os.Stat(path)
- if err == nil {
- return true
- }
- if os.IsNotExist(err) {
- return false
- }
- return false
- }
|