From 94ec34e453b3497dbdf3212f3c35331b8225fda2 Mon Sep 17 00:00:00 2001
From: Pavel Karpy <carpawell@nspcc.ru>
Date: Tue, 31 May 2022 14:42:41 +0300
Subject: [PATCH] [#1460] meta: Add a benchmark on `Get` operation

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
---
 pkg/local_object_storage/metabase/get_test.go | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/pkg/local_object_storage/metabase/get_test.go b/pkg/local_object_storage/metabase/get_test.go
index e9f7eca490..95c345bcd0 100644
--- a/pkg/local_object_storage/metabase/get_test.go
+++ b/pkg/local_object_storage/metabase/get_test.go
@@ -2,6 +2,8 @@ package meta_test
 
 import (
 	"bytes"
+	"fmt"
+	"os"
 	"testing"
 
 	"github.com/nspcc-dev/neofs-node/pkg/core/object"
@@ -9,6 +11,7 @@ import (
 	apistatus "github.com/nspcc-dev/neofs-sdk-go/client/status"
 	cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test"
 	objectSDK "github.com/nspcc-dev/neofs-sdk-go/object"
+	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
 	oidtest "github.com/nspcc-dev/neofs-sdk-go/object/id/test"
 	"github.com/stretchr/testify/require"
 )
@@ -149,3 +152,54 @@ func binaryEqual(a, b *objectSDK.Object) bool {
 
 	return bytes.Equal(binaryA, binaryB)
 }
+
+func BenchmarkGet(b *testing.B) {
+	numOfObjects := [...]int{
+		1,
+		10,
+		100,
+	}
+
+	for _, num := range numOfObjects {
+		b.Run(fmt.Sprintf("%d objects", num), func(b *testing.B) {
+			benchmarkGet(b, num)
+		})
+	}
+}
+
+var obj *objectSDK.Object
+
+func benchmarkGet(b *testing.B, numOfObj int) {
+	db := newDB(b)
+	addrs := make([]oid.Address, 0, numOfObj)
+
+	for i := 0; i < numOfObj; i++ {
+		raw := generateObject(b)
+		addrs = append(addrs, object.AddressOf(raw))
+
+		err := putBig(db, raw)
+		require.NoError(b, err)
+	}
+
+	b.Cleanup(func() {
+		_ = db.Close()
+		_ = os.RemoveAll(b.Name())
+	})
+	b.ReportAllocs()
+	b.ResetTimer()
+
+	var getPrm meta.GetPrm
+	getPrm.WithAddress(addrs[len(addrs)/2])
+
+	for i := 0; i < b.N; i++ {
+		for _, addr := range addrs {
+			getPrm.WithAddress(addr)
+
+			res, err := db.Get(getPrm)
+			require.NoError(b, err)
+
+			obj = res.Header()
+		}
+
+	}
+}