76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package shard
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
shardconfig "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-node/config/engine/shard"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
|
|
oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func BenchmarkRefillMetabase(b *testing.B) {
|
|
b.Run("100 objects", func(b *testing.B) {
|
|
benchRefillMetabase(b, 100)
|
|
})
|
|
|
|
b.Run("1000 objects", func(b *testing.B) {
|
|
benchRefillMetabase(b, 1000)
|
|
})
|
|
|
|
b.Run("2000 objects", func(b *testing.B) {
|
|
benchRefillMetabase(b, 2000)
|
|
})
|
|
|
|
b.Run("5000 objects", func(b *testing.B) {
|
|
benchRefillMetabase(b, 5000)
|
|
})
|
|
}
|
|
|
|
func benchRefillMetabase(b *testing.B, objectsCount int) {
|
|
sh := newCustomShard(b, false, shardOptions{
|
|
additionalShardOptions: []Option{WithRefillMetabaseWorkersCount(shardconfig.RefillMetabaseWorkersCountDefault)},
|
|
})
|
|
|
|
defer func() { require.NoError(b, sh.Close(context.Background())) }()
|
|
|
|
var putPrm PutPrm
|
|
|
|
for range objectsCount / 2 {
|
|
obj := testutil.GenerateObject()
|
|
testutil.AddAttribute(obj, "foo", "bar")
|
|
testutil.AddPayload(obj, 1<<5) // blobvnicza tree obj
|
|
|
|
putPrm.SetObject(obj)
|
|
|
|
_, err := sh.Put(context.Background(), putPrm)
|
|
require.NoError(b, err)
|
|
}
|
|
|
|
for range objectsCount / 2 {
|
|
obj := testutil.GenerateObject()
|
|
testutil.AddAttribute(obj, "foo", "bar")
|
|
obj.SetID(oidtest.ID())
|
|
testutil.AddPayload(obj, 1<<20) // fstree obj
|
|
|
|
putPrm.SetObject(obj)
|
|
|
|
_, err := sh.Put(context.Background(), putPrm)
|
|
require.NoError(b, err)
|
|
}
|
|
|
|
require.NoError(b, sh.Close(context.Background()))
|
|
require.NoError(b, os.Remove(sh.metaBase.DumpInfo().Path))
|
|
|
|
require.NoError(b, sh.Open(context.Background()))
|
|
sh.cfg.refillMetabase = true
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
require.NoError(b, sh.Init(context.Background()))
|
|
|
|
require.NoError(b, sh.Close(context.Background()))
|
|
}
|