forked from TrueCloudLab/frostfs-node
[#1450] engine: Add benchmark for Inhume
operation
Signed-off-by: Aleksey Savchuk <a.savchuk@yadro.com>
This commit is contained in:
parent
f7cb6b4d87
commit
b62919f595
1 changed files with 60 additions and 0 deletions
|
@ -2,6 +2,7 @@ package engine
|
|||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
|
||||
|
@ -9,7 +10,11 @@ import (
|
|||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
||||
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
|
||||
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
||||
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
||||
oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
|
||||
objecttest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/test"
|
||||
"github.com/stretchr/testify/require"
|
||||
"golang.org/x/sync/errgroup"
|
||||
)
|
||||
|
||||
func TestStorageEngine_Inhume(t *testing.T) {
|
||||
|
@ -84,3 +89,58 @@ func TestStorageEngine_Inhume(t *testing.T) {
|
|||
require.Empty(t, addrs)
|
||||
})
|
||||
}
|
||||
|
||||
func BenchmarkInhumeMultipart(b *testing.B) {
|
||||
// The benchmark result insignificantly depends on the number of shards,
|
||||
// so doesn't use this as a benchmark parameter, just set it big enough.
|
||||
numShards := 100
|
||||
|
||||
for numObjects := 1; numObjects <= 10000; numObjects *= 10 {
|
||||
b.Run(fmt.Sprintf("%d_objects", numObjects), func(b *testing.B) {
|
||||
benchmarkInhumeMultipart(b, numShards, numObjects)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func benchmarkInhumeMultipart(b *testing.B, numShards, numObjects int) {
|
||||
b.StopTimer()
|
||||
|
||||
for range b.N {
|
||||
engine := testNewEngine(b, WithShardPoolSize(uint32(numObjects))).
|
||||
setShardsNum(b, numShards).
|
||||
prepare(b).
|
||||
engine
|
||||
|
||||
cnt := cidtest.ID()
|
||||
var addrs []oid.Address
|
||||
|
||||
eg := errgroup.Group{}
|
||||
|
||||
for range numObjects {
|
||||
prm := PutPrm{}
|
||||
|
||||
prm.Object = objecttest.Object().Parent()
|
||||
prm.Object.SetContainerID(cnt)
|
||||
prm.Object.SetType(objectSDK.TypeRegular)
|
||||
addrs = append(addrs, object.AddressOf(prm.Object))
|
||||
|
||||
eg.Go(func() error {
|
||||
return engine.Put(context.Background(), prm)
|
||||
})
|
||||
}
|
||||
require.NoError(b, eg.Wait())
|
||||
|
||||
ts := oidtest.Address()
|
||||
ts.SetContainer(cnt)
|
||||
|
||||
prm := InhumePrm{}
|
||||
prm.WithTarget(ts, addrs...)
|
||||
|
||||
b.StartTimer()
|
||||
_, err := engine.Inhume(context.Background(), prm)
|
||||
require.NoError(b, err)
|
||||
b.StopTimer()
|
||||
|
||||
require.NoError(b, engine.Close(context.Background()))
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue