feat: add log level flag to rules-evaluator

hsmatulis · bwplotka · commit 4a68fa47649d · 2024-10-31T18:14:23.000+01:00
Signed-off-by: Henrique Spanoudis Matulis &lt;hmatulis@google.com&gt;
diff --git a/cmd/rule-evaluator/main.go b/cmd/rule-evaluator/main.go
@@ -102,6 +102,7 @@ func main() {
 	logger = log.With(logger, "caller", log.DefaultCaller)
 
 	a := kingpin.New("rule", "The Prometheus Rule Evaluator")
+	logLevel := a.Flag("log-level", "The level of logging").Default("info").Enum("debug", "info", "warn", "error")
 
 	a.HelpFlag.Short('h')
 
@@ -159,6 +160,16 @@ func main() {
 		a.Usage(os.Args[1:])
 		os.Exit(2)
 	}
+	switch strings.ToLower(*logLevel) {
+	case "debug":
+		logger = level.NewFilter(logger, level.AllowDebug())
+	case "warn":
+		logger = level.NewFilter(logger, level.AllowWarn())
+	case "error":
+		logger = level.NewFilter(logger, level.AllowError())
+	default:
+		logger = level.NewFilter(logger, level.AllowInfo())
+	}
 
 	if err := defaultEvaluatorOpts.validate(); err != nil {
 		_ = level.Error(logger).Log("msg", "invalid command line argument", "err", err)
