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

Reset proto before unmarshalling ()

Browse Source
This commit is contained in:
Yongzheng Lai
2017-05-06 04:23:47 +08:00
committed by dfawley
parent 844f573616
commit 07f4e6bf4e
1 changed files with 1 additions and 0 deletions

1
codec.go

@ -96,6 +96,7 @@ func (p protoCodec) Marshal(v interface{}) ([]byte, error) {
func (p protoCodec) Unmarshal(data []byte, v interface{}) error { func (p protoCodec) Unmarshal(data []byte, v interface{}) error {
cb := protoBufferPool.Get().(*cachedProtoBuffer) cb := protoBufferPool.Get().(*cachedProtoBuffer)
cb.SetBuf(data) cb.SetBuf(data)
v.(proto.Message).Reset()
err := cb.Unmarshal(v.(proto.Message)) err := cb.Unmarshal(v.(proto.Message))
cb.SetBuf(nil) cb.SetBuf(nil)
protoBufferPool.Put(cb) protoBufferPool.Put(cb)