Function statsMerge

pkg/querier/tripperware/merge.go:215–261

statsMerge merge the stats from 2 responses this function is similar to matrixMerge

(shouldSumStats bool, resps []*PrometheusResponse)

// statsMerge merge the stats from 2 responses this function is similar to matrixMerge
func statsMerge(shouldSumStats bool, resps []*PrometheusResponse) *PrometheusResponseStats {
	output := map[int64]*PrometheusResponseQueryableSamplesStatsPerStep{}
	hasStats := false
	var peakSamples int64
	for _, resp := range resps {
		if resp.Data.Stats == nil {
			continue
		}

		hasStats = true
		if resp.Data.Stats.Samples == nil {
			continue
		}

		for _, s := range resp.Data.Stats.Samples.TotalQueryableSamplesPerStep {
			if shouldSumStats {
				if stats, ok := output[s.GetTimestampMs()]; ok {
					stats.Value += s.Value
				} else {
					output[s.GetTimestampMs()] = s
				}
			} else {
				output[s.GetTimestampMs()] = s
			}
		}
		peakSamples = max(peakSamples, resp.Data.Stats.Samples.PeakSamples)
	}

	if !hasStats {
		return nil
	}
	keys := make([]int64, 0, len(output))
	for key := range output {
		keys = append(keys, key)
	}

	slices.Sort(keys)

	result := &PrometheusResponseStats{Samples: &PrometheusResponseSamplesStats{}}
	for _, key := range keys {
		result.Samples.TotalQueryableSamplesPerStep = append(result.Samples.TotalQueryableSamplesPerStep, output[key])
		result.Samples.TotalQueryableSamples += output[key].Value
	}
	result.Samples.PeakSamples = peakSamples

	return result
}

type sortPlan int