fstree: add bench test to GetRange

Signed-off-by: Andrey Butusov <[email protected]>

Author: End-rey

pkg/local_object_storage/blobstor/fstree/range_bench_test.go

@@ -0,0 +1,94 @@
+package fstree_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/nspcc-dev/neofs-node/pkg/core/object"
+	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor/compression"
+	"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor/fstree"
+	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkFSTree_GetRange(b *testing.B) {
+	const (
+		KB = 1024
+		MB = 1024 * 1024
+	)
+
+	testCases := []struct {
+		from       uint64
+		length     uint64
+		objectSize int
+	}{
+		{from: 1 * MB, length: 4 * KB, objectSize: 10 * MB},
+		{from: 0, length: 10 * MB, objectSize: 10 * MB},
+		{from: 0, length: 0, objectSize: 10 * MB},
+		{from: 0, length: 4 * KB, objectSize: 4 * KB},
+		{from: 0, length: 0, objectSize: 4 * KB},
+		{from: 1 * KB, length: 1 * KB, objectSize: 4 * KB},
+		{from: 9 * MB, length: 4 * KB, objectSize: 10 * MB},
+	}
+
+	for _, tc := range testCases {
+		b.Run(fmt.Sprintf("size=%s,off=%s,len=%s",
+			generateSizeLabel(tc.objectSize), generateSizeLabel(int(tc.from)), generateSizeLabel(int(tc.length))),
+			func(b *testing.B) {
+				tmpDir := b.TempDir()
+
+				fsTree := fstree.New(fstree.WithPath(tmpDir))
+
+				obj := generateTestObject(tc.objectSize)
+				addr := object.AddressOf(obj)
+
+				b.Run("regular", func(b *testing.B) {
+					require.NoError(b, fsTree.Put(addr, obj.Marshal()))
+					b.ResetTimer()
+					for range b.N {
+						_, err := fsTree.GetRange(addr, tc.from, tc.length)
+						if err != nil {
+							b.Fatal(err)
+						}
+					}
+				})
+
+				b.Run("compressed", func(b *testing.B) {
+					compressConfig := &compression.Config{
+						Enabled: true,
+					}
+					require.NoError(b, compressConfig.Init())
+					require.NoError(b, fsTree.Put(addr, obj.Marshal()))
+
+					b.ResetTimer()
+					for range b.N {
+						_, err := fsTree.GetRange(addr, tc.from, tc.length)
+						if err != nil {
+							b.Fatal(err)
+						}
+					}
+				})
+
+				b.Run("combined", func(b *testing.B) {
+					const numObjects = 10
+
+					objMap := make(map[oid.Address][]byte, numObjects)
+					addrs := make([]oid.Address, numObjects)
+					for i := range numObjects {
+						o := generateTestObject(tc.objectSize)
+						objMap[object.AddressOf(o)] = o.Marshal()
+						addrs[i] = object.AddressOf(o)
+					}
+					require.NoError(b, fsTree.PutBatch(objMap))
+
+					b.ResetTimer()
+					for k := range b.N {
+						_, err := fsTree.GetRange(addrs[k%numObjects], tc.from, tc.length)
+						if err != nil {
+							b.Fatal(err)
+						}
+					}
+				})
+			})
+	}
+}