mq_coder/primitive_encoder.go

113 lines
2.7 KiB
Go
Raw Normal View History

2022-10-10 14:47:49 +08:00
package mq_coder
import (
"bytes"
"git.hpds.cc/Component/mq_coder/encoding"
)
// PrimitivePacketEncoder used for encode a primitive packet
type PrimitivePacketEncoder struct {
*encoder
}
// NewPrimitivePacketEncoder return an Encoder for primitive packet
func NewPrimitivePacketEncoder(sid byte) *PrimitivePacketEncoder {
prim := &PrimitivePacketEncoder{
encoder: &encoder{
isNode: false,
buf: new(bytes.Buffer),
},
}
prim.seqID = sid
return prim
}
// SetInt32Value encode int32 value
func (enc *PrimitivePacketEncoder) SetInt32Value(v int32) {
size := encoding.SizeOfNVarInt32(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeNVarInt32(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetUInt32Value encode uint32 value
func (enc *PrimitivePacketEncoder) SetUInt32Value(v uint32) {
size := encoding.SizeOfNVarUInt32(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeNVarUInt32(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetInt64Value encode int64 value
func (enc *PrimitivePacketEncoder) SetInt64Value(v int64) {
size := encoding.SizeOfNVarInt64(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeNVarInt64(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetUInt64Value encode uint64 value
func (enc *PrimitivePacketEncoder) SetUInt64Value(v uint64) {
size := encoding.SizeOfNVarUInt64(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeNVarUInt64(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetFloat32Value encode float32 value
func (enc *PrimitivePacketEncoder) SetFloat32Value(v float32) {
var size = encoding.SizeOfVarFloat32(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeVarFloat32(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetFloat64Value encode float64 value
func (enc *PrimitivePacketEncoder) SetFloat64Value(v float64) {
var size = encoding.SizeOfVarFloat64(v)
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodeVarFloat64(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetBoolValue encode bool value
func (enc *PrimitivePacketEncoder) SetBoolValue(v bool) {
var size = encoding.SizeOfPVarUInt32(uint32(1))
codec := encoding.VarCodec{Size: size}
2023-03-10 23:58:36 +08:00
enc.valBuf = make([]byte, size)
err := codec.EncodePVarBool(enc.valBuf, v)
2022-10-10 14:47:49 +08:00
if err != nil {
panic(err)
}
}
// SetStringValue encode string
func (enc *PrimitivePacketEncoder) SetStringValue(v string) {
2023-03-10 23:58:36 +08:00
enc.valBuf = []byte(v)
2022-10-10 14:47:49 +08:00
}
// SetBytesValue encode []byte
func (enc *PrimitivePacketEncoder) SetBytesValue(v []byte) {
2023-03-10 23:58:36 +08:00
enc.valBuf = v
2022-10-10 14:47:49 +08:00
}