265 lines
5.2 KiB
Go
265 lines
5.2 KiB
Go
|
package refs
|
||
|
|
||
|
import (
|
||
|
"encoding/binary"
|
||
|
|
||
|
refs "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs/grpc"
|
||
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/message"
|
||
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/util/proto"
|
||
|
"google.golang.org/protobuf/encoding/protowire"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
ownerIDValField = 1
|
||
|
|
||
|
containerIDValField = 1
|
||
|
|
||
|
objectIDValField = 1
|
||
|
|
||
|
addressContainerField = 1
|
||
|
addressObjectField = 2
|
||
|
|
||
|
checksumTypeField = 1
|
||
|
checksumValueField = 2
|
||
|
|
||
|
signatureKeyField = 1
|
||
|
signatureValueField = 2
|
||
|
signatureSchemeField = 3
|
||
|
|
||
|
versionMajorField = 1
|
||
|
versionMinorField = 2
|
||
|
)
|
||
|
|
||
|
func (o *OwnerID) StableMarshal(buf []byte) []byte {
|
||
|
if o == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, o.StableSize())
|
||
|
}
|
||
|
|
||
|
proto.BytesMarshal(ownerIDValField, buf, o.val)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (o *OwnerID) StableSize() int {
|
||
|
if o == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
return proto.BytesSize(ownerIDValField, o.val)
|
||
|
}
|
||
|
|
||
|
func (o *OwnerID) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(o, data, new(refs.OwnerID))
|
||
|
}
|
||
|
|
||
|
func (c *ContainerID) StableMarshal(buf []byte) []byte {
|
||
|
if c == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, c.StableSize())
|
||
|
}
|
||
|
|
||
|
proto.BytesMarshal(containerIDValField, buf, c.val)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (c *ContainerID) StableSize() int {
|
||
|
if c == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
return proto.BytesSize(containerIDValField, c.val)
|
||
|
}
|
||
|
|
||
|
func (c *ContainerID) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(c, data, new(refs.ContainerID))
|
||
|
}
|
||
|
|
||
|
func (o *ObjectID) StableMarshal(buf []byte) []byte {
|
||
|
if o == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, o.StableSize())
|
||
|
}
|
||
|
|
||
|
proto.BytesMarshal(objectIDValField, buf, o.val)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
// ObjectIDNestedListSize returns byte length of nested
|
||
|
// repeated ObjectID field with fNum number.
|
||
|
func ObjectIDNestedListSize(fNum int64, ids []ObjectID) (sz int) {
|
||
|
for i := range ids {
|
||
|
sz += proto.NestedStructureSize(fNum, &ids[i])
|
||
|
}
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func (o *ObjectID) StableSize() int {
|
||
|
if o == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
return proto.BytesSize(objectIDValField, o.val)
|
||
|
}
|
||
|
|
||
|
// ObjectIDNestedListMarshal writes protobuf repeated ObjectID field
|
||
|
// with fNum number to buf.
|
||
|
func ObjectIDNestedListMarshal(fNum int64, buf []byte, ids []ObjectID) (off int) {
|
||
|
prefix := protowire.EncodeTag(protowire.Number(fNum), protowire.BytesType)
|
||
|
for i := range ids {
|
||
|
off += binary.PutUvarint(buf[off:], prefix)
|
||
|
|
||
|
n := ids[i].StableSize()
|
||
|
off += binary.PutUvarint(buf[off:], uint64(n))
|
||
|
off += proto.BytesMarshal(objectIDValField, buf[off:], ids[i].val)
|
||
|
}
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
func (o *ObjectID) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(o, data, new(refs.ObjectID))
|
||
|
}
|
||
|
|
||
|
func (a *Address) StableMarshal(buf []byte) []byte {
|
||
|
if a == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, a.StableSize())
|
||
|
}
|
||
|
|
||
|
var offset int
|
||
|
|
||
|
offset += proto.NestedStructureMarshal(addressContainerField, buf[offset:], a.cid)
|
||
|
proto.NestedStructureMarshal(addressObjectField, buf[offset:], a.oid)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (a *Address) StableSize() (size int) {
|
||
|
if a == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
size += proto.NestedStructureSize(addressContainerField, a.cid)
|
||
|
size += proto.NestedStructureSize(addressObjectField, a.oid)
|
||
|
|
||
|
return size
|
||
|
}
|
||
|
|
||
|
func (a *Address) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(a, data, new(refs.Address))
|
||
|
}
|
||
|
|
||
|
func (c *Checksum) StableMarshal(buf []byte) []byte {
|
||
|
if c == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, c.StableSize())
|
||
|
}
|
||
|
|
||
|
var offset int
|
||
|
|
||
|
offset += proto.EnumMarshal(checksumTypeField, buf[offset:], int32(c.typ))
|
||
|
proto.BytesMarshal(checksumValueField, buf[offset:], c.sum)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (c *Checksum) StableSize() (size int) {
|
||
|
if c == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
size += proto.EnumSize(checksumTypeField, int32(c.typ))
|
||
|
size += proto.BytesSize(checksumValueField, c.sum)
|
||
|
|
||
|
return size
|
||
|
}
|
||
|
|
||
|
func (c *Checksum) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(c, data, new(refs.Checksum))
|
||
|
}
|
||
|
|
||
|
func (s *Signature) StableMarshal(buf []byte) []byte {
|
||
|
if s == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, s.StableSize())
|
||
|
}
|
||
|
|
||
|
var offset int
|
||
|
|
||
|
offset += proto.BytesMarshal(signatureKeyField, buf[offset:], s.key)
|
||
|
offset += proto.BytesMarshal(signatureValueField, buf[offset:], s.sign)
|
||
|
proto.EnumMarshal(signatureSchemeField, buf[offset:], int32(s.scheme))
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (s *Signature) StableSize() (size int) {
|
||
|
if s == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
size += proto.BytesSize(signatureKeyField, s.key)
|
||
|
size += proto.BytesSize(signatureValueField, s.sign)
|
||
|
size += proto.EnumSize(signatureSchemeField, int32(s.scheme))
|
||
|
|
||
|
return size
|
||
|
}
|
||
|
|
||
|
func (s *Signature) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(s, data, new(refs.Signature))
|
||
|
}
|
||
|
|
||
|
func (v *Version) StableMarshal(buf []byte) []byte {
|
||
|
if v == nil {
|
||
|
return []byte{}
|
||
|
}
|
||
|
|
||
|
if buf == nil {
|
||
|
buf = make([]byte, v.StableSize())
|
||
|
}
|
||
|
|
||
|
var offset int
|
||
|
|
||
|
offset += proto.UInt32Marshal(versionMajorField, buf[offset:], v.major)
|
||
|
proto.UInt32Marshal(versionMinorField, buf[offset:], v.minor)
|
||
|
|
||
|
return buf
|
||
|
}
|
||
|
|
||
|
func (v *Version) StableSize() (size int) {
|
||
|
if v == nil {
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
size += proto.UInt32Size(versionMajorField, v.major)
|
||
|
size += proto.UInt32Size(versionMinorField, v.minor)
|
||
|
|
||
|
return size
|
||
|
}
|
||
|
|
||
|
func (v *Version) Unmarshal(data []byte) error {
|
||
|
return message.Unmarshal(v, data, new(refs.Version))
|
||
|
}
|