Add configurable log level to badger logging

needed to clean up logs to just what i wanted
This commit is contained in:
2023-12-28 12:59:41 -06:00
parent 30b98a36f3
commit 7d13958532
2 changed files with 26 additions and 6 deletions

21
logs.go
View File

@@ -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{}) { func (l badgerZerologLogger) Errorf(format string, args ...interface{}) {
if l.level <= ERROR {
log.Error().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) log.Error().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n"))
}
} }
func (l badgerZerologLogger) Warningf(format string, args ...interface{}) { func (l badgerZerologLogger) Warningf(format string, args ...interface{}) {
if l.level <= WARNING {
log.Warn().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) log.Warn().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n"))
}
} }
func (l badgerZerologLogger) Infof(format string, args ...interface{}) { func (l badgerZerologLogger) Infof(format string, args ...interface{}) {
if l.level <= INFO {
log.Info().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) log.Info().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n"))
}
} }
func (l badgerZerologLogger) Debugf(format string, args ...interface{}) { func (l badgerZerologLogger) Debugf(format string, args ...interface{}) {
if l.level <= DEBUG {
log.Debug().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n")) log.Debug().Msg(strings.TrimRight(fmt.Sprintf(format, args...), "\n"))
}
} }

View File

@@ -23,7 +23,7 @@ func init() {
// Initialize Badger db store // Initialize Badger db store
var err error var err error
options := badger.DefaultOptions("./db/").WithLogger(badgerZerologLogger{}) options := badger.DefaultOptions("./db/").WithLogger(badgerZerologLogger{level: WARNING})
db, err = badger.Open(options) db, err = badger.Open(options)
if err != nil { if err != nil {
log.Fatal().Err(err).Msg("Failed to open database") 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 { client.Jar.SetCookies(utsaUrl, lo.Map(cookies, func(cookie http.Cookie, _ int) *http.Cookie {
return &cookie return &cookie
})) }))
log.Debug().Int("count", len(cookies)).Msg("Cookies Loaded from DB")
} }
func main() { func main() {