From 1ac6f2eeebcdcc7e9d0159a3e65be20571731e1f Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Wed, 16 Sep 2020 14:16:58 +0300 Subject: [PATCH] [#145] sdk/object: Refactor logic of field setters In previous implementation RawObject constructor initialized internal v2 Object structure recursively so that further you can directly set the values of nested fields. This caused the object created by constructor New to be different from the object created from the empty object v2. Remove recursive initialization of v2 Object from New and NewRaw constructors. Make setters to initialize nested structures on demand. Signed-off-by: Leonard Lyubich --- pkg/object/object.go | 2 +- pkg/object/raw.go | 2 +- pkg/object/rw.go | 112 +++++++++++++++++++++++-------------------- 3 files changed, 61 insertions(+), 55 deletions(-) diff --git a/pkg/object/object.go b/pkg/object/object.go index c743414..e447be5 100644 --- a/pkg/object/object.go +++ b/pkg/object/object.go @@ -26,7 +26,7 @@ func NewFromV2(oV2 *object.Object) *Object { // // Works similar as NewFromV2(new(Object)). func New() *Object { - return NewFromV2(initObjectRecursive()) + return NewFromV2(new(object.Object)) } // ToV2 converts Object to v2 Object message. diff --git a/pkg/object/raw.go b/pkg/object/raw.go index 18edac3..0878f48 100644 --- a/pkg/object/raw.go +++ b/pkg/object/raw.go @@ -25,7 +25,7 @@ func NewRawFromV2(oV2 *object.Object) *RawObject { // // Works similar as NewRawFromV2(new(Object)). func NewRaw() *RawObject { - return NewRawFromV2(initObjectRecursive()) + return NewRawFromV2(new(object.Object)) } // Object returns read-only object instance. diff --git a/pkg/object/rw.go b/pkg/object/rw.go index a9c7647..5ed9f5e 100644 --- a/pkg/object/rw.go +++ b/pkg/object/rw.go @@ -12,18 +12,6 @@ import ( // public getter and private setters. type rwObject object.Object -func initObjectRecursive() *object.Object { - obj := new(object.Object) - - hdr := new(object.Header) - - hdr.SetSplit(new(object.SplitHeader)) - - obj.SetHeader(hdr) - - return obj -} - // TODO: add session token methods // ToV2 converts Object to v2 Object message. @@ -31,6 +19,30 @@ func (o *rwObject) ToV2() *object.Object { return (*object.Object)(o) } +func (o *rwObject) setHeaderField(setter func(*object.Header)) { + obj := (*object.Object)(o) + h := obj.GetHeader() + + if h == nil { + h = new(object.Header) + obj.SetHeader(h) + } + + setter(h) +} + +func (o *rwObject) setSplitFields(setter func(*object.SplitHeader)) { + o.setHeaderField(func(h *object.Header) { + split := h.GetSplit() + if split == nil { + split = new(object.SplitHeader) + h.SetSplit(split) + } + + setter(split) + }) +} + // GetID returns object identifier. func (o *rwObject) GetID() *ID { return NewIDFromV2( @@ -78,9 +90,9 @@ func (o *rwObject) GetVersion() *pkg.Version { } func (o *rwObject) setVersion(v *pkg.Version) { - (*object.Object)(o). - GetHeader(). - SetVersion(v.ToV2()) + o.setHeaderField(func(h *object.Header) { + h.SetVersion(v.ToV2()) + }) } // GetPayloadSize returns payload length of the object. @@ -91,9 +103,9 @@ func (o *rwObject) GetPayloadSize() uint64 { } func (o *rwObject) setPayloadSize(v uint64) { - (*object.Object)(o). - GetHeader(). - SetPayloadLength(v) + o.setHeaderField(func(h *object.Header) { + h.SetPayloadLength(v) + }) } // GetContainerID returns identifier of the related container. @@ -106,9 +118,9 @@ func (o *rwObject) GetContainerID() *container.ID { } func (o *rwObject) setContainerID(v *container.ID) { - (*object.Object)(o). - GetHeader(). - SetContainerID(v.ToV2()) + o.setHeaderField(func(h *object.Header) { + h.SetContainerID(v.ToV2()) + }) } // GetOwnerID returns identifier of the object owner. @@ -121,9 +133,9 @@ func (o *rwObject) GetOwnerID() *owner.ID { } func (o *rwObject) setOwnerID(v *owner.ID) { - (*object.Object)(o). - GetHeader(). - SetOwnerID(v.ToV2()) + o.setHeaderField(func(h *object.Header) { + h.SetOwnerID(v.ToV2()) + }) } // GetCreationEpoch returns epoch number in which object was created. @@ -134,9 +146,9 @@ func (o *rwObject) GetCreationEpoch() uint64 { } func (o *rwObject) setCreationEpoch(v uint64) { - (*object.Object)(o). - GetHeader(). - SetCreationEpoch(v) + o.setHeaderField(func(h *object.Header) { + h.SetCreationEpoch(v) + }) } // GetPayloadChecksum returns checksum of the object payload. @@ -149,9 +161,9 @@ func (o *rwObject) GetPayloadChecksum() *pkg.Checksum { } func (o *rwObject) setPayloadChecksum(v *pkg.Checksum) { - (*object.Object)(o). - GetHeader(). - SetPayloadHash(v.ToV2()) + o.setHeaderField(func(h *object.Header) { + h.SetPayloadHash(v.ToV2()) + }) } // GetPayloadHomomorphicHash returns homomorphic hash of the object payload. @@ -164,9 +176,9 @@ func (o *rwObject) GetPayloadHomomorphicHash() *pkg.Checksum { } func (o *rwObject) setPayloadHomomorphicHash(v *pkg.Checksum) { - (*object.Object)(o). - GetHeader(). - SetHomomorphicHash(v.ToV2()) + o.setHeaderField(func(h *object.Header) { + h.SetHomomorphicHash(v.ToV2()) + }) } // GetAttributes returns object attributes. @@ -185,16 +197,15 @@ func (o *rwObject) GetAttributes() []*Attribute { } func (o *rwObject) setAttributes(v ...*Attribute) { - h := (*object.Object)(o). - GetHeader() - attrs := make([]*object.Attribute, 0, len(v)) for i := range v { attrs = append(attrs, v[i].ToV2()) } - h.SetAttributes(attrs) + o.setHeaderField(func(h *object.Header) { + h.SetAttributes(attrs) + }) } // GetPreviousID returns identifier of the previous sibling object. @@ -208,10 +219,9 @@ func (o *rwObject) GetPreviousID() *ID { } func (o *rwObject) setPreviousID(v *ID) { - (*object.Object)(o). - GetHeader(). - GetSplit(). - SetPrevious(v.ToV2()) + o.setSplitFields(func(split *object.SplitHeader) { + split.SetPrevious(v.ToV2()) + }) } // GetChildren return list of the identifiers of the child objects. @@ -231,17 +241,15 @@ func (o *rwObject) GetChildren() []*ID { } func (o *rwObject) setChildren(v ...*ID) { - h := (*object.Object)(o). - GetHeader(). - GetSplit() - ids := make([]*refs.ObjectID, 0, len(v)) for i := range v { ids = append(ids, v[i].ToV2()) } - h.SetChildren(ids) + o.setSplitFields(func(split *object.SplitHeader) { + split.SetChildren(ids) + }) } // GetParent returns parent object w/o payload. @@ -259,11 +267,9 @@ func (o *rwObject) GetParent() *Object { } func (o *rwObject) setParent(v *Object) { - h := (*object.Object)(o). - GetHeader(). - GetSplit() - - h.SetParent((*object.Object)(v.rwObject).GetObjectID()) - h.SetParentSignature((*object.Object)(v.rwObject).GetSignature()) - h.SetParentHeader((*object.Object)(v.rwObject).GetHeader()) + o.setSplitFields(func(split *object.SplitHeader) { + split.SetParent((*object.Object)(v.rwObject).GetObjectID()) + split.SetParentSignature((*object.Object)(v.rwObject).GetSignature()) + split.SetParentHeader((*object.Object)(v.rwObject).GetHeader()) + }) }