Function newAsyncTraceIDSharder

modules/frontend/traceid_sharder.go:30–47

(cfg *TraceByIDConfig, jobsPerQuery *prometheus.HistogramVec, logger log.Logger)

}

func newAsyncTraceIDSharder(cfg *TraceByIDConfig, jobsPerQuery *prometheus.HistogramVec, logger log.Logger) pipeline.AsyncMiddleware[combiner.PipelineResponse] {
	return pipeline.AsyncMiddlewareFunc[combiner.PipelineResponse](func(next pipeline.AsyncRoundTripper[combiner.PipelineResponse]) pipeline.AsyncRoundTripper[combiner.PipelineResponse] {
		// Calculate block boundaries:
		// - If external is enabled: N-2 block shards (1 ingester + 1 external + N-2 blocks = N total)
		// - If external is disabled: N-1 block shards (1 ingester + N-1 blocks = N total)
		numBlockShards := cfg.QueryShards - 1
		if cfg.ExternalEnabled {
			numBlockShards = cfg.QueryShards - 2
		}
		return asyncTraceSharder{
			next:            next,
			cfg:             cfg,
			logger:          logger,
			blockBoundaries: blockboundary.CreateBlockBoundaries(numBlockShards),
			jobsPerQuery:    jobsPerQuery,
		}
	})
}

// RoundTrip implements http.RoundTripper
func (s asyncTraceSharder) RoundTrip(pipelineRequest pipeline.Request) (pipeline.Responses[combiner.PipelineResponse], error) {