Function BenchmarkMemberlistReceiveWithRingDesc

ring/bench/ring_memberlist_test.go:62–107

Benchmark the memberlist receive path when it is being used as the ring backing store.

(b *testing.B)

// Benchmark the memberlist receive path when it is being used as the ring backing store.
func BenchmarkMemberlistReceiveWithRingDesc(b *testing.B) {
	c := ring.GetCodec()

	var cfg memberlist.KVConfig
	flagext.DefaultValues(&cfg)
	cfg.TCPTransport = memberlist.TCPTransportConfig{}
	cfg.Codecs = []codec.Codec{c}

	mkv := memberlist.NewKV(cfg, log.NewNopLogger(), &dnsProviderMock{}, prometheus.NewPedanticRegistry())
	require.NoError(b, services.StartAndAwaitRunning(context.Background(), mkv))
	defer services.StopAndAwaitTerminated(context.Background(), mkv) //nolint:errcheck

	// Build the initial ring state:
	// - The ring isn't actually in use, so the fields such as address/zone are not important.
	// - The number of keys in the store has no impact for this test, so simulate a single ring.
	// - The number of instances in the ring does have a big impact.
	const numInstances = 600
	const numTokens = 128
	initialDesc := ring.NewDesc()
	{
		for i := 0; i < numInstances; i++ {
			tokens := generateUniqueTokens(i, numTokens)
			initialDesc.AddIngester(fmt.Sprintf("instance-%d", i), "127.0.0.1", "zone", tokens, ring.ACTIVE, time.Now(), false, time.Time{}, nil)
		}
		// Send a single update to populate the store.
		msg := encodeMessage(b, "ring", initialDesc)
		mkv.NotifyMsg(msg)
	}

	// Ensure that each received message updates the ring.
	testMsgs := make([][]byte, b.N)
	for i := range testMsgs {
		instance := initialDesc.Ingesters["instance-0"]
		instance.Timestamp = initialDesc.Ingesters["instance-0"].RegisteredTimestamp + int64(i)

		testDesc := ring.NewDesc()
		testDesc.Ingesters["instance-0"] = instance
		testMsgs[i] = encodeMessage(b, "ring", testDesc)
	}

	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		mkv.NotifyMsg(testMsgs[i])
	}
}