113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
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}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeNVarInt32(enc.valbuf, v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// SetUInt32Value encode uint32 value
|
|
func (enc *PrimitivePacketEncoder) SetUInt32Value(v uint32) {
|
|
size := encoding.SizeOfNVarUInt32(v)
|
|
codec := encoding.VarCodec{Size: size}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeNVarUInt32(enc.valbuf, v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// SetInt64Value encode int64 value
|
|
func (enc *PrimitivePacketEncoder) SetInt64Value(v int64) {
|
|
size := encoding.SizeOfNVarInt64(v)
|
|
codec := encoding.VarCodec{Size: size}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeNVarInt64(enc.valbuf, v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// SetUInt64Value encode uint64 value
|
|
func (enc *PrimitivePacketEncoder) SetUInt64Value(v uint64) {
|
|
size := encoding.SizeOfNVarUInt64(v)
|
|
codec := encoding.VarCodec{Size: size}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeNVarUInt64(enc.valbuf, v)
|
|
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}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeVarFloat32(enc.valbuf, v)
|
|
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}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodeVarFloat64(enc.valbuf, v)
|
|
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}
|
|
enc.valbuf = make([]byte, size)
|
|
err := codec.EncodePVarBool(enc.valbuf, v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// SetStringValue encode string
|
|
func (enc *PrimitivePacketEncoder) SetStringValue(v string) {
|
|
enc.valbuf = []byte(v)
|
|
}
|
|
|
|
// SetBytesValue encode []byte
|
|
func (enc *PrimitivePacketEncoder) SetBytesValue(v []byte) {
|
|
enc.valbuf = v
|
|
}
|