fslog/fslog_instance.go

package fslog

import (
	"fmt"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
	"gopkg.in/natefinch/lumberjack.v2"
	"os"
	"path/filepath"
)

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

const logDir = "logs"

func init() {
	if !pathExists(logDir) {
		err := os.MkdirAll(logDir, os.ModeDir|os.ModePerm)
		if err != nil {
			panic(err)
		}
	}

	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) {
	logger.Info(message)
}

func Error(err error) {
	message := fmt.Sprintf("\n%+v\n", err)
	logger.Error(message)
}

func Debug(message string) {
	logger.Debug(message)
}

func Warn(message string) {
	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
}