Function AddSeriesConcurrentBench

pkg/util/limiter/query_limiter_test.go:148–189

(b *testing.B, batchSize int)

}

func AddSeriesConcurrentBench(b *testing.B, batchSize int) {
	b.ResetTimer()
	const (
		metricName = "test_metric"
	)

	limiter := NewQueryLimiter(b.N+1, 0, 0, 0)

	// Concurrent goroutines trying to add duplicated series
	const numWorkers = 100
	var wg sync.WaitGroup

	worker := func(w int) {
		defer wg.Done()
		var series []labels.Labels
		for i := 0; b.Loop(); i++ {
			series = append(series,
				labels.FromMap(map[string]string{
					labels.MetricName: metricName + "_1",
					"series1":         fmt.Sprint(i),
				}))
		}

		for i := 0; i < len(series); i += batchSize {
			s := make([][]cortexpb.LabelAdapter, 0, batchSize)
			j := min(i+batchSize, len(series))
			for k := i; k < j; k++ {
				s = append(s, cortexpb.FromLabelsToLabelAdapters(series[k]))
			}

			err := limiter.AddSeries(s...)
			assert.NoError(b, err)
		}
	}

	for w := 1; w <= numWorkers; w++ {
		wg.Add(1)
		go worker(w)
	}

	wg.Wait()
}