diff --git a/logs.go b/logs.go index 5e0da4e..0a82078 100644 --- a/logs.go +++ b/logs.go @@ -33,20 +33,39 @@ func (l logSplitter) WriteLevel(level zerolog.Level, p []byte) (n int, err error } } -type badgerZerologLogger struct{} +type loggingLevel int + +const ( + DEBUG loggingLevel = iota + INFO + WARNING + ERROR +) + +type badgerZerologLogger struct { + level loggingLevel +} func (l badgerZerologLogger) Errorf(format string, args ...interface{}) { - log.Error().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + if l.level <= ERROR { + log.Error().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + } } func (l badgerZerologLogger) Warningf(format string, args ...interface{}) { - log.Warn().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + if l.level <= WARNING { + log.Warn().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + } } func (l badgerZerologLogger) Infof(format string, args ...interface{}) { - log.Info().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + if l.level <= INFO { + log.Info().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + } } func (l badgerZerologLogger) Debugf(format string, args ...interface{}) { - log.Debug().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + if l.level <= DEBUG { + log.Debug().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) + } } diff --git a/main.go b/main.go index 1246e0e..e0cdf7d 100644 --- a/main.go +++ b/main.go @@ -23,7 +23,7 @@ func init() { // Initialize Badger db store var err error - options := badger.DefaultOptions("./db/").WithLogger(badgerZerologLogger{}) + options := badger.DefaultOptions("./db/").WithLogger(badgerZerologLogger{level: WARNING}) db, err = badger.Open(options) if err != nil { log.Fatal().Err(err).Msg("Failed to open database") @@ -91,6 +91,7 @@ func LoadCookies() { client.Jar.SetCookies(utsaUrl, lo.Map(cookies, func(cookie http.Cookie, _ int) *http.Cookie { return &cookie })) + log.Debug().Int("count", len(cookies)).Msg("Cookies Loaded from DB") } func main() {