Method collectAttributeStats

cmd/tempo-cli/cmd-gen-attrindex.go:95–132

()

}

func (cmd *attrIndexCmd) collectAttributeStats() (*fileStats, error) {
	stats := fileStats{
		Attributes: make(map[string]attributeInfo, 200),
	}

	in, pf, err := openParquetFile(cmd.In)
	if err != nil {
		return nil, err
	}
	defer in.Close()

	reader := parquet.NewGenericReader[vp4.Trace](pf)
	defer reader.Close()

	var (
		traceBuffer = make([]vp4.Trace, 1024)
		readCount   int
	)

	for {
		readCount, err = reader.Read(traceBuffer)
		if err != nil {
			if !errors.Is(err, io.EOF) {
				return nil, err
			}
			break
		}

		if readCount > 0 {
			cmd.collectAttributeStatsForTraces(&stats, traceBuffer[:readCount])
		}
	}
	if readCount > 0 {
		cmd.collectAttributeStatsForTraces(&stats, traceBuffer[:readCount])
	}

	return &stats, nil
}

func (cmd *attrIndexCmd) collectAttributeStatsForTraces(stats *fileStats, traces []vp4.Trace) {
	row := pq.EmptyRowNumber()