configure log level

Author: dmoll1974

cmd/root.go

@@ -30,6 +30,7 @@ import (
 	"os"
 	"os/exec"
 	"os/signal"
+	"strings"
 	"syscall"
 
 	"github.com/perfana/x2i/gatlingparser"
@@ -111,6 +112,15 @@ var rootCmd = &cobra.Command{
 // This is called by main.main(). It only needs to happen once to the rootCmd.
 func Execute() {
 	// Initiating logger before any other processes start
+	level, _ := rootCmd.Flags().GetString("log-level")
+	switch strings.ToUpper(level) {
+	case "DEBUG":
+		l.SetLogLevel(l.DEBUG)
+	case "INFO":
+		l.SetLogLevel(l.INFO)
+	default:
+		l.SetLogLevel(l.ERROR)
+	}
 	logPath, _ := rootCmd.Flags().GetString("log")
 	err := l.InitLogger(logPath)
 	if err != nil {
@@ -132,7 +142,8 @@ func init() {
 	rootCmd.Flags().StringP("password", "p", "", "Password credential for InfluxDB instance")
 	rootCmd.Flags().StringP("database", "b", "", "Database name in InfluxDB")
 	rootCmd.Flags().StringP("testtool", "i", "gatling", "Testtool used, can be gatling, jmeter or k6")
-	rootCmd.Flags().StringP("log", "l", "x2i.log", "File path to x2i log file")
+	rootCmd.PersistentFlags().StringP("log-level", "l", "INFO", "Set log level (DEBUG, INFO, ERROR)")
+	rootCmd.Flags().StringP("log", "o", "x2i.log", "Log file path")
 	rootCmd.Flags().StringP("test-environment", "t", "", "Test environment identifier")
 	rootCmd.Flags().StringP("system-under-test", "y", "", "System under test identifier")
 	rootCmd.Flags().UintP("stop-timeout", "s", 120, "Time (seconds) to exit if no new log lines found")
@@ -141,3 +152,5 @@ func init() {
 	// set up global context
 	ctx, cancel = context.WithCancel(context.Background())
 }
+
+

logger/logger.go

@@ -31,12 +31,19 @@ import (
 	"path/filepath"
 )
 
-// that will add prefixes to log lines
+type LogLevel int
+
+const (
+	DEBUG LogLevel = iota
+	INFO
+	ERROR
+)
+
 var (
-	// logger is a single local logger implementation
-	logger *log.Logger
-	sw     io.Writer
-	ew     io.Writer
+	logger    *log.Logger
+	sw        io.Writer
+	ew        io.Writer
+	logLevel  LogLevel = INFO // Default to INFO level
 )
 
 // InitLogger sets up a new instance of logger that writes to file and STDOUT
@@ -62,44 +69,52 @@ func InitLogger(fileName string) error {
 	return nil
 }
 
-// Errorln writes a line to STDERR and log file prepending message with ERROR
+func SetLogLevel(level LogLevel) {
+	logLevel = level
+}
+
 func Errorln(v ...interface{}) {
+	// ERROR level always shows
 	logger.SetOutput(ew)
 	logger.SetPrefix("ERROR ")
 	logger.Println(v...)
 }
 
-// Errorf writes a formatted output to STDERR and log file prepending message with ERROR
 func Errorf(format string, v ...interface{}) {
+	// ERROR level always shows
 	logger.SetOutput(ew)
 	logger.SetPrefix("ERROR ")
 	logger.Printf(format, v...)
 }
 
-// Infoln writes a line to STDERR and log file prepending message with INFO
 func Infoln(v ...interface{}) {
-	logger.SetOutput(sw)
-	logger.SetPrefix("INFO ")
-	logger.Println(v...)
+	if logLevel >= INFO {
+		logger.SetOutput(sw)
+		logger.SetPrefix("INFO ")
+		logger.Println(v...)
+	}
 }
 
-// Infof writes a formatted output to STDERR and log file prepending message with INFO
 func Infof(format string, v ...interface{}) {
-	logger.SetOutput(sw)
-	logger.SetPrefix("INFO ")
-	logger.Printf(format, v...)
+	if logLevel >= INFO {
+		logger.SetOutput(sw)
+		logger.SetPrefix("INFO ")
+		logger.Printf(format, v...)
+	}
 }
 
-// Debugln writes a line to STDERR and log file prepending message with DEBUG
 func Debugln(v ...interface{}) {
-	logger.SetOutput(sw)
-	logger.SetPrefix("DEBUG ")
-	logger.Println(v...)
+	if logLevel <= DEBUG {
+		logger.SetOutput(sw)
+		logger.SetPrefix("DEBUG ")
+		logger.Println(v...)
+	}
 }
 
-// Debugf writes a formatted output to STDERR and log file prepending message with DEBUG
 func Debugf(format string, v ...interface{}) {
-	logger.SetOutput(sw)
-	logger.SetPrefix("DEBUG ")
-	logger.Printf(format, v...)
-}
+	if logLevel <= DEBUG {
+		logger.SetOutput(sw)
+		logger.SetPrefix("DEBUG ")
+		logger.Printf(format, v...)
+	}
+}