From ea76c989ca933aa6904d74ba438b74a729b3d433 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Wed, 25 May 2022 15:56:45 +0300
Subject: [PATCH] [#1433] metabase: Add a benchmark for `Select`

Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
---
 .../metabase/select_test.go                   | 53 +++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/pkg/local_object_storage/metabase/select_test.go b/pkg/local_object_storage/metabase/select_test.go
index 2d7353666..0961fcf41 100644
--- a/pkg/local_object_storage/metabase/select_test.go
+++ b/pkg/local_object_storage/metabase/select_test.go
@@ -2,11 +2,13 @@ package meta_test
 
 import (
 	"encoding/hex"
+	"strconv"
 	"testing"
 
 	v2object "github.com/nspcc-dev/neofs-api-go/v2/object"
 	"github.com/nspcc-dev/neofs-node/pkg/core/object"
 	meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
+	cidSDK "github.com/nspcc-dev/neofs-sdk-go/container/id"
 	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"
@@ -727,3 +729,54 @@ func TestDB_SelectContainerID(t *testing.T) {
 		testSelect(t, db, cnr, fs)
 	})
 }
+
+func BenchmarkSelect(b *testing.B) {
+	const objCount = 1000
+	db := newDB(b)
+	cid := cidtest.ID()
+
+	for i := 0; i < objCount; i++ {
+		var attr objectSDK.Attribute
+		attr.SetKey("myHeader")
+		attr.SetValue(strconv.Itoa(i))
+		obj := generateObjectWithCID(b, cid)
+		obj.SetAttributes(attr)
+		require.NoError(b, meta.Put(db, obj, nil))
+	}
+
+	b.Run("string equal", func(b *testing.B) {
+		fs := objectSDK.SearchFilters{}
+		fs.AddFilter("myHeader", strconv.Itoa(objCount/2), objectSDK.MatchStringEqual)
+		benchmarkSelect(b, db, cid, fs, 1)
+	})
+	b.Run("string not equal", func(b *testing.B) {
+		fs := objectSDK.SearchFilters{}
+		fs.AddFilter("myHeader", strconv.Itoa(objCount/2), objectSDK.MatchStringNotEqual)
+		benchmarkSelect(b, db, cid, fs, objCount-1)
+	})
+	b.Run("common prefix", func(b *testing.B) {
+		prefix := "99"
+		n := 1 /* 99 */ + 10 /* 990..999 */
+
+		fs := objectSDK.SearchFilters{}
+		fs.AddFilter("myHeader", prefix, objectSDK.MatchCommonPrefix)
+		benchmarkSelect(b, db, cid, fs, n)
+	})
+	b.Run("unknown", func(b *testing.B) {
+		fs := objectSDK.SearchFilters{}
+		fs.AddFilter("myHeader", strconv.Itoa(objCount/2), objectSDK.MatchUnknown)
+		benchmarkSelect(b, db, cid, fs, 0)
+	})
+}
+
+func benchmarkSelect(b *testing.B, db *meta.DB, cid cidSDK.ID, fs objectSDK.SearchFilters, expected int) {
+	for i := 0; i < b.N; i++ {
+		res, err := meta.Select(db, &cid, fs)
+		if err != nil {
+			b.Fatal(err)
+		}
+		if len(res) != expected {
+			b.Fatalf("expected %d items, got %d", expected, len(res))
+		}
+	}
+}