benchmark colored output

ammario · ammario · commit 0abda43f6e51 · 2023-05-05T15:17:21.000-05:00
diff --git a/internal/entryhuman/entry_test.go b/internal/entryhuman/entry_test.go
@@ -1,6 +1,8 @@
 package entryhuman_test
 
 import (
+	"fmt"
+	"io"
 	"io/ioutil"
 	"testing"
 	"time"
@@ -92,3 +94,38 @@ func TestEntry(t *testing.T) {
 		assert.Equal(t, "entry", "\x1b[0m\x1b[0m0001-01-01 00:00:00.000 \x1b[91m[CRITICAL]\x1b[0m\t\x1b[36m<.:0>	\x1b[0m\t\"\"\t{\x1b[34m\"hey\"\x1b[0m: \x1b[32m\"hi\"\x1b[0m}", act)
 	})
 }
+
+func BenchmarkFmt(b *testing.B) {
+	bench := func(b *testing.B, color bool) {
+		nfs := []int{1, 4, 16}
+		for _, nf := range nfs {
+			name := fmt.Sprintf("nf=%v", nf)
+			if color {
+				name = "Colored-" + name
+			}
+			b.Run(name, func(b *testing.B) {
+				fs := make([]slog.Field, nf)
+				for i := 0; i < nf; i++ {
+					fs[i] = slog.F("key", "value")
+				}
+				se := slog.SinkEntry{
+					Level: slog.LevelCritical,
+					Fields: slog.M(
+						fs...,
+					),
+				}
+				w := io.Discard
+				if color {
+					w = entryhuman.ForceColorWriter
+				}
+				b.ResetTimer()
+				b.ReportAllocs()
+				for i := 0; i < b.N; i++ {
+					_ = entryhuman.Fmt(w, se)
+				}
+			})
+		}
+	}
+	bench(b, true)
+	bench(b, false)
+}
