VictoriaMetrics/lib/logstorage/stats_median.go

package logstorage

import (
	"slices"
	"unsafe"
)

type statsMedian struct {
	fields       []string
	containsStar bool
}

func (sm *statsMedian) String() string {
	return "median(" + fieldNamesString(sm.fields) + ")"
}

func (sm *statsMedian) neededFields() []string {
	return sm.fields
}

func (sm *statsMedian) newStatsProcessor() (statsProcessor, int) {
	smp := &statsMedianProcessor{
		sqp: &statsQuantileProcessor{
			sq: &statsQuantile{
				fields:       sm.fields,
				containsStar: sm.containsStar,
				phi:          0.5,
			},
		},
	}
	return smp, int(unsafe.Sizeof(*smp)) + int(unsafe.Sizeof(*smp.sqp)) + int(unsafe.Sizeof(*smp.sqp.sq))
}

type statsMedianProcessor struct {
	sqp *statsQuantileProcessor
}

func (smp *statsMedianProcessor) updateStatsForAllRows(br *blockResult) int {
	return smp.sqp.updateStatsForAllRows(br)
}

func (smp *statsMedianProcessor) updateStatsForRow(br *blockResult, rowIdx int) int {
	return smp.sqp.updateStatsForRow(br, rowIdx)
}

func (smp *statsMedianProcessor) mergeState(sfp statsProcessor) {
	src := sfp.(*statsMedianProcessor)
	smp.sqp.mergeState(src.sqp)
}

func (smp *statsMedianProcessor) finalizeStats() string {
	return smp.sqp.finalizeStats()
}

func parseStatsMedian(lex *lexer) (*statsMedian, error) {
	fields, err := parseFieldNamesForStatsFunc(lex, "median")
	if err != nil {
		return nil, err
	}
	sm := &statsMedian{
		fields:       fields,
		containsStar: slices.Contains(fields, "*"),
	}
	return sm, nil
}