Function callQueryRange

integration/api/query_range_test.go:761–807

(t *testing.T, h *util.TempoHarness, req queryRangeRequest, fn func(*tempopb.QueryRangeResponse, error))

}

func callQueryRange(t *testing.T, h *util.TempoHarness, req queryRangeRequest, fn func(*tempopb.QueryRangeResponse, error)) {
	req.SetDefaults()

	apiClient := h.APIClientHTTP("")
	fn(apiClient.MetricsQueryRange(req.Query, req.Start.UnixNano(), req.End.UnixNano(), req.Step, req.Exemplars))

	step := time.Duration(0)
	if req.Step != "" {
		var err error
		step, err = time.ParseDuration(req.Step)
		require.NoError(t, err)
	}

	grpcClient, ctx, err := h.APIClientGRPC("")
	require.NoError(t, err)

	stream, err := grpcClient.MetricsQueryRange(ctx, &tempopb.QueryRangeRequest{
		Query:     req.Query,
		Start:     uint64(req.Start.UnixNano()),
		End:       uint64(req.End.UnixNano()),
		Step:      uint64(step.Nanoseconds()),
		Exemplars: uint32(req.Exemplars),
	})
	require.NoError(t, err)

	finalResponse := &tempopb.QueryRangeResponse{
		Metrics: &tempopb.SearchMetrics{},
	}
	var finalError error
	for {
		t.Logf("recv")
		resp, err := stream.Recv()
		if resp != nil {
			t.Logf("resp: %+v, count: %d", resp, len(resp.GetSeries()))
			naiveQueryRangeCombine(resp, finalResponse)
			t.Logf("resp: %d, finalResponse: %d", len(resp.GetSeries()), len(finalResponse.GetSeries()))
		}
		if errors.Is(err, io.EOF) {
			break
		}
		if err != nil {
			finalError = err
			break
		}
	}
	fn(finalResponse, finalError)
}

func strptr(s string) *string {
	return &s