100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package ormfield
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
"github.com/cosmos/cosmos-sdk/orm/types/ormerrors"
|
|
)
|
|
|
|
// BytesCodec encodes bytes as raw bytes. It errors if the byte array is longer
|
|
// than 255 bytes.
|
|
type BytesCodec struct{}
|
|
|
|
func (b BytesCodec) FixedBufferSize() int {
|
|
return -1
|
|
}
|
|
|
|
func (b BytesCodec) ComputeBufferSize(value protoreflect.Value) (int, error) {
|
|
return bytesSize(value)
|
|
}
|
|
|
|
func bytesSize(value protoreflect.Value) (int, error) {
|
|
bz := value.Bytes()
|
|
n := len(bz)
|
|
if n > 255 {
|
|
return -1, ormerrors.BytesFieldTooLong
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
func (b BytesCodec) IsOrdered() bool {
|
|
return false
|
|
}
|
|
|
|
func (b BytesCodec) Decode(r Reader) (protoreflect.Value, error) {
|
|
bz, err := io.ReadAll(r)
|
|
return protoreflect.ValueOfBytes(bz), err
|
|
}
|
|
|
|
func (b BytesCodec) Encode(value protoreflect.Value, w io.Writer) error {
|
|
_, err := w.Write(value.Bytes())
|
|
return err
|
|
}
|
|
|
|
func (b BytesCodec) Compare(v1, v2 protoreflect.Value) int {
|
|
return bytes.Compare(v1.Bytes(), v2.Bytes())
|
|
}
|
|
|
|
// NonTerminalBytesCodec encodes bytes as raw bytes length prefixed by a single
|
|
// byte. It errors if the byte array is longer than 255 bytes.
|
|
type NonTerminalBytesCodec struct{}
|
|
|
|
func (b NonTerminalBytesCodec) FixedBufferSize() int {
|
|
return -1
|
|
}
|
|
|
|
func (b NonTerminalBytesCodec) ComputeBufferSize(value protoreflect.Value) (int, error) {
|
|
n, err := bytesSize(value)
|
|
return n + 1, err
|
|
}
|
|
|
|
func (b NonTerminalBytesCodec) IsOrdered() bool {
|
|
return false
|
|
}
|
|
|
|
func (b NonTerminalBytesCodec) Compare(v1, v2 protoreflect.Value) int {
|
|
return bytes.Compare(v1.Bytes(), v2.Bytes())
|
|
}
|
|
|
|
func (b NonTerminalBytesCodec) Decode(r Reader) (protoreflect.Value, error) {
|
|
n, err := r.ReadByte()
|
|
if err != nil {
|
|
return protoreflect.Value{}, err
|
|
}
|
|
|
|
if n == 0 {
|
|
return protoreflect.ValueOfBytes([]byte{}), nil
|
|
}
|
|
|
|
bz := make([]byte, n)
|
|
_, err = r.Read(bz)
|
|
return protoreflect.ValueOfBytes(bz), err
|
|
}
|
|
|
|
func (b NonTerminalBytesCodec) Encode(value protoreflect.Value, w io.Writer) error {
|
|
bz := value.Bytes()
|
|
n := len(bz)
|
|
if n > 255 {
|
|
return ormerrors.BytesFieldTooLong
|
|
}
|
|
_, err := w.Write([]byte{byte(n)})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = w.Write(bz)
|
|
return err
|
|
}
|