prometheus/pkg/textparse/parse.go

//go:generate golex -o=lex.l.go lex.l
package textparse

import (
	"errors"
	"io"
	"reflect"
	"sort"
	"unsafe"

	"github.com/prometheus/prometheus/pkg/labels"
)

type lexer struct {
	b      []byte
	i      int
	vstart int

	err          error
	val          float64
	offsets      []int
	mstart, mend int
}

const eof = 0

func (l *lexer) next() byte {
	l.i++
	if l.i >= len(l.b) {
		l.err = io.EOF
		return eof
	}
	c := l.b[l.i]
	return c
}

func (l *lexer) Error(es string) {
	l.err = errors.New(es)
}

type Parser struct {
	l   *lexer
	err error
	val float64
}

func New(b []byte) *Parser {
	return &Parser{l: &lexer{b: b}}
}

func (p *Parser) Next() bool {
	switch p.l.Lex() {
	case 0, -1:
		return false
	case 1:
		return true
	}
	panic("unexpected")
}

func (p *Parser) At() ([]byte, *int64, float64) {
	return p.l.b[p.l.mstart:p.l.mend], nil, p.l.val
}

func (p *Parser) Err() error {
	if p.err != nil {
		return p.err
	}
	if p.l.err == io.EOF {
		return nil
	}
	return p.l.err
}

func (p *Parser) Metric(l *labels.Labels) {
	*l = append(*l, labels.Label{
		Name:  labels.MetricName,
		Value: string(p.l.b[p.l.mstart:p.l.offsets[0]]),
	})

	for i := 1; i < len(p.l.offsets); i += 3 {
		a, b, c := p.l.offsets[i], p.l.offsets[i+1], p.l.offsets[i+2]

		*l = append(*l, labels.Label{
			Name:  string(p.l.b[a:b]),
			Value: string(p.l.b[b+2 : c]),
		})
	}

	sort.Sort((*l)[1:])
}

func yoloString(b []byte) string {
	sh := (*reflect.SliceHeader)(unsafe.Pointer(&b))

	h := reflect.StringHeader{
		Data: sh.Data,
		Len:  sh.Len,
	}
	return *((*string)(unsafe.Pointer(&h)))
}