[#2068] blobovnicza: Add Exists
method
Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>
This commit is contained in:
parent
9eccf3bbf5
commit
6ad2b5d5b8
2 changed files with 38 additions and 1 deletions
29
pkg/local_object_storage/blobovnicza/exists.go
Normal file
29
pkg/local_object_storage/blobovnicza/exists.go
Normal 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
|
||||||
|
}
|
|
@ -40,12 +40,16 @@ func TestBlobovnicza_Get(t *testing.T) {
|
||||||
addr := oidtest.Address()
|
addr := oidtest.Address()
|
||||||
obj := make([]byte, firstBucketBound+1)
|
obj := make([]byte, firstBucketBound+1)
|
||||||
|
|
||||||
|
exists, err := blz.Exists(addr)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.False(t, exists)
|
||||||
|
|
||||||
var prmPut PutPrm
|
var prmPut PutPrm
|
||||||
prmPut.SetAddress(addr)
|
prmPut.SetAddress(addr)
|
||||||
prmPut.SetMarshaledObject(obj)
|
prmPut.SetMarshaledObject(obj)
|
||||||
|
|
||||||
// place object to [32K:64K] bucket
|
// place object to [32K:64K] bucket
|
||||||
_, err := blz.Put(prmPut)
|
_, err = blz.Put(prmPut)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
var prmGet GetPrm
|
var prmGet GetPrm
|
||||||
|
@ -55,6 +59,10 @@ func TestBlobovnicza_Get(t *testing.T) {
|
||||||
res, err := blz.Get(prmGet)
|
res, err := blz.Get(prmGet)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
require.Equal(t, obj, res.Object())
|
require.Equal(t, obj, res.Object())
|
||||||
|
|
||||||
|
exists, err := blz.Exists(addr)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.True(t, exists)
|
||||||
}
|
}
|
||||||
|
|
||||||
// object should be available
|
// object should be available
|
||||||
|
|
Loading…
Reference in a new issue