Function preparePartitionRingWithActivePartitions

ring/partition_ring_test.go:1162–1176

(partitions int)

var cachedRings = map[int]*PartitionRing{}

func preparePartitionRingWithActivePartitions(partitions int) *PartitionRing {
	pr := cachedRings[partitions]
	if pr != nil {
		return pr
	}

	prd := NewPartitionRingDesc()

	for pid := 0; pid < partitions; pid++ {
		prd.AddPartition(int32(pid), PartitionActive, time.Time{}) // benchmark doesn't use state or time.
	}
	pr, _ = NewPartitionRing(*prd)
	cachedRings[partitions] = pr
	return pr
}

func BenchmarkPartitionRingGetTokenRangesForPartition(b *testing.B) {
	partitionsCount := []int{1, 3, 9, 27, 81, 243, 729}