mirror of
https://github.com/netbirdio/netbird.git
synced 2025-08-09 23:27:58 +02:00
[management] fix github run id (#3705)
This commit is contained in:
@ -16,7 +16,6 @@ import (
|
||||
|
||||
"github.com/golang-jwt/jwt"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/push"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
|
||||
@ -75,6 +74,14 @@ const (
|
||||
OperationGetAll = "get_all"
|
||||
)
|
||||
|
||||
var BenchmarkDuration = prometheus.NewGaugeVec(
|
||||
prometheus.GaugeOpts{
|
||||
Name: "benchmark_duration_ms",
|
||||
Help: "Benchmark duration per op in ms",
|
||||
},
|
||||
[]string{"module", "operation", "test_case", "branch"},
|
||||
)
|
||||
|
||||
type TB interface {
|
||||
Cleanup(func())
|
||||
Helper()
|
||||
@ -324,38 +331,15 @@ func EvaluateBenchmarkResults(b *testing.B, testCase string, duration time.Durat
|
||||
b.Fatalf("environment variable GIT_BRANCH is not set")
|
||||
}
|
||||
|
||||
storeEngine := os.Getenv("NETBIRD_STORE_ENGINE")
|
||||
if storeEngine == "" {
|
||||
b.Fatalf("environment variable NETBIRD_STORE_ENGINE is not set")
|
||||
}
|
||||
|
||||
if recorder.Code != http.StatusOK {
|
||||
b.Fatalf("Benchmark %s failed: unexpected status code %d", testCase, recorder.Code)
|
||||
}
|
||||
|
||||
msPerOp := float64(duration.Nanoseconds()) / float64(b.N) / 1e6
|
||||
|
||||
gauge := prometheus.NewGauge(prometheus.GaugeOpts{
|
||||
Name: "benchmark_duration_ms",
|
||||
Help: "Benchmark duration per op in ms",
|
||||
ConstLabels: prometheus.Labels{
|
||||
"store_engine": storeEngine,
|
||||
"module": module,
|
||||
"operation": operation,
|
||||
"test_case": testCase,
|
||||
"branch": branch,
|
||||
},
|
||||
})
|
||||
|
||||
gauge := BenchmarkDuration.WithLabelValues(module, operation, testCase, branch)
|
||||
gauge.Set(msPerOp)
|
||||
|
||||
if err := push.New("http://localhost:9091", "api_benchmark").
|
||||
Collector(gauge).
|
||||
Grouping("ci_run", os.Getenv("GITHUB_RUN_ID")).
|
||||
Push(); err != nil {
|
||||
b.Fatalf("Could not push benchmark metric: %v", err)
|
||||
}
|
||||
|
||||
b.ReportMetric(msPerOp, "ms/op")
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user