From 2a5a79660af339ed6871cee4599e48db76a25587 Mon Sep 17 00:00:00 2001 From: Lin <767763591@qq.com> Date: Sat, 9 Jan 2021 20:47:24 +0800 Subject: [PATCH] NetworkReader Pref improve --- binary/reader.go | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/binary/reader.go b/binary/reader.go index 6a616930..e4a55759 100644 --- a/binary/reader.go +++ b/binary/reader.go @@ -2,6 +2,7 @@ package binary import ( "bytes" + "io" "net" ) @@ -137,14 +138,15 @@ func (r *NetworkReader) ReadByte() (byte, error) { func (r *NetworkReader) ReadBytes(len int) ([]byte, error) { buf := make([]byte, len) - for i := 0; i < len; i++ { - b, err := r.ReadByte() - if err != nil { - return nil, err - } - buf[i] = b - } - return buf, nil + _, err := io.ReadFull(r.conn, buf) + //for i := 0; i < len; i++ { + // b, err := r.ReadByte() + // if err != nil { + // return nil, err + // } + // buf[i] = b + //} + return buf, err } func (r *NetworkReader) ReadInt32() (int32, error) {