diff --git a/tsdb/head_bench_test.go b/tsdb/head_bench_test.go index 0dd30b55d..a8595ed8c 100644 --- a/tsdb/head_bench_test.go +++ b/tsdb/head_bench_test.go @@ -17,11 +17,11 @@ import ( "io/ioutil" "os" "strconv" - "sync/atomic" "testing" "github.com/prometheus/prometheus/pkg/labels" "github.com/prometheus/prometheus/util/testutil" + "go.uber.org/atomic" ) func BenchmarkHeadStripeSeriesCreate(b *testing.B) { @@ -51,11 +51,11 @@ func BenchmarkHeadStripeSeriesCreateParallel(b *testing.B) { testutil.Ok(b, err) defer h.Close() - var count int64 + var count atomic.Int64 b.RunParallel(func(pb *testing.PB) { for pb.Next() { - i := atomic.AddInt64(&count, 1) + i := count.Inc() h.getOrCreate(uint64(i), labels.FromStrings("a", strconv.Itoa(int(i)))) } })