[#2068] blobovnicza: Add Exists method

Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>
This commit is contained in:
Evgenii Stratonikov 2022-11-17 09:06:59 +03:00 committed by fyrchik
parent 9eccf3bbf5
commit 6ad2b5d5b8
2 changed files with 38 additions and 1 deletions

View file

@ -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
}

View file

@ -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