prometheus/pkg/textparse/parse_test.go

package textparse

import (
	"fmt"
	"io/ioutil"
	"os"
	"testing"

	"github.com/prometheus/prometheus/pkg/labels"
	"github.com/stretchr/testify/require"
)

func TestParse(t *testing.T) {
	input := `# HELP go_gc_duration_seconds A summary of the GC invocation durations.
# TYPE go_gc_duration_seconds summary
go_gc_duration_seconds{quantile="0"} 4.9351e-05
go_gc_duration_seconds{quantile="0.25"} 7.424100000000001e-05
go_gc_duration_seconds{quantile="0.5",a="b"} 8.3835e-05
go_gc_duration_seconds_count 99
# HELP go_goroutines Number of goroutines that currently exist.
# TYPE go_goroutines gauge
go_goroutines 33`

	exp := []struct {
		lset labels.Labels
		m    string
		v    float64
	}{
		{
			m:    `go_gc_duration_seconds{quantile="0"}`,
			v:    4.9351e-05,
			lset: labels.FromStrings("__name__", "go_gc_duration_seconds", "quantile", "0"),
		}, {
			m:    `go_gc_duration_seconds{quantile="0.25"}`,
			v:    7.424100000000001e-05,
			lset: labels.FromStrings("__name__", "go_gc_duration_seconds", "quantile", "0.25"),
		}, {
			m:    `go_gc_duration_seconds{quantile="0.5",a="b"}`,
			v:    8.3835e-05,
			lset: labels.FromStrings("__name__", "go_gc_duration_seconds", "quantile", "0.5", "a", "b"),
		}, {
			m:    `go_gc_duration_seconds_count`,
			v:    99,
			lset: labels.FromStrings("__name__", "go_gc_duration_seconds_count"),
		}, {
			m:    `go_goroutines`,
			v:    33,
			lset: labels.FromStrings("__name__", "go_goroutines"),
		},
	}

	p := New([]byte(input))
	i := 0

	var res labels.Labels

	for p.Next() {
		m, _, v := p.At()

		p.Metric(&res)

		require.Equal(t, exp[i].m, string(m))
		require.Equal(t, exp[i].v, v)
		require.Equal(t, exp[i].lset, res)

		i++
		res = res[:0]
	}

	require.NoError(t, p.Err())

}

func BenchmarkParse(b *testing.B) {
	f, err := os.Open("testdata.txt")
	require.NoError(b, err)
	defer f.Close()

	var tbuf []byte

	buf, err := ioutil.ReadAll(f)
	require.NoError(b, err)

	for i := 0; i*527 < b.N; i++ {
		tbuf = append(tbuf, buf...)
		tbuf = append(tbuf, '\n')
	}

	p := New(tbuf)
	i := 0
	var m labels.Labels
	total := 0

	b.ResetTimer()
	b.ReportAllocs()
	b.SetBytes(int64(len(tbuf)))

	for p.Next() && i < b.N {
		p.At()

		res := make(labels.Labels, 0, 5)
		p.Metric(&res)

		total += len(res)
		i++
		res = res[:0]
	}
	_ = m
	fmt.Println(total)
	require.NoError(b, p.Err())
}