c1f1aacffe
There's no reason it should be at the top-level. Change-Id: I35b06e7baa0e425c6ff9a82964d0a1570d4eb6d0
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
||
// See LICENSE for copying information.
|
||
|
||
package testsuite
|
||
|
||
import (
|
||
"path"
|
||
"strconv"
|
||
"testing"
|
||
|
||
"golang.org/x/sync/errgroup"
|
||
|
||
"storj.io/common/testcontext"
|
||
"storj.io/storj/private/kvstore"
|
||
)
|
||
|
||
// RunBenchmarks runs common kvstore.Store benchmarks.
|
||
func RunBenchmarks(b *testing.B, store kvstore.Store) {
|
||
var words = []string{
|
||
"alpha", "beta", "gamma", "delta", "iota", "kappa", "lambda", "mu",
|
||
"άλφα", "βήτα", "γάμμα", "δέλτα", "έψιλον", "ζήτα", "ήτα", "θήτα", "ιώτα", "κάππα", "λάμδα", "μυ",
|
||
"nu", "xi", "omicron", "pi", "rho", "sigma", "tau", "upsilon", "phi", "chi", "psi", "omega",
|
||
"νυ", "ξι", "όμικρον", "πι", "ρώ", "σίγμα", "ταυ", "ύψιλον", "φι", "χι", "ψι", "ωμέγα",
|
||
}
|
||
|
||
words = words[:20] // branching factor
|
||
if testing.Short() {
|
||
words = words[:2]
|
||
}
|
||
|
||
var items kvstore.Items
|
||
|
||
k := 0
|
||
for _, a := range words {
|
||
for _, b := range words {
|
||
for _, c := range words {
|
||
items = append(items, kvstore.Item{
|
||
Key: kvstore.Key(path.Join(a, b, c)),
|
||
Value: kvstore.Value(strconv.Itoa(k)),
|
||
})
|
||
k++
|
||
}
|
||
}
|
||
}
|
||
|
||
ctx := testcontext.New(b)
|
||
defer ctx.Cleanup()
|
||
|
||
defer cleanupItems(b, ctx, store, items)
|
||
|
||
b.Run("Put", func(b *testing.B) {
|
||
b.SetBytes(int64(len(items)))
|
||
for k := 0; k < b.N; k++ {
|
||
var group errgroup.Group
|
||
for _, item := range items {
|
||
key := item.Key
|
||
value := item.Value
|
||
|
||
group.Go(func() error {
|
||
return store.Put(ctx, key, value)
|
||
})
|
||
}
|
||
|
||
if err := group.Wait(); err != nil {
|
||
b.Fatalf("Put: %v", err)
|
||
}
|
||
}
|
||
})
|
||
|
||
b.Run("Get", func(b *testing.B) {
|
||
b.SetBytes(int64(len(items)))
|
||
for k := 0; k < b.N; k++ {
|
||
for _, item := range items {
|
||
_, err := store.Get(ctx, item.Key)
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
})
|
||
}
|