frostfs-node/pkg/local_object_storage/blobstor/get_big.go
Leonard Lyubich fb6857a1cb [#211] blobstor: Rename Get operation to GetBig
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
2020-12-11 17:19:37 +03:00

83 lines
1.8 KiB
Go

package blobstor
import (
"io/ioutil"
"os"
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
"github.com/nspcc-dev/neofs-node/pkg/core/object"
"github.com/pkg/errors"
)
// GetBigPrm groups the parameters of GetBig operation.
type GetBigPrm struct {
addr *objectSDK.Address
}
// GetBigRes groups resulting values of GetBig operation.
type GetBigRes struct {
obj *object.Object
}
// ErrObjectNotFound is returns on read operations requested on a missing object.
var ErrObjectNotFound = errors.New("object not found")
// WithAddress is a GetBig option to set the address of the requested object.
//
// Option is required.
func (p *GetBigPrm) WithAddress(addr *objectSDK.Address) *GetBigPrm {
if p != nil {
p.addr = addr
}
return p
}
// Object returns the requested object.
func (r *GetBigRes) Object() *object.Object {
return r.obj
}
// GetBig reads the object from shallow dir of BLOB storage by address.
//
// Returns any error encountered that
// did not allow to completely read the object.
func (b *BlobStor) GetBig(prm *GetBigPrm) (*GetBigRes, error) {
b.mtx.RLock()
defer b.mtx.RUnlock()
// get compressed object data
data, err := b.fsTree.get(prm.addr)
if err != nil {
if errors.Is(err, errFileNotFound) {
return nil, ErrObjectNotFound
}
return nil, errors.Wrap(err, "could not read object from fs tree")
}
data, err = b.decompressor(data)
if err != nil {
return nil, errors.Wrap(err, "could not decompress object data")
}
// unmarshal the object
obj := object.New()
if err := obj.Unmarshal(data); err != nil {
return nil, errors.Wrap(err, "could not unmarshal the object")
}
return &GetBigRes{
obj: obj,
}, nil
}
func (t *fsTree) get(addr *objectSDK.Address) ([]byte, error) {
p := t.treePath(addr)
if _, err := os.Stat(p); os.IsNotExist(err) {
return nil, errFileNotFound
}
return ioutil.ReadFile(p)
}