From 6ad2b5d5b87519a532b0dfa9f90c50c3214d288c Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@morphbits.ru>
Date: Thu, 17 Nov 2022 09:06:59 +0300
Subject: [PATCH] [#2068] blobovnicza: Add `Exists` method

Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>
---
 .../blobovnicza/exists.go                     | 29 +++++++++++++++++++
 .../blobovnicza/get_test.go                   | 10 ++++++-
 2 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 pkg/local_object_storage/blobovnicza/exists.go

diff --git a/pkg/local_object_storage/blobovnicza/exists.go b/pkg/local_object_storage/blobovnicza/exists.go
new file mode 100644
index 00000000..47d57d6f
--- /dev/null
+++ b/pkg/local_object_storage/blobovnicza/exists.go
@@ -0,0 +1,29 @@
+package blobovnicza
+
+import (
+	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
+	"go.etcd.io/bbolt"
+)
+
+// Exists check if object with the specified address is stored in b.
+func (b *Blobovnicza) Exists(addr oid.Address) (bool, error) {
+	var (
+		exists  bool
+		addrKey = addressKey(addr)
+	)
+
+	err := b.boltDB.View(func(tx *bbolt.Tx) error {
+		return tx.ForEach(func(_ []byte, buck *bbolt.Bucket) error {
+			exists = buck.Get(addrKey) != nil
+			if exists {
+				return errInterruptForEach
+			}
+			return nil
+		})
+	})
+
+	if err == errInterruptForEach {
+		err = nil
+	}
+	return exists, err
+}
diff --git a/pkg/local_object_storage/blobovnicza/get_test.go b/pkg/local_object_storage/blobovnicza/get_test.go
index 91b56fc5..147f0066 100644
--- a/pkg/local_object_storage/blobovnicza/get_test.go
+++ b/pkg/local_object_storage/blobovnicza/get_test.go
@@ -40,12 +40,16 @@ func TestBlobovnicza_Get(t *testing.T) {
 		addr := oidtest.Address()
 		obj := make([]byte, firstBucketBound+1)
 
+		exists, err := blz.Exists(addr)
+		require.NoError(t, err)
+		require.False(t, exists)
+
 		var prmPut PutPrm
 		prmPut.SetAddress(addr)
 		prmPut.SetMarshaledObject(obj)
 
 		// place object to [32K:64K] bucket
-		_, err := blz.Put(prmPut)
+		_, err = blz.Put(prmPut)
 		require.NoError(t, err)
 
 		var prmGet GetPrm
@@ -55,6 +59,10 @@ func TestBlobovnicza_Get(t *testing.T) {
 			res, err := blz.Get(prmGet)
 			require.NoError(t, err)
 			require.Equal(t, obj, res.Object())
+
+			exists, err := blz.Exists(addr)
+			require.NoError(t, err)
+			require.True(t, exists)
 		}
 
 		// object should be available