From 35019d32bd408b7b67dd4ad8511392b66efe7059 Mon Sep 17 00:00:00 2001 From: Damien Mathieu <42@dmathieu.com> Date: Thu, 22 Sep 2022 17:02:13 +0200 Subject: [PATCH] Add prometheus exporter benchmark (#3213) * add prometheus exporter benchmark * Update exporters/prometheus/benchmark_test.go Co-authored-by: Tyler Yahn * Update exporters/prometheus/benchmark_test.go Co-authored-by: Tyler Yahn Co-authored-by: Tyler Yahn Co-authored-by: Chester Cheung Co-authored-by: Tyler Yahn --- exporters/prometheus/benchmark_test.go | 56 ++++++++++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 exporters/prometheus/benchmark_test.go diff --git a/exporters/prometheus/benchmark_test.go b/exporters/prometheus/benchmark_test.go new file mode 100644 index 000000000..dee0814ed --- /dev/null +++ b/exporters/prometheus/benchmark_test.go @@ -0,0 +1,56 @@ +// Copyright The OpenTelemetry Authors +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package prometheus + +import ( + "context" + "fmt" + "testing" + + "github.com/prometheus/client_golang/prometheus" + "github.com/stretchr/testify/require" + + "go.opentelemetry.io/otel/sdk/metric" +) + +func benchmarkCollect(b *testing.B, n int) { + ctx := context.Background() + exporter := New() + provider := metric.NewMeterProvider(metric.WithReader(exporter)) + meter := provider.Meter("testmeter") + + registry := prometheus.NewRegistry() + err := registry.Register(exporter.Collector) + require.NoError(b, err) + + for i := 0; i < n; i++ { + counter, err := meter.SyncFloat64().Counter(fmt.Sprintf("foo_%d", i)) + require.NoError(b, err) + counter.Add(ctx, float64(i)) + } + + b.ReportAllocs() + b.ResetTimer() + for i := 0; i < b.N; i++ { + _, err := registry.Gather() + require.NoError(b, err) + } +} + +func BenchmarkCollect1(b *testing.B) { benchmarkCollect(b, 1) } +func BenchmarkCollect10(b *testing.B) { benchmarkCollect(b, 10) } +func BenchmarkCollect100(b *testing.B) { benchmarkCollect(b, 100) } +func BenchmarkCollect1000(b *testing.B) { benchmarkCollect(b, 1000) } +func BenchmarkCollect10000(b *testing.B) { benchmarkCollect(b, 10000) }