[#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()
|
||||
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
|
||||
|
|
Loading…
Reference in a new issue