lib/logstorage: add benchmark for streamID.marshalString

Author: valyala

lib/logstorage/stream_id_test.go

@@ -6,23 +6,27 @@ import (
 )
 
 func TestStreamIDMarshalUnmarshalString(t *testing.T) {
-	f := func(sid *streamID) {
+	f := func(sid *streamID, resultExpected string) {
 		t.Helper()
 
-		s := string(sid.marshalString(nil))
+		result := string(sid.marshalString(nil))
+
+		if result != resultExpected {
+			t.Fatalf("unexpected result\ngot\n%q\nwant\n%q", result, resultExpected)
+		}
 
 		var sid2 streamID
-		if !sid2.tryUnmarshalFromString(s) {
-			t.Fatalf("cannot unmarshal streamID from %q", s)
+		if !sid2.tryUnmarshalFromString(result) {
+			t.Fatalf("cannot unmarshal streamID from %q", result)
 		}
 
-		s2 := string(sid2.marshalString(nil))
-		if s != s2 {
-			t.Fatalf("unexpected marshaled streamID; got %s; want %s", s2, s)
+		result2 := string(sid2.marshalString(nil))
+		if result != result2 {
+			t.Fatalf("unexpected marshaled streamID; got %s; want %s", result2, result)
 		}
 	}
 
-	f(&streamID{})
+	f(&streamID{}, "000000000000000000000000000000000000000000000000")
 	f(&streamID{
 		tenantID: TenantID{
 			AccountID: 123,
@@ -32,7 +36,7 @@ func TestStreamIDMarshalUnmarshalString(t *testing.T) {
 			lo: 89,
 			hi: 344334,
 		},
-	})
+	}, "0000007b000001c8000000000005410e0000000000000059")
 }
 
 func TestStreamIDMarshalUnmarshal(t *testing.T) {

lib/logstorage/stream_id_timing_test.go

@@ -0,0 +1,27 @@
+package logstorage
+
+import (
+	"testing"
+)
+
+func BenchmarkStreamIDMarshalString(b *testing.B) {
+	b.ReportAllocs()
+	b.SetBytes(1)
+
+	sid := &streamID{
+		tenantID: TenantID{
+			AccountID: 123,
+			ProjectID: 456,
+		},
+		id: u128{
+			lo: 89,
+			hi: 344334,
+		},
+	}
+	b.RunParallel(func(pb *testing.PB) {
+		var b []byte
+		for pb.Next() {
+			b = sid.marshalString(b[:0])
+		}
+	})
+}