From a886e5b0f86971488e642369d307416e3cc5f3c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=94=AF=E5=B0=A4=E5=88=9A=5BC=26N=20Dev=5D?=
 <ning.yougang@navercorp.com>
Date: Tue, 14 May 2024 02:00:24 -0400
Subject: [PATCH] Add ExcludeKey option to TextFormatter

---
 text_formatter.go      | 19 +++++++++++++++----
 text_formatter_test.go | 19 +++++++++++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/text_formatter.go b/text_formatter.go
index be2c6ef..0fecc58 100644
--- a/text_formatter.go
+++ b/text_formatter.go
@@ -28,6 +28,10 @@ func init() {
 
 // TextFormatter formats logs into text
 type TextFormatter struct {
+	// Set to true if want the log not to contain the startingKey, e.g.
+	// 0001-01-01T00:00:00Z warning message
+	ExcludeKey bool
+
 	// Set to true to bypass checking for a TTY before outputting colors.
 	ForceColors bool
 
@@ -320,8 +324,10 @@ func (f *TextFormatter) appendKeyValue(b *bytes.Buffer, key string, value interf
 	if b.Len() > 0 {
 		b.WriteByte(' ')
 	}
-	b.WriteString(key)
-	b.WriteByte('=')
+	if !f.ExcludeKey {
+		b.WriteString(key)
+		b.WriteByte('=')
+	}
 	f.appendValue(b, value)
 }
 
@@ -331,9 +337,14 @@ func (f *TextFormatter) appendValue(b *bytes.Buffer, value interface{}) {
 		stringVal = fmt.Sprint(value)
 	}
 
-	if !f.needsQuoting(stringVal) {
+	// write value without quoting directly when ExcludeKey is true
+	if f.ExcludeKey {
 		b.WriteString(stringVal)
 	} else {
-		b.WriteString(fmt.Sprintf("%q", stringVal))
+		if !f.needsQuoting(stringVal) {
+			b.WriteString(stringVal)
+		} else {
+			b.WriteString(fmt.Sprintf("%q", stringVal))
+		}
 	}
 }
diff --git a/text_formatter_test.go b/text_formatter_test.go
index 5b1cc0a..782e3c1 100644
--- a/text_formatter_test.go
+++ b/text_formatter_test.go
@@ -598,3 +598,22 @@ func TestCustomSorting(t *testing.T) {
 	require.NoError(t, err)
 	require.True(t, strings.HasPrefix(string(b), "prefix="), "format output is %q", string(b))
 }
+
+func TestFormattingNotContainStartingKey(t *testing.T) {
+	tf := &TextFormatter{ExcludeKey: true}
+
+	testCases := []struct {
+		value    string
+		expected string
+	}{
+		{`foo`, "0001-01-01T00:00:00Z panic foo\n"},
+	}
+
+	for _, tc := range testCases {
+		b, _ := tf.Format(WithField("test", tc.value))
+
+		if string(b) != tc.expected {
+			t.Errorf("formatting expected for %q (result was %q instead of %q)", tc.value, string(b), tc.expected)
+		}
+	}
+}