Function NewPartitionRingWithOptions

ring/partition_ring.go:82–107

NewPartitionRingWithOptions creates a new PartitionRing with custom options.

(desc PartitionRingDesc, opts PartitionRingOptions)

// NewPartitionRingWithOptions creates a new PartitionRing with custom options.
func NewPartitionRingWithOptions(desc PartitionRingDesc, opts PartitionRingOptions) (*PartitionRing, error) {
	shuffleShardCache, err := newPartitionRingShuffleShardCache(opts.ShuffleShardCacheSize)
	if err != nil {
		return nil, fmt.Errorf("failed to create shuffle shard cache: %w", err)
	}

	ringTokens := desc.tokens()
	partitionByToken := desc.partitionByToken()
	ringPartitionIDs, ringPartitionActive, err := buildRingTokenPartitionLookups(ringTokens, partitionByToken, desc.Partitions)
	if err != nil {
		return nil, err
	}

	return &PartitionRing{
		desc:                  desc,
		ringTokens:            ringTokens,
		partitionByToken:      partitionByToken,
		ringPartitionIDs:      ringPartitionIDs,
		ringPartitionActive:   ringPartitionActive,
		ownersByPartition:     desc.ownersByPartition(),
		activePartitionsCount: desc.activePartitionsCount(),
		maxPartitionID:        desc.maxPartitionID(),
		shuffleShardCache:     shuffleShardCache,
		opts:                  opts,
	}, nil
}

// buildRingTokenPartitionLookups builds two slices parallel to ringTokens:
// - ringPartitionIDs[i] is the partition ID that owns ringTokens[i]