Add log level flag

This commit is contained in:
2023-12-30 16:24:16 -06:00
parent f1b3dd5ffa
commit 4e01cf0dbb

10
main.go
View File

@@ -16,8 +16,9 @@ import (
) )
var ( var (
client *http.Client client *http.Client
db *badger.DB db *badger.DB
flagLevel = flag.String("level", "info", "log level")
// A channel that will be used to buffer incomplete entries that need to be queried properly // A channel that will be used to buffer incomplete entries that need to be queried properly
incompleteEntries = make(chan Entry) incompleteEntries = make(chan Entry)
@@ -27,6 +28,11 @@ var (
) )
func init() { func init() {
// Acquire log level from flag
flag.Parse()
parsedLevel, _ := zerolog.ParseLevel(*flagLevel)
zerolog.SetGlobalLevel(parsedLevel)
log.Logger = zerolog.New(logSplitter{}).With().Timestamp().Logger() log.Logger = zerolog.New(logSplitter{}).With().Timestamp().Logger()
// Initialize Badger db store // Initialize Badger db store