opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change the type of length

Browse Source
This commit is contained in:
iamqizhao
2015-10-15 13:22:54 -07:00
parent 61c1259226
commit 984280ba37
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rpc_util.go
View File

@ -164,7 +164,7 @@ func encode(c Codec, msg interface{}, pf payloadFormat) ([]byte, error) {
// Write message into the fixed header. // Write message into the fixed header.
buf.WriteByte(uint8(pf)) buf.WriteByte(uint8(pf))
var b []byte var b []byte
var length uint32 var length uint
if msg != nil { if msg != nil {
var err error var err error
// TODO(zhaoq): optimize to reduce memory alloc and copying. // TODO(zhaoq): optimize to reduce memory alloc and copying.
@ -172,13 +172,13 @@ func encode(c Codec, msg interface{}, pf payloadFormat) ([]byte, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
length = uint32(len(b)) length = uint(len(b))
} }
if length > math.MaxUint32 { if length > math.MaxUint32 {
return nil, Errorf(codes.InvalidArgument, "grpc: message too large (%d bytes)", length) return nil, Errorf(codes.InvalidArgument, "grpc: message too large (%d bytes)", length)
} }
var szHdr [4]byte var szHdr [4]byte
binary.BigEndian.PutUint32(szHdr[:], length) binary.BigEndian.PutUint32(szHdr[:], uint32(length))
buf.Write(szHdr[:]) buf.Write(szHdr[:])
buf.Write(b) buf.Write(b)
return buf.Bytes(), nil return buf.Bytes(), nil