tablets: colocate gc benchmark with tablet benches

Author: dkropachev

tablets/gcshape_bench_test.go

@@ -6,6 +6,7 @@ package tablets
 import (
 	"fmt"
 	"runtime"
+	"runtime/debug"
 	"sync/atomic"
 	"testing"
 )
@@ -65,6 +66,55 @@ func BenchmarkCowTabletList(b *testing.B) {
 	})
 }
 
+func BenchmarkCowTabletListGCShape(b *testing.B) {
+	const (
+		rf                = 3
+		hostsCount        = 16
+		tabletsPerTable   = 100000
+		tokenRangeCount   = int64(tabletsPerTable)
+		lookupBatchSize   = 2048
+		gcEveryIterations = 8
+	)
+
+	oldGCPercent := debug.SetGCPercent(10)
+	defer debug.SetGCPercent(oldGCPercent)
+
+	hosts := GenerateHostUUIDs(hostsCount)
+	cl := NewCowTabletList()
+	defer cl.Close()
+
+	cl.BulkAddTablets(createTablets("ks", "tbl", hosts, rf, tabletsPerTable, tokenRangeCount))
+	cl.Flush()
+
+	runtime.GC()
+	runtime.GC()
+
+	var before, after runtime.MemStats
+	runtime.ReadMemStats(&before)
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		for j := 0; j < lookupBatchSize; j++ {
+			token := int64((i*lookupBatchSize + j) % tabletsPerTable)
+			_, _ = cl.FindTabletForToken("ks", "tbl", token)
+		}
+		if i%gcEveryIterations == 0 {
+			runtime.GC()
+		}
+	}
+	b.StopTimer()
+
+	runtime.GC()
+	runtime.GC()
+	runtime.KeepAlive(cl)
+	runtime.ReadMemStats(&after)
+
+	b.ReportMetric(float64(after.HeapObjects), "heap_objects")
+	b.ReportMetric(float64(after.HeapAlloc), "heap_bytes")
+	b.ReportMetric(float64(after.NumGC-before.NumGC), "gc_cycles")
+	b.ReportMetric(float64(after.PauseTotalNs-before.PauseTotalNs), "gc_pause_ns")
+}
+
 func runCowTabletListTestSuit(b *testing.B, name string, hostsCount, parallelism, rf, totalTablets, extraTables int) {
 	b.Run(name, func(b *testing.B) {