forked from TrueCloudLab/frostfs-node
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package shard
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// PutPrm groups the parameters of Put operation.
|
|
type PutPrm struct {
|
|
obj *object.Object
|
|
}
|
|
|
|
// PutRes groups resulting values of Put operation.
|
|
type PutRes struct{}
|
|
|
|
// WithObject is a Put option to set object to save.
|
|
func (p *PutPrm) WithObject(obj *object.Object) *PutPrm {
|
|
if p != nil {
|
|
p.obj = obj
|
|
}
|
|
|
|
return p
|
|
}
|
|
|
|
// Put saves the object in shard.
|
|
//
|
|
// Returns any error encountered that
|
|
// did not allow to completely save the object.
|
|
func (s *Shard) Put(prm *PutPrm) (*PutRes, error) {
|
|
s.mtx.Lock()
|
|
defer s.mtx.Unlock()
|
|
|
|
// check object existence
|
|
ex, err := s.metaBase.Exists(prm.obj.Address())
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not check object existence")
|
|
} else if ex {
|
|
return nil, nil
|
|
}
|
|
|
|
// try to put to WriteCache
|
|
// TODO: implement
|
|
|
|
// put to BlobStor
|
|
if _, err := s.blobStor.Put(
|
|
new(blobstor.PutPrm).
|
|
WithObject(prm.obj),
|
|
); err != nil {
|
|
return nil, errors.Wrap(err, "could not put object to BLOB storage")
|
|
}
|
|
|
|
// put to metabase
|
|
if err := s.metaBase.Put(prm.obj); err != nil {
|
|
// may we need to handle this case in a special way
|
|
// since the object has been successfully written to BlobStor
|
|
return nil, errors.Wrap(err, "could not put object to metabase")
|
|
}
|
|
|
|
return nil, nil
|
|
}
|