diff --git a/tidb-server/server/packetio.go b/tidb-server/server/packetio.go index 150cc7c4d299f..1a2ef79a3f09c 100644 --- a/tidb-server/server/packetio.go +++ b/tidb-server/server/packetio.go @@ -43,6 +43,11 @@ import ( "github.com/pingcap/tidb/mysql" ) +const ( + defaultReaderSize = 16 * 1024 + defaultWriterSize = 16 * 1024 +) + type packetIO struct { rb *bufio.Reader wb *bufio.Writer @@ -52,17 +57,17 @@ type packetIO struct { func newPacketIO(conn net.Conn) *packetIO { p := &packetIO{ - rb: bufio.NewReaderSize(conn, 2048), - wb: bufio.NewWriterSize(conn, 2048), + rb: bufio.NewReaderSize(conn, defaultReaderSize), + wb: bufio.NewWriterSize(conn, defaultWriterSize), } return p } func (p *packetIO) readPacket() ([]byte, error) { - header := []byte{0, 0, 0, 0} + var header [4]byte - if _, err := io.ReadFull(p.rb, header); err != nil { + if _, err := io.ReadFull(p.rb, header[:]); err != nil { return nil, errors.Trace(err) }