perf: drop bytes.Buffer, use slice append

perf: drop bytes.Buffer, use slice append

net/packets/packet.go

@@ -1,7 +1,6 @@
 package packets
 
 import (
-	"bytes"
 	"encoding/binary"
 	"math"
 )
@@ -16,64 +15,58 @@ type IPacket interface {
 
 type Packet struct {
 	id  byte
-	buf *bytes.Buffer
+	buf []byte
 	raw []byte
 	pos int
 }
 
 func NewPacket(id byte) *Packet {
-	return &Packet{id: id, buf: &bytes.Buffer{}}
+	return &Packet{id: id}
 }
 
-func (p *Packet) GetID() byte        { return p.id }
+func (p *Packet) GetID() byte         { return p.id }
-func (p *Packet) GetBuffer() []byte  { return p.buf.Bytes() }
+func (p *Packet) GetBuffer() []byte   { return p.buf }
 func (p *Packet) SetBuffer(data []byte) { p.raw = data; p.pos = 0 }
 
-func (p *Packet) PutByte(v byte)    { p.buf.WriteByte(v) }
+func (p *Packet) PutByte(v byte)    { p.buf = append(p.buf, v) }
-func (p *Packet) PutBytes(v []byte) { p.buf.Write(v) }
+func (p *Packet) PutBytes(v []byte) { p.buf = append(p.buf, v...) }
 
 func (p *Packet) PutBool(v bool) {
 	if v {
-		p.buf.WriteByte(0x01)
+		p.buf = append(p.buf, 0x01)
 	} else {
-		p.buf.WriteByte(0x00)
+		p.buf = append(p.buf, 0x00)
 	}
 }
 
 func (p *Packet) PutShort(v int16) {
-	b := make([]byte, 2)
+	p.buf = append(p.buf, byte(v>>8), byte(v))
-	binary.BigEndian.PutUint16(b, uint16(v))
-	p.buf.Write(b)
 }
 
 func (p *Packet) PutLShort(v uint16) {
-	b := make([]byte, 2)
+	p.buf = append(p.buf, byte(v), byte(v>>8))
-	binary.LittleEndian.PutUint16(b, v)
-	p.buf.Write(b)
 }
 
 func (p *Packet) PutInt(v int32) {
-	b := make([]byte, 4)
+	p.buf = append(p.buf, byte(v>>24), byte(v>>16), byte(v>>8), byte(v))
-	binary.BigEndian.PutUint32(b, uint32(v))
-	p.buf.Write(b)
 }
 
 func (p *Packet) PutLInt(v int32) {
-	b := make([]byte, 4)
+	u := uint32(v)
-	binary.LittleEndian.PutUint32(b, uint32(v))
+	p.buf = append(p.buf, byte(u), byte(u>>8), byte(u>>16), byte(u>>24))
-	p.buf.Write(b)
 }
 
 func (p *Packet) PutLLong(v int64) {
-	b := make([]byte, 8)
+	u := uint64(v)
-	binary.LittleEndian.PutUint64(b, uint64(v))
+	p.buf = append(p.buf,
-	p.buf.Write(b)
+		byte(u), byte(u>>8), byte(u>>16), byte(u>>24),
+		byte(u>>32), byte(u>>40), byte(u>>48), byte(u>>56),
+	)
 }
 
 func (p *Packet) PutLFloat(v float32) {
-	b := make([]byte, 4)
+	u := math.Float32bits(v)
-	binary.LittleEndian.PutUint32(b, math.Float32bits(v))
+	p.buf = append(p.buf, byte(u), byte(u>>8), byte(u>>16), byte(u>>24))
-	p.buf.Write(b)
 }
 
 func (p *Packet) PutVarInt(v int32) {
@@ -82,10 +75,10 @@ func (p *Packet) PutVarInt(v int32) {
 
 func (p *Packet) PutUVarInt(v uint32) {
 	for v >= 0x80 {
-		p.buf.WriteByte(byte(v&0x7f) | 0x80)
+		p.buf = append(p.buf, byte(v&0x7f)|0x80)
 		v >>= 7
 	}
-	p.buf.WriteByte(byte(v))
+	p.buf = append(p.buf, byte(v))
 }
 
 func (p *Packet) PutVarLong(v int64) {
@@ -94,65 +87,79 @@ func (p *Packet) PutVarLong(v int64) {
 
 func (p *Packet) PutUVarLong(v uint64) {
 	for v >= 0x80 {
-		p.buf.WriteByte(byte(v&0x7f) | 0x80)
+		p.buf = append(p.buf, byte(v&0x7f)|0x80)
 		v >>= 7
 	}
-	p.buf.WriteByte(byte(v))
+	p.buf = append(p.buf, byte(v))
 }
 
 func (p *Packet) PutString(v string) {
 	p.PutUVarInt(uint32(len(v)))
-	p.buf.WriteString(v)
+	p.buf = append(p.buf, v...)
 }
 
 func (p *Packet) checkRead(n int) bool { return p.pos+n <= len(p.raw) }
 
 func (p *Packet) GetByte() byte {
-	if !p.checkRead(1) { return 0 }
+	if !p.checkRead(1) {
-	b := p.raw[p.pos]; p.pos++; return b
+		return 0
+	}
+	b := p.raw[p.pos]
+	p.pos++
+	return b
 }
 
 func (p *Packet) GetBytes(n int) []byte {
-	if !p.checkRead(n) { return nil }
+	if !p.checkRead(n) {
-	b := p.raw[p.pos : p.pos+n]; p.pos += n; return b
+		return nil
+	}
+	b := p.raw[p.pos : p.pos+n]
+	p.pos += n
+	return b
 }
 
-func (p *Packet) GetBool() bool { return p.GetByte() != 0x00 }
+func (p *Packet) GetBool() bool { return p.GetByte() != 0 }
 
 func (p *Packet) GetShort() int16 {
-	b := p.GetBytes(2)
+	if b := p.GetBytes(2); b != nil {
-	if b == nil { return 0 }
+		return int16(binary.BigEndian.Uint16(b))
-	return int16(binary.BigEndian.Uint16(b))
+	}
+	return 0
 }
 
 func (p *Packet) GetLShort() uint16 {
-	b := p.GetBytes(2)
+	if b := p.GetBytes(2); b != nil {
-	if b == nil { return 0 }
+		return binary.LittleEndian.Uint16(b)
-	return binary.LittleEndian.Uint16(b)
+	}
+	return 0
 }
 
 func (p *Packet) GetInt() int32 {
-	b := p.GetBytes(4)
+	if b := p.GetBytes(4); b != nil {
-	if b == nil { return 0 }
+		return int32(binary.BigEndian.Uint32(b))
-	return int32(binary.BigEndian.Uint32(b))
+	}
+	return 0
 }
 
 func (p *Packet) GetLInt() int32 {
-	b := p.GetBytes(4)
+	if b := p.GetBytes(4); b != nil {
-	if b == nil { return 0 }
+		return int32(binary.LittleEndian.Uint32(b))
-	return int32(binary.LittleEndian.Uint32(b))
+	}
+	return 0
 }
 
 func (p *Packet) GetLLong() int64 {
-	b := p.GetBytes(8)
+	if b := p.GetBytes(8); b != nil {
-	if b == nil { return 0 }
+		return int64(binary.LittleEndian.Uint64(b))
-	return int64(binary.LittleEndian.Uint64(b))
+	}
+	return 0
 }
 
 func (p *Packet) GetLFloat() float32 {
-	b := p.GetBytes(4)
+	if b := p.GetBytes(4); b != nil {
-	if b == nil { return 0 }
+		return math.Float32frombits(binary.LittleEndian.Uint32(b))
-	return math.Float32frombits(binary.LittleEndian.Uint32(b))
+	}
+	return 0
 }
 
 func (p *Packet) GetVarInt() int32 {
@@ -165,7 +172,9 @@ func (p *Packet) GetUVarInt() uint32 {
 	for i := uint(0); i < 35; i += 7 {
 		b := p.GetByte()
 		v |= uint32(b&0x7f) << i
-		if b&0x80 == 0 { break }
+		if b&0x80 == 0 {
+			break
+		}
 	}
 	return v
 }
@@ -180,16 +189,22 @@ func (p *Packet) GetUVarLong() uint64 {
 	for i := uint(0); i < 70; i += 7 {
 		b := p.GetByte()
 		v |= uint64(b&0x7f) << i
-		if b&0x80 == 0 { break }
+		if b&0x80 == 0 {
+			break
+		}
 	}
 	return v
 }
 
 func (p *Packet) GetString() string {
-	return string(p.GetBytes(int(p.GetUVarInt())))
+	n := p.GetUVarInt()
+	b := p.GetBytes(int(n))
+	return string(b)
 }
 
 func (p *Packet) Remaining() int {
-	if p.pos >= len(p.raw) { return 0 }
+	if p.pos >= len(p.raw) {
+		return 0
+	}
 	return len(p.raw) - p.pos
 }