expfmt_bench_test.go

package metrics

import (
	"bytes"
	"testing"
)

func BenchmarkExpfmtWriter(b *testing.B) {
	w := ExpfmtWriter{
		b: bytes.NewBuffer(nil),
	}

	family := MustIdent("really_cool_metric")
	tags := MustTags(
		"foo", "bar",
		"a", "b",
	)
	manyTags := MustTags(
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
		"foo", "bar",
	)

	b.Run("name", func(b *testing.B) {
		w.WriteMetricName(MetricName{Family: family})

		b.ReportAllocs()
		for b.Loop() {
			w.b.Reset()

			w.WriteMetricName(MetricName{Family: family})
			b.SetBytes(int64(w.b.Len()))
		}
	})

	b.Run("name with tags", func(b *testing.B) {
		w.WriteMetricName(MetricName{Family: family, Tags: tags})

		b.ReportAllocs()
		for b.Loop() {
			w.b.Reset()
			w.WriteMetricName(MetricName{Family: family, Tags: tags})
			b.SetBytes(int64(w.b.Len()))
		}
	})

	b.Run("name with many tags", func(b *testing.B) {
		w.WriteMetricName(MetricName{Family: family, Tags: manyTags})

		b.ReportAllocs()
		for b.Loop() {
			w.b.Reset()
			w.WriteMetricName(MetricName{Family: family, Tags: manyTags})
			b.SetBytes(int64(w.b.Len()))
		}
	})

	b.Run("uint64", func(b *testing.B) {
		const value = 100
		w.WriteMetricName(MetricName{Family: family, Tags: tags})
		w.WriteUint64(value)

		b.ReportAllocs()
		for b.Loop() {
			w.b.Reset()
			w.WriteMetricName(MetricName{Family: family, Tags: tags})
			w.WriteUint64(value)
			b.SetBytes(int64(w.b.Len()))
		}
	})

	b.Run("float64", func(b *testing.B) {
		const value = 100.5
		w.WriteMetricName(MetricName{Family: family, Tags: tags})
		w.WriteFloat64(value)

		b.ReportAllocs()
		for b.Loop() {
			w.b.Reset()
			w.WriteMetricName(MetricName{Family: family, Tags: tags})
			w.WriteFloat64(value)
			b.SetBytes(int64(w.b.Len()))
		}
	})
}