Add configurable log level to badger logging

needed to clean up logs to just what i wanted

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"))
+	}
 }

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() {