mirror of
https://github.com/Xevion/scla-unsubscribe.git
synced 2025-12-06 07:16:15 -06:00
Add configurable log level to badger logging
needed to clean up logs to just what i wanted
This commit is contained in:
21
logs.go
21
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{}) {
|
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"))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
3
main.go
3
main.go
@@ -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() {
|
||||||
|
|||||||
Reference in New Issue
Block a user