grpc
/
grpc-go
mirror of https://github.com/grpc/grpc-go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #385 from nuss-justin/master 

Avoid some allocations in recvMsg
This commit is contained in:
Qi Zhao 2015-10-06 22:39:18 -07:00
parent 02fd7b3ab4 e2cb78fabb
commit 6c8d213d08
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
rpc_util.go
View File

@ -139,10 +139,21 @@ type msgFixedHeader struct {
// EOF is returned with nil msg and 0 pf if the entire stream is done. Other
// non-nil error is returned if something is wrong on reading.
func (p *parser) recvMsg() (pf payloadFormat, msg []byte, err error) {
const (
headerSize = 5
formatIndex = 1
)
var hdr msgFixedHeader
if err := binary.Read(p.s, binary.BigEndian, &hdr); err != nil {
var buf [headerSize]byte
if _, err := io.ReadFull(p.s, buf[:]); err != nil {
return 0, nil, err
}
hdr.T = payloadFormat(buf[formatIndex])
hdr.Length = binary.BigEndian.Uint32(buf[formatIndex:])
if hdr.Length == 0 {
return hdr.T, nil, nil
}