From 78779ebcbb819eaef719b9ad9ca37c85ccfdbf70 Mon Sep 17 00:00:00 2001
From: Sven Rebhan <srebhan@influxdata.com>
Date: Mon, 2 Sep 2024 21:40:00 +0200
Subject: [PATCH] fix(logger): Handle trace level for standard log

---
 logger/stdlog_redirector.go | 36 ++++++++++++++----------------------
 1 file changed, 14 insertions(+), 22 deletions(-)

diff --git a/logger/stdlog_redirector.go b/logger/stdlog_redirector.go
index 4c34c1d70c768..7615ad993a853 100644
--- a/logger/stdlog_redirector.go
+++ b/logger/stdlog_redirector.go
@@ -2,11 +2,8 @@ package logger
 
 import (
 	"bytes"
-	"regexp"
 )
 
-var prefixRegex = regexp.MustCompile("^[DIWE]!")
-
 type stdlogRedirector struct {
 	log logger
 }
@@ -14,26 +11,21 @@ type stdlogRedirector struct {
 func (s *stdlogRedirector) Write(b []byte) (n int, err error) {
 	msg := bytes.Trim(b, " \t\r\n")
 
-	// Extract the log-level indicator; use info by default
-	loc := prefixRegex.FindIndex(b)
-	level := 'I'
-	if loc != nil {
-		level = rune(b[loc[0]])
-		msg = bytes.Trim(msg[loc[1]:], " \t\r\n")
-	}
-
-	// Log with the given level
-	switch level {
-	case 'T':
-		s.log.Trace(string(msg))
-	case 'D':
-		s.log.Debug(string(msg))
-	case 'I':
+	// Check a potential log-level indicator  and log with the given level or
+	// use info by default
+	switch {
+	case bytes.HasPrefix(msg, []byte("E! ")):
+		s.log.Error(string(msg[3:]))
+	case bytes.HasPrefix(msg, []byte("W! ")):
+		s.log.Warn(string(msg[3:]))
+	case bytes.HasPrefix(msg, []byte("I! ")):
+		s.log.Info(string(msg[3:]))
+	case bytes.HasPrefix(msg, []byte("D! ")):
+		s.log.Debug(string(msg[3:]))
+	case bytes.HasPrefix(msg, []byte("T! ")):
+		s.log.Trace(string(msg[3:]))
+	default:
 		s.log.Info(string(msg))
-	case 'W':
-		s.log.Warn(string(msg))
-	case 'E':
-		s.log.Error(string(msg))
 	}
 
 	return len(b), nil