Switch from OMQ_LOG_LEVEL to -l

Author: mkuratczyk

README.md

@@ -88,6 +88,7 @@ messages published with perf-test can be consumed by `omq` or vice versa, and th
       --consumer-uri string                 URI for consuming
   -y, --consumers int                       The number of consumers to start (default 1)
   -h, --help                                help for omq
+  -l, --log-level log-level                 Log level (debug, info, error) (default info)
       --log-out-of-order-messages           Print a log line when a message is received that is older than the previously received message
   -d, --message-durability                  Mark messages as durable (default true)
       --message-priority string             Message priority (0-255, default=unset)

cmd/root.go

@@ -178,11 +178,14 @@ func RootCmd() *cobra.Command {
 			}
 			metricsServer := metrics.GetMetricsServer()
 			metricsServer.Start()
+			log.Setup()
 
 		},
 		PersistentPostRun: func(cmd *cobra.Command, args []string) {
 		},
 	}
+	rootCmd.PersistentFlags().
+		VarP(enumflag.New(&log.Level, "log-level", log.Levels, enumflag.EnumCaseInsensitive), "log-level", "l", "Log level (debug, info, error)")
 	rootCmd.PersistentFlags().StringVarP(&cfg.PublisherUri, "publisher-uri", "", "", "URI for publishing")
 	rootCmd.PersistentFlags().StringVarP(&cfg.ConsumerUri, "consumer-uri", "", "", "URI for consuming")
 	rootCmd.PersistentFlags().IntVarP(&cfg.Publishers, "publishers", "x", 1, "The number of publishers to start")

pkg/log/log.go

@@ -2,27 +2,22 @@ package log
 
 import (
 	"os"
-	"strings"
 
 	"log/slog"
 )
 
 var logger *slog.Logger
 
-func init() {
-	var logLevelEnv = os.Getenv("OMQ_LOG_LEVEL")
-	var logLevel slog.Level
-	switch strings.ToUpper(logLevelEnv) {
-	case "DEBUG":
-		logLevel = slog.LevelDebug
-	case "INFO":
-		logLevel = slog.LevelInfo
-	case "ERROR":
-		logLevel = slog.LevelError
-	default:
-		logLevel = slog.LevelInfo
-	}
-	logger = slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: logLevel}))
+var Levels = map[slog.Level][]string{
+	slog.LevelDebug: {"debug"},
+	slog.LevelInfo:  {"info"},
+	slog.LevelError: {"error"},
+}
+
+var Level slog.Level
+
+func Setup() {
+	logger = slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: Level}))
 }
 
 func Debug(format string, v ...interface{}) {