fstree: add Get and Head bench tests

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

Author: End-rey

pkg/local_object_storage/blobstor/fstree/bench_test.go

@@ -0,0 +1,99 @@
+package fstree_test
+
+import (
+	"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"
+	objectSDK "github.com/nspcc-dev/neofs-sdk-go/object"
+	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkFSTree_Head(b *testing.B) {
+	for _, size := range payloadSizes {
+		b.Run(generateSizeLabel(size), func(b *testing.B) {
+			fsTree := fstree.New(fstree.WithPath(b.TempDir()))
+
+			require.NoError(b, fsTree.Open(false))
+			require.NoError(b, fsTree.Init())
+
+			testReadOp(b, fsTree, fsTree.Head, "Head", size)
+		})
+	}
+}
+
+func BenchmarkFSTree_Get(b *testing.B) {
+	for _, size := range payloadSizes {
+		b.Run(generateSizeLabel(size), func(b *testing.B) {
+			fsTree := fstree.New(fstree.WithPath(b.TempDir()))
+
+			require.NoError(b, fsTree.Open(false))
+			require.NoError(b, fsTree.Init())
+
+			testReadOp(b, fsTree, fsTree.Get, "Get", size)
+		})
+	}
+}
+
+func testReadOp(b *testing.B, fsTree *fstree.FSTree, read func(address oid.Address) (*objectSDK.Object, error),
+	name string, payloadSize int) {
+	b.Run(name+"_regular", func(b *testing.B) {
+		obj := generateTestObject(payloadSize)
+		addr := object.AddressOf(obj)
+
+		require.NoError(b, fsTree.Put(addr, obj.Marshal()))
+		b.ReportAllocs()
+		b.ResetTimer()
+		for range b.N {
+			_, err := read(addr)
+			if err != nil {
+				b.Fatal(err)
+			}
+		}
+	})
+
+	b.Run(name+"_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(payloadSize)
+			objMap[object.AddressOf(o)] = o.Marshal()
+			addrs[i] = object.AddressOf(o)
+		}
+		require.NoError(b, fsTree.PutBatch(objMap))
+
+		b.ReportAllocs()
+		b.ResetTimer()
+		for k := range b.N {
+			_, err := read(addrs[k%numObjects])
+			if err != nil {
+				b.Fatal(err)
+			}
+		}
+	})
+
+	b.Run(name+"_compressed", func(b *testing.B) {
+		obj := generateTestObject(payloadSize)
+		addr := object.AddressOf(obj)
+
+		compressConfig := &compression.Config{
+			Enabled: true,
+		}
+		require.NoError(b, compressConfig.Init())
+		fsTree.SetCompressor(compressConfig)
+		require.NoError(b, fsTree.Put(addr, obj.Marshal()))
+
+		b.ReportAllocs()
+		b.ResetTimer()
+		for range b.N {
+			_, err := read(addr)
+			if err != nil {
+				b.Fatal(err)
+			}
+		}
+	})
+}