Write entire compressed packet at once (#236)

This commit is contained in:
Richard Grover
2023-03-31 19:57:16 -06:00
committed by GitHub
parent 6bbfeec2c7
commit a8f9c4bee5
2 changed files with 52 additions and 88 deletions

View File

@ -252,21 +252,28 @@ func (l *Long) ReadFrom(r io.Reader) (n int64, err error) {
}
func (v VarInt) WriteTo(w io.Writer) (n int64, err error) {
vi := make([]byte, 0, MaxVarIntLen)
vi := make([]byte, MaxVarIntLen)
nn := v.WriteToBytes(vi)
_, err = w.Write(vi[:nn])
return int64(nn), err
}
func (v VarInt) WriteToBytes(buf []byte) int {
num := uint32(v)
i := 0
for {
b := num & 0x7F
num >>= 7
if num != 0 {
b |= 0x80
}
vi = append(vi, byte(b))
buf[i] = byte(b)
i++
if num == 0 {
break
}
}
nn, err := w.Write(vi)
return int64(nn), err
return i
}
func (v *VarInt) ReadFrom(r io.Reader) (n int64, err error) {
@ -308,21 +315,28 @@ func (v VarInt) Len() int {
}
func (v VarLong) WriteTo(w io.Writer) (n int64, err error) {
vi := make([]byte, 0, MaxVarLongLen)
vi := make([]byte, MaxVarLongLen)
nn := v.WriteToBytes(vi)
_, err = w.Write(vi[:nn])
return int64(nn), err
}
func (v VarLong) WriteToBytes(buf []byte) int {
num := uint64(v)
i := 0
for {
b := num & 0x7F
num >>= 7
if num != 0 {
b |= 0x80
}
vi = append(vi, byte(b))
buf[i] = byte(b)
i++
if num == 0 {
break
}
}
nn, err := w.Write(vi)
return int64(nn), err
return i
}
func (v *VarLong) ReadFrom(r io.Reader) (n int64, err error) {