fslog_instance.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. package fslog
  2. import (
  3. "fmt"
  4. "go.uber.org/zap"
  5. "go.uber.org/zap/zapcore"
  6. "gopkg.in/natefinch/lumberjack.v2"
  7. "os"
  8. "path/filepath"
  9. "sync"
  10. )
  11. const (
  12. LoggerLevelInfo = "info"
  13. LoggerLevelError = "error"
  14. LoggerLevelDebug = "debug"
  15. LoggerLevelWarn = "Warn"
  16. )
  17. var LoggerLever = LoggerLevelDebug
  18. func chooseLoggerLever() zapcore.Level {
  19. switch LoggerLever {
  20. case LoggerLevelInfo:
  21. return zapcore.InfoLevel
  22. case LoggerLevelError:
  23. return zapcore.ErrorLevel
  24. case LoggerLevelDebug:
  25. return zapcore.DebugLevel
  26. case LoggerLevelWarn:
  27. return zapcore.WarnLevel
  28. default:
  29. return zapcore.DebugLevel
  30. }
  31. }
  32. var logger *zap.Logger
  33. var initLoggerOnce sync.Once
  34. const logDir = "logs"
  35. func lazyInitialize() {
  36. initLoggerOnce.Do(func() {
  37. logFilePath := filepath.Join(logDir, "log")
  38. lumberJackLogger := &lumberjack.Logger{
  39. Filename: logFilePath,
  40. MaxAge: 10,
  41. }
  42. developmentEncoderConfig := zap.NewDevelopmentEncoderConfig()
  43. encoder := zapcore.NewConsoleEncoder(developmentEncoderConfig)
  44. core := zapcore.NewTee(
  45. zapcore.NewCore(encoder, zapcore.AddSync(lumberJackLogger), chooseLoggerLever()),
  46. zapcore.NewCore(encoder, zapcore.AddSync(os.Stdout), chooseLoggerLever()),
  47. )
  48. logger = zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))
  49. })
  50. }
  51. func GetLogger() *zap.Logger {
  52. return logger
  53. }
  54. func Info(message string) {
  55. lazyInitialize()
  56. logger.Info(message)
  57. }
  58. func Error(err error) {
  59. lazyInitialize()
  60. message := fmt.Sprintf("\n%+v\n", err)
  61. logger.Error(message)
  62. }
  63. func Debug(message string) {
  64. lazyInitialize()
  65. logger.Debug(message)
  66. }
  67. func Warn(message string) {
  68. lazyInitialize()
  69. logger.Warn(message)
  70. }
  71. func pathExists(path string) bool {
  72. _, err := os.Stat(path)
  73. if err == nil {
  74. return true
  75. }
  76. if os.IsNotExist(err) {
  77. return false
  78. }
  79. return false
  80. }