Updated the log logic (#125)

* Updated the log logic This PR replaces the tap.debug flag with a more generic logLevel flag. The logLevel flag can be set to any one of: debug, info, warning, error, fatal, panic. The default logLevel is warning The new logic set the global log level based on the new helm value. --------- Co-authored-by: Ilya Gavrilov <[email protected]>
kubeshark · Dec 17, 2024 · 9cc6305 · 9cc6305
1 parent 05a7e93
commit 9cc6305
Showing 1 changed file with 18 additions and 9 deletions.
diff --git a/main.go b/main.go
@@ -38,8 +38,7 @@ var port = flag.Int("port", 80, "Port number of the HTTP server")
 // capture
 var procfs = flag.String("procfs", "/proc", "The procfs directory, used when mapping host volumes into a container")
 
-// development
-var debug = flag.Bool("debug", false, "Enable debug mode")
+var logLevel = flag.String("loglevel", "warning", "The minimum log level to output. Possible values: debug, info, warning")
 
 var initBPF = flag.Bool("init-bpf", false, "Use to initialize bpf filesystem. Common usage is from init containers.")
 
@@ -49,6 +48,23 @@ var disableTlsLog = flag.Bool("disable-tls-log", false, "Disable tls logging")
 var tracer *Tracer
 
 func main() {
+	()
+
+	// Set log level
+	var level zerolog.Level
+	switch strings.ToLower(*logLevel) {
+	case "debug":
+		level = zerolog.DebugLevel
+	case "info":
+		level = zerolog.InfoLevel
+	case "warning":
+		level = zerolog.WarnLevel
+	default:
+		level = zerolog.WarnLevel
+		log.Warn().Msgf("Invalid log level '%s'. Defaulting to 'warning'.", *logLevel)
+	}
+	zerolog.SetGlobalLevel(level)
+
 	var sentryDSN string
 	if sentrypkg.IsSentryEnabled() {
 		sentryDSN, err := sentrypkg.GetDSN(context.Background(), "tracer", version.Ver)
@@ -72,8 +88,6 @@ func main() {
 			defer sentry.Flush(2 * time.Second)
 		}
 	}
-
-	zerolog.SetGlobalLevel(zerolog.InfoLevel)
 	w, err := zlogsentry.New(
 		sentryDSN,
 	)
@@ -86,13 +100,8 @@ func main() {
 	multi := zerolog.MultiLevelWriter(w, zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339})
 	log.Logger = zerolog.New(multi).With().Timestamp().Caller().Logger()
 
-	flag.Parse()
 	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339}).With().Caller().Logger()
 
-	if *debug {
-		zerolog.SetGlobalLevel(zerolog.DebugLevel)
-	}
-
 	runtimeDebug.SetPanicOnFault(true)
 	defer func() {
 		if err := recover(); err != nil {