GetValueAsUInt32()

Author: fanweixiao

pkg/spec/decode.go

@@ -45,6 +45,14 @@ func FromBytes(buf []byte) (p *Packet, err error) {
 	return p, nil
 }
 
+// GetValueAsUInt32 decode value as uint32
+func (p *Packet) GetValueAsUInt32() (uint32, error) {
+	var val uint32
+	codec := encoding.VarCodec{}
+	err := codec.DecodePVarUInt32(p.valbuf, &val)
+	return val, err
+}
+
 func readTag(buffer []byte, position int, val *uint64) (cursor int, err error) {
 	cursor, err = readPVarUInt64(buffer, position, val)
 	return cursor, err

pkg/spec/decode_test.go

@@ -73,3 +73,12 @@ func testReadLength(t *testing.T, buffer []byte, position int, expectValue uint6
 		t.Errorf("expect cursor=%d, actual=%d", expectedCursor, cursor)
 	}
 }
+
+func TestGetValueAsUInt32(t *testing.T) {
+	buf := []byte{0x03, 0x01, 0x02}
+	p, _ := FromBytes(buf)
+	actual, _ := p.GetValueAsUInt32()
+	if actual != 2 {
+		t.Errorf("expect=%d, actual=%d", 2, actual)
+	}
+}