mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2026-06-18 00:03:47 +03:00
1572e1e5c3
This commit adds the following changes: * use consistent hashing for the remote write sharding. * properly count metric of remote write samples drop rate when `shardByURL` was enabled. Related issues: https://github.com/VictoriaMetrics/VictoriaMetrics/issues/8546 https://github.com/VictoriaMetrics/VictoriaMetrics/issues/8702
41 lines
673 B
Go
41 lines
673 B
Go
package consistenthash
|
|
|
|
import (
|
|
"math/rand"
|
|
"sync/atomic"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkConsistentHash(b *testing.B) {
|
|
nodes := []string{
|
|
"node1",
|
|
"node2",
|
|
"node3",
|
|
"node4",
|
|
}
|
|
rh := NewConsistentHash(nodes, 0)
|
|
b.ReportAllocs()
|
|
b.SetBytes(int64(len(benchKeys)))
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
sum := 0
|
|
for pb.Next() {
|
|
for _, k := range benchKeys {
|
|
idx := rh.GetNodeIdx(k, nil)
|
|
sum += idx
|
|
}
|
|
}
|
|
BenchSink.Add(uint64(sum))
|
|
})
|
|
}
|
|
|
|
var benchKeys = func() []uint64 {
|
|
r := rand.New(rand.NewSource(1))
|
|
keys := make([]uint64, 10000)
|
|
for i := 0; i < len(keys); i++ {
|
|
keys[i] = r.Uint64()
|
|
}
|
|
return keys
|
|
}()
|
|
|
|
var BenchSink atomic.Uint64
|