Function benchmarkBatch

ring/ring_test.go:62–111

(b *testing.B, numInstances, numKeys int)

}

func benchmarkBatch(b *testing.B, numInstances, numKeys int) {
	seed := time.Now().UnixNano()
	rnd := rand.New(rand.NewSource(seed))
	gen := NewRandomTokenGeneratorWithSeed(seed)

	// Make a random ring with N instances, and M tokens per ingests
	desc := NewDesc()
	var takenTokens []uint32
	for i := 0; i < numInstances; i++ {
		tokens := gen.GenerateTokens(numTokens, takenTokens)
		takenTokens = append(takenTokens, tokens...)
		desc.AddIngester(fmt.Sprintf("%d", i), fmt.Sprintf("instance-%d", i), strconv.Itoa(i), tokens, ACTIVE, time.Now(), false, time.Time{}, nil)
	}

	cfg := Config{}
	flagext.DefaultValues(&cfg)
	cfg.HeartbeatTimeout = time.Hour // A minute is not enough to run all benchmarks.
	r, err := NewWithStoreClientAndStrategy(cfg, testRingName, testRingKey, nil, NewDefaultReplicationStrategy(), prometheus.NewRegistry(), log.NewNopLogger())
	require.NoError(b, err)
	r.updateRingState(desc)

	ctx := context.Background()
	callback := func(InstanceDesc, []int) error { return deepStack(64) }
	keys := make([]uint32, numKeys)
	// Generate a batch of N random keys, and look them up
	b.ResetTimer()

	// run with `benchstat -col /go` to get stats on different go= options.
	// ref: https://pkg.go.dev/golang.org/x/perf/cmd/benchstat

	b.Run("go=default", func(b *testing.B) {
		for i := 0; i < b.N; i++ {
			generateKeys(rnd, numKeys, keys)
			err := DoBatchWithOptions(ctx, Write, r, keys, callback, DoBatchOptions{})
			require.NoError(b, err)
		}
	})

	b.Run("go=concurrency.ReusableGoroutinesPool", func(b *testing.B) {
		pool := concurrency.NewReusableGoroutinesPool(100)
		b.Cleanup(pool.Close)

		b.ResetTimer()
		for i := 0; i < b.N; i++ {
			generateKeys(rnd, numKeys, keys)
			err := DoBatchWithOptions(ctx, Write, r, keys, callback, DoBatchOptions{Go: pool.Go})
			require.NoError(b, err)
		}
	})
}

func generateKeys(r *rand.Rand, numTokens int, dest []uint32) {
	for i := 0; i < numTokens; i++ {