Add stable marshal for refs package
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin
Stanislav Bogatyrev
parent
f8e6908e50
commit
2c33e22db5
2 changed files with 246 additions and 0 deletions
157
v2/refs/marshal.go
Normal file
157
v2/refs/marshal.go
Normal file
|
|
@ -0,0 +1,157 @@
|
|||
package refs
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
|
||||
"github.com/nspcc-dev/neofs-api-go/util/proto"
|
||||
)
|
||||
|
||||
const (
|
||||
OwnerIDValField = 1
|
||||
|
||||
ContainerIDValField = 1
|
||||
|
||||
ObjectIDValField = 1
|
||||
|
||||
AddressContainerField = 1
|
||||
AddressObjectField = 2
|
||||
)
|
||||
|
||||
func (o *OwnerID) StableMarshal(buf []byte) ([]byte, error) {
|
||||
if o == nil {
|
||||
return []byte{}, nil
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
buf = make([]byte, o.StableSize())
|
||||
}
|
||||
|
||||
_, err := proto.BytesMarshal(OwnerIDValField, buf, o.val)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (o *OwnerID) StableSize() int {
|
||||
if o == nil {
|
||||
return 0
|
||||
}
|
||||
return proto.BytesSize(OwnerIDValField, o.val)
|
||||
}
|
||||
|
||||
func (c *ContainerID) StableMarshal(buf []byte) ([]byte, error) {
|
||||
if c == nil {
|
||||
return []byte{}, nil
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
buf = make([]byte, c.StableSize())
|
||||
}
|
||||
|
||||
_, err := proto.BytesMarshal(ContainerIDValField, buf, c.val)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (c *ContainerID) StableSize() int {
|
||||
if c == nil {
|
||||
return 0
|
||||
}
|
||||
return proto.BytesSize(ContainerIDValField, c.val)
|
||||
}
|
||||
|
||||
func (o *ObjectID) StableMarshal(buf []byte) ([]byte, error) {
|
||||
if o == nil {
|
||||
return []byte{}, nil
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
buf = make([]byte, o.StableSize())
|
||||
}
|
||||
|
||||
_, err := proto.BytesMarshal(ObjectIDValField, buf, o.val)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (o *ObjectID) StableSize() int {
|
||||
if o == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
return proto.BytesSize(ObjectIDValField, o.val)
|
||||
}
|
||||
|
||||
func (a *Address) StableMarshal(buf []byte) ([]byte, error) {
|
||||
if a == nil {
|
||||
return []byte{}, nil
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
buf = make([]byte, a.StableSize())
|
||||
}
|
||||
|
||||
var (
|
||||
offset, n int
|
||||
prefix uint64
|
||||
err error
|
||||
)
|
||||
|
||||
if a.cid != nil {
|
||||
prefix, _ = proto.NestedStructurePrefix(AddressContainerField)
|
||||
offset = binary.PutUvarint(buf, prefix)
|
||||
|
||||
n = a.cid.StableSize()
|
||||
offset += binary.PutUvarint(buf[offset:], uint64(n))
|
||||
|
||||
_, err = a.cid.StableMarshal(buf[offset:])
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
offset += n
|
||||
}
|
||||
|
||||
if a.oid != nil {
|
||||
prefix, _ = proto.NestedStructurePrefix(AddressObjectField)
|
||||
offset += binary.PutUvarint(buf[offset:], prefix)
|
||||
|
||||
n = a.oid.StableSize()
|
||||
offset += binary.PutUvarint(buf[offset:], uint64(n))
|
||||
|
||||
_, err = a.oid.StableMarshal(buf[offset:])
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (a *Address) StableSize() (size int) {
|
||||
if a == nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
if a.cid != nil {
|
||||
_, ln := proto.NestedStructurePrefix(AddressContainerField)
|
||||
n := a.cid.StableSize()
|
||||
size += ln + proto.VarUIntSize(uint64(n)) + n
|
||||
}
|
||||
|
||||
if a.oid != nil {
|
||||
_, ln := proto.NestedStructurePrefix(AddressObjectField)
|
||||
n := a.oid.StableSize()
|
||||
size += ln + proto.VarUIntSize(uint64(n)) + n
|
||||
}
|
||||
|
||||
return size
|
||||
}
|
||||
89
v2/refs/marshal_test.go
Normal file
89
v2/refs/marshal_test.go
Normal file
|
|
@ -0,0 +1,89 @@
|
|||
package refs_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
||||
grpc "github.com/nspcc-dev/neofs-api-go/v2/refs/grpc"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestOwnerID_StableMarshal(t *testing.T) {
|
||||
ownerFrom := new(refs.OwnerID)
|
||||
ownerTransport := new(grpc.OwnerID)
|
||||
|
||||
t.Run("non empty", func(t *testing.T) {
|
||||
ownerFrom.SetValue([]byte("Owner ID"))
|
||||
|
||||
wire, err := ownerFrom.StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = ownerTransport.Unmarshal(wire)
|
||||
require.NoError(t, err)
|
||||
|
||||
ownerTo := refs.OwnerIDFromGRPCMessage(ownerTransport)
|
||||
require.Equal(t, ownerFrom, ownerTo)
|
||||
})
|
||||
}
|
||||
|
||||
func TestContainerID_StableMarshal(t *testing.T) {
|
||||
cnrFrom := new(refs.ContainerID)
|
||||
cnrTransport := new(grpc.ContainerID)
|
||||
|
||||
t.Run("non empty", func(t *testing.T) {
|
||||
cnrFrom.SetValue([]byte("Container ID"))
|
||||
|
||||
wire, err := cnrFrom.StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = cnrTransport.Unmarshal(wire)
|
||||
require.NoError(t, err)
|
||||
|
||||
cnrTo := refs.ContainerIDFromGRPCMessage(cnrTransport)
|
||||
require.Equal(t, cnrFrom, cnrTo)
|
||||
})
|
||||
}
|
||||
|
||||
func TestObjectID_StableMarshal(t *testing.T) {
|
||||
objectIDFrom := new(refs.ObjectID)
|
||||
objectIDTransport := new(grpc.ObjectID)
|
||||
|
||||
t.Run("non empty", func(t *testing.T) {
|
||||
objectIDFrom.SetValue([]byte("Object ID"))
|
||||
|
||||
wire, err := objectIDFrom.StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = objectIDTransport.Unmarshal(wire)
|
||||
require.NoError(t, err)
|
||||
|
||||
objectIDTo := refs.ObjectIDFromGRPCMessage(objectIDTransport)
|
||||
require.Equal(t, objectIDFrom, objectIDTo)
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddress_StableMarshal(t *testing.T) {
|
||||
addressFrom := new(refs.Address)
|
||||
|
||||
cnr := new(refs.ContainerID)
|
||||
cnr.SetValue([]byte("Container ID"))
|
||||
|
||||
objectID := new(refs.ObjectID)
|
||||
objectID.SetValue([]byte("Object ID"))
|
||||
|
||||
addressTransport := new(grpc.Address)
|
||||
|
||||
t.Run("non empty", func(t *testing.T) {
|
||||
addressFrom.SetContainerID(cnr)
|
||||
addressFrom.SetObjectID(objectID)
|
||||
|
||||
wire, err := addressFrom.StableMarshal(nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = addressTransport.Unmarshal(wire)
|
||||
require.NoError(t, err)
|
||||
|
||||
addressTo := refs.AddressFromGRPCMessage(addressTransport)
|
||||
require.Equal(t, addressFrom, addressTo)
|
||||
})
|
||||
}
|
||||
Loading…
Reference in a new issue