Function testCheckingOfKeyOwnership

ring/token_range_test.go:192–268

(t *testing.T, randomizeInstanceStates bool)

}

func testCheckingOfKeyOwnership(t *testing.T, randomizeInstanceStates bool) {
	const instancesPerZone = 20
	const numZones = 3
	const numTokens = 512
	const replicationFactor = numZones // This is the only config supported by GetTokenRangesForInstance right now.

	seed := time.Now().UnixNano()
	t.Log("token generator seed:", seed)
	gen := NewRandomTokenGeneratorWithSeed(seed)

	stateRand := rand.New(rand.NewSource(seed))

	// Generate users with different number of tokens
	userTokens := map[string][]uint32{}
	shardSizes := map[string]int{}
	for _, cnt := range []int{1000, 5000, 10000, 25000, 50000, 100000} {
		uid := fmt.Sprintf("%dk", cnt/1000)
		userTokens[uid] = gen.GenerateTokens(cnt, nil)

		shardSize := cnt / 7500
		shardSize = (shardSize / numZones) * numZones // round down to numZones
		if shardSize < numZones {
			shardSize = numZones
		}
		shardSizes[uid] = shardSize
	}

	// Generate ring
	ringDesc := &Desc{Ingesters: generateRingInstances(gen, instancesPerZone*numZones, numZones, numTokens)}

	if randomizeInstanceStates {
		for ins, ing := range ringDesc.Ingesters {
			ing.State = InstanceState(stateRand.Int31n(int32(LEFT))) // LEFT is not state that clients can see, so we don't test it.
			ringDesc.Ingesters[ins] = ing
		}
	}

	ring := Ring{
		cfg:                  Config{HeartbeatTimeout: time.Hour, ZoneAwarenessEnabled: true, SubringCacheDisabled: false, ReplicationFactor: replicationFactor},
		ringDesc:             ringDesc,
		ringTokens:           ringDesc.GetTokens(),
		ringTokensByZone:     ringDesc.getTokensByZone(),
		ringInstanceByToken:  ringDesc.getTokensInfo(),
		ringZones:            getZones(ringDesc.getTokensByZone()),
		shuffledSubringCache: map[subringCacheKey]*Ring{},
		strategy:             nopReplicationStrategy{},
		lastTopologyChange:   time.Now(),
	}

	bufInstances, bufHosts, bufZones := MakeBuffersForGet()

	for uid, tokens := range userTokens {
		shardSize := shardSizes[uid]

		subRing := ring.ShuffleShard(uid, shardSize)
		sr := subRing.(*Ring)

		for instanceID := range sr.ringDesc.Ingesters {