PutBool & PutBytes & PutUTF8String

Author: fanweixiao

v2.go

@@ -26,11 +26,6 @@ func NewPacket(sid int) (*Packet, error) {
 	return p, nil
 }
 
-// // Length return the length of value
-// func (p *Packet) Length() int {
-// 	return p.length
-// }
-
 func (p *Packet) setLengthBuffer() error {
 	len := uint64(p.length)
 	size := encoding.SizeOfPVarUInt64(len)
@@ -133,6 +128,27 @@ func (p *Packet) PutFloat64(v float64) error {
 	return nil
 }
 
+// PutBool set boolean value
+func (p *Packet) PutBool(v bool) error {
+	var val uint64 = 0
+	if v {
+		val = 1
+	}
+	return p.PutUInt64(val)
+}
+
+// PutUTF8String set string value
+func (p *Packet) PutUTF8String(v string) {
+	p.valbuf = []byte(v)
+	p.length = len(p.valbuf)
+}
+
+// PutBytes set bytes value
+func (p *Packet) PutBytes(v []byte) {
+	p.valbuf = v
+	p.length = len(p.valbuf)
+}
+
 // AddNode add a child Packet
 func (p *Packet) AddNode(child *Packet) (*Packet, error) {
 	childBuffer, err := child.Encode()

v2_test.go

@@ -58,6 +58,32 @@ func TestV2EncodeFloat64(t *testing.T) {
 	testV2EncodeFloat64(t, 0x09, 0.01171875, []byte{0x09, 0x02, 0x3F, 0x88})
 }
 
+func TestV2EncodeString(t *testing.T) {
+	p, _ := NewPacket(0x01)
+	p.PutUTF8String("C")
+	compareBytes(t, p, []byte{0x01, 0x01, 0x43})
+	p.PutUTF8String("CC")
+	compareBytes(t, p, []byte{0x01, 0x02, 0x43, 0x43})
+	p.PutUTF8String("Yona")
+	compareBytes(t, p, []byte{0x01, 0x04, 0x59, 0x6F, 0x6E, 0x61})
+	p.PutUTF8String("https://yomo.run")
+	compareBytes(t, p, []byte{0x01, 0x10, 0x68, 0x74, 0x74, 0x70, 0x73, 0x3A, 0x2F, 0x2F, 0x79, 0x6F, 0x6D, 0x6F, 0x2E, 0x72, 0x75, 0x6E})
+}
+
+func TestV2EncodeBytes(t *testing.T) {
+	p, _ := NewPacket(0x01)
+	p.PutBytes([]byte{0x03, 0x06, 0x09, 0x0C, 0x0F})
+	compareBytes(t, p, []byte{0x01, 0x05, 0x03, 0x06, 0x09, 0x0C, 0x0F})
+}
+
+func TestV2EncodeBool(t *testing.T) {
+	p, _ := NewPacket(0x01)
+	p.PutBool(true)
+	compareBytes(t, p, []byte{0x01, 0x01, 0x01})
+	p.PutBool(false)
+	compareBytes(t, p, []byte{0x01, 0x01, 0x00})
+}
+
 func TestV2AddNode(t *testing.T) {
 	parent, _ := NewPacket(0x01)