From 1005bf4f56d98cf09b285ee56ba1d447c508ffd8 Mon Sep 17 00:00:00 2001
From: Dmitrii Stepanov <d.stepanov@yadro.com>
Date: Tue, 5 Mar 2024 13:41:15 +0300
Subject: [PATCH] [#1024] shard: Add refill metabase benchmark

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
---
 pkg/local_object_storage/shard/refill_test.go | 72 +++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 pkg/local_object_storage/shard/refill_test.go

diff --git a/pkg/local_object_storage/shard/refill_test.go b/pkg/local_object_storage/shard/refill_test.go
new file mode 100644
index 000000000..07ec9180c
--- /dev/null
+++ b/pkg/local_object_storage/shard/refill_test.go
@@ -0,0 +1,72 @@
+package shard
+
+import (
+	"context"
+	"os"
+	"testing"
+
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
+	oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkRefillMetabase(b *testing.B) {
+	b.Run("100 objects", func(b *testing.B) {
+		benchRefillMetabase(b, 100)
+	})
+
+	b.Run("1000 objects", func(b *testing.B) {
+		benchRefillMetabase(b, 1000)
+	})
+
+	b.Run("2000 objects", func(b *testing.B) {
+		benchRefillMetabase(b, 2000)
+	})
+
+	b.Run("5000 objects", func(b *testing.B) {
+		benchRefillMetabase(b, 5000)
+	})
+}
+
+func benchRefillMetabase(b *testing.B, objectsCount int) {
+	sh := newShard(b, false)
+	defer func() { require.NoError(b, sh.Close()) }()
+
+	var putPrm PutPrm
+
+	for i := 0; i < objectsCount/2; i++ {
+		obj := testutil.GenerateObject()
+		testutil.AddAttribute(obj, "foo", "bar")
+		testutil.AddPayload(obj, 1<<5) // blobvnicza tree obj
+
+		putPrm.SetObject(obj)
+
+		_, err := sh.Put(context.Background(), putPrm)
+		require.NoError(b, err)
+	}
+
+	for i := 0; i < objectsCount/2; i++ {
+		obj := testutil.GenerateObject()
+		testutil.AddAttribute(obj, "foo", "bar")
+		obj.SetID(oidtest.ID())
+		testutil.AddPayload(obj, 1<<20) // fstree obj
+
+		putPrm.SetObject(obj)
+
+		_, err := sh.Put(context.Background(), putPrm)
+		require.NoError(b, err)
+	}
+
+	require.NoError(b, sh.Close())
+	require.NoError(b, os.Remove(sh.metaBase.DumpInfo().Path))
+
+	require.NoError(b, sh.Open(context.Background()))
+	sh.cfg.refillMetabase = true
+
+	b.ReportAllocs()
+	b.ResetTimer()
+
+	require.NoError(b, sh.Init(context.Background()))
+
+	require.NoError(b, sh.Close())
+}