forked from TrueCloudLab/frostfs-node
[#1491] engine/test: Move BenchmarkExists
to exists_test.go
Move `BenchmarkExists` from `engine_test.go` to `exists_test.go` for better organization and clarity. Signed-off-by: Aleksey Savchuk <a.savchuk@yadro.com>
This commit is contained in:
parent
c6066d6ee4
commit
7ef36749d0
2 changed files with 51 additions and 44 deletions
|
@ -10,17 +10,11 @@ import (
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/fstree"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/fstree"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/teststore"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/teststore"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
|
|
||||||
meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
|
meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/pilorama"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/pilorama"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger/test"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger/test"
|
||||||
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
|
|
||||||
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
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"
|
|
||||||
"git.frostfs.info/TrueCloudLab/hrw"
|
|
||||||
"github.com/panjf2000/ants/v2"
|
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -30,44 +24,6 @@ func (s epochState) CurrentEpoch() uint64 {
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func BenchmarkExists(b *testing.B) {
|
|
||||||
b.Run("2 shards", func(b *testing.B) {
|
|
||||||
benchmarkExists(b, 2)
|
|
||||||
})
|
|
||||||
b.Run("4 shards", func(b *testing.B) {
|
|
||||||
benchmarkExists(b, 4)
|
|
||||||
})
|
|
||||||
b.Run("8 shards", func(b *testing.B) {
|
|
||||||
benchmarkExists(b, 8)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func benchmarkExists(b *testing.B, shardNum int) {
|
|
||||||
e := testNewEngine(b).setShardsNum(b, shardNum).engine
|
|
||||||
defer func() { require.NoError(b, e.Close(context.Background())) }()
|
|
||||||
|
|
||||||
addr := oidtest.Address()
|
|
||||||
for range 100 {
|
|
||||||
obj := testutil.GenerateObjectWithCID(cidtest.ID())
|
|
||||||
err := Put(context.Background(), e, obj, false)
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
b.ReportAllocs()
|
|
||||||
b.ResetTimer()
|
|
||||||
for range b.N {
|
|
||||||
var shPrm shard.ExistsPrm
|
|
||||||
shPrm.Address = addr
|
|
||||||
shPrm.ParentAddress = oid.Address{}
|
|
||||||
ok, _, err := e.exists(context.Background(), shPrm)
|
|
||||||
if err != nil || ok {
|
|
||||||
b.Fatalf("%t %v", ok, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type testEngineWrapper struct {
|
type testEngineWrapper struct {
|
||||||
engine *StorageEngine
|
engine *StorageEngine
|
||||||
shardIDs []*shard.ID
|
shardIDs []*shard.ID
|
||||||
|
|
51
pkg/local_object_storage/engine/exists_test.go
Normal file
51
pkg/local_object_storage/engine/exists_test.go
Normal file
|
@ -0,0 +1,51 @@
|
||||||
|
package engine
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
|
||||||
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
||||||
|
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
|
||||||
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
||||||
|
oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func BenchmarkExists(b *testing.B) {
|
||||||
|
b.Run("2 shards", func(b *testing.B) {
|
||||||
|
benchmarkExists(b, 2)
|
||||||
|
})
|
||||||
|
b.Run("4 shards", func(b *testing.B) {
|
||||||
|
benchmarkExists(b, 4)
|
||||||
|
})
|
||||||
|
b.Run("8 shards", func(b *testing.B) {
|
||||||
|
benchmarkExists(b, 8)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func benchmarkExists(b *testing.B, shardNum int) {
|
||||||
|
e := testNewEngine(b).setShardsNum(b, shardNum).engine
|
||||||
|
defer func() { require.NoError(b, e.Close(context.Background())) }()
|
||||||
|
|
||||||
|
addr := oidtest.Address()
|
||||||
|
for range 100 {
|
||||||
|
obj := testutil.GenerateObjectWithCID(cidtest.ID())
|
||||||
|
err := Put(context.Background(), e, obj, false)
|
||||||
|
if err != nil {
|
||||||
|
b.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
b.ReportAllocs()
|
||||||
|
b.ResetTimer()
|
||||||
|
for range b.N {
|
||||||
|
var shPrm shard.ExistsPrm
|
||||||
|
shPrm.Address = addr
|
||||||
|
shPrm.ParentAddress = oid.Address{}
|
||||||
|
ok, _, err := e.exists(context.Background(), shPrm)
|
||||||
|
if err != nil || ok {
|
||||||
|
b.Fatalf("%t %v", ok, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue