diff --git a/main.go b/main.go index c717718..fce5888 100644 --- a/main.go +++ b/main.go @@ -16,8 +16,9 @@ import ( ) var ( - client *http.Client - db *badger.DB + client *http.Client + 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 incompleteEntries = make(chan Entry) @@ -27,6 +28,11 @@ var ( ) func init() { + // Acquire log level from flag + flag.Parse() + parsedLevel, _ := zerolog.ParseLevel(*flagLevel) + zerolog.SetGlobalLevel(parsedLevel) + log.Logger = zerolog.New(logSplitter{}).With().Timestamp().Logger() // Initialize Badger db store