Method AddResponse

modules/frontend/combiner/trace_by_id.go:56–106

(r PipelineResponse)

}

func (c *TraceByIDCombiner) AddResponse(r PipelineResponse) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if c.shouldQuit() {
		return nil
	}

	res := r.HTTPResponse()
	if res.StatusCode == http.StatusNotFound {
		// 404s are not considered errors, so we don't need to do anything.
		return nil
	}
	c.code = res.StatusCode

	if res.StatusCode != http.StatusOK {
		bytesMsg, err := io.ReadAll(res.Body)
		if err != nil {
			return fmt.Errorf("error reading response body: %w", err)
		}
		c.statusMessage = string(bytesMsg)
		return nil
	}

	// Read the body
	buff, err := tempo_io.ReadAllWithEstimate(res.Body, res.ContentLength)
	if err != nil {
		c.statusMessage = internalErrorMsg
		return fmt.Errorf("error reading response body: %w", err)
	}
	_ = res.Body.Close()

	// Unmarshal the body
	resp := &tempopb.TraceByIDResponse{}
	err = resp.Unmarshal(buff)
	if err != nil {
		c.statusMessage = internalErrorMsg
		return fmt.Errorf("error unmarshalling response body: %w", err)
	}

	// Consume the trace
	_, err = c.c.Consume(resp.Trace)
	if errors.Is(err, trace.ErrTraceTooLarge) {
		c.code = http.StatusUnprocessableEntity
		c.statusMessage = fmt.Sprint(err)
		return nil
	}
	c.MetricsCombiner.Combine(resp.Metrics, r)

	return err
}

func (c *TraceByIDCombiner) HTTPFinal() (*http.Response, error) {
	c.mu.Lock()