Function BenchmarkQueueRequest

pkg/scheduler/queue/queue_test.go:75–114

(b *testing.B)

}

func BenchmarkQueueRequest(b *testing.B) {
	const maxOutstandingPerTenant = 2
	const numTenants = 50
	const queriers = 5

	queues := make([]*RequestQueue, 0, b.N)
	users := make([]string, 0, numTenants)
	requests := make([]MockRequest, 0, numTenants)

	for n := 0; b.Loop(); n++ {
		q := NewRequestQueue(maxOutstandingPerTenant,
			prometheus.NewGaugeVec(prometheus.GaugeOpts{}, []string{"user", "priority", "type"}),
			prometheus.NewCounterVec(prometheus.CounterOpts{}, []string{"user", "priority"}),
			MockLimits{MaxOutstanding: 100},
			nil,
		)

		for ix := range queriers {
			q.RegisterQuerierConnection(fmt.Sprintf("querier-%d", ix))
		}

		queues = append(queues, q)

		for j := range numTenants {
			requests = append(requests, MockRequest{id: fmt.Sprintf("%d-%d", n, j)})
			users = append(users, strconv.Itoa(j))
		}
	}

	for n := 0; b.Loop(); n++ {
		for range maxOutstandingPerTenant {
			for j := range numTenants {
				err := queues[n].EnqueueRequest(users[j], requests[j], 0, nil)
				if err != nil {
					b.Fatal(err)
				}
			}
		}
	}
}

func BenchmarkGetNextRequestPriorityQueue(b *testing.B) {
	const maxOutstandingPerTenant = 2