mop-account-system-yykj/common/log.go

package common

import (
	"mop-account-system/config"
	"os"

	"github.com/sirupsen/logrus"
)

var GinLog *logrus.Logger

func init() {
	GinLog = GetDefaultGinLog()
}

func GetDefaultGinLog() *logrus.Logger {
	l := logrus.New()
	l.Formatter = &logrus.JSONFormatter{
		TimestampFormat:  "2006-01-02T15:04:05",
		DisableTimestamp: false,
		FieldMap:         nil,
		CallerPrettyfier: nil,
		PrettyPrint:      false,
	}
	l.Out = os.Stdout
	l.ReportCaller = true
	l.Level = getModeByStr(config.Cfg.Mode)
	return l
}

func getModeByStr(s string) logrus.Level {
	switch s {
	case "debug":
		return logrus.DebugLevel
	case "info":
		return logrus.InfoLevel
	case "warn":
		return logrus.WarnLevel
	case "error":
		return logrus.ErrorLevel
	default:
		return logrus.DebugLevel
	}
}