/
unmarshal.go
43 lines (40 loc) · 1002 Bytes
/
unmarshal.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package app
import (
"gx/ipfs/QmZ4Qi3GaRbjcx28Sme5eMH7RQjGkt8wHxt2a65oLaeFEV/gogo-protobuf/proto"
"reflect"
"errors"
)
//Cache un-marshalling. This does cost a lot more memory
func (b *Decentralizer) getUnmarshalKey(buf []byte) uint32 {
b.crcTable.Reset()
b.crcTable.Write(buf)
return b.crcTable.Sum32()
}
func (d *Decentralizer) unmarshal(buf []byte, pb proto.Message) error {
//Experimental.
var cacheKey uint32
if d.unmarshalCache != nil {
cacheKey = d.getUnmarshalKey(buf)
if cacheVal, ok := d.unmarshalCache.Get(cacheKey); ok {
if val, ok := cacheVal.(proto.Message); ok {
v := reflect.ValueOf(pb).Elem()
v.Set(reflect.ValueOf(val).Elem())
}
if val, ok := cacheVal.(string); ok {
return errors.New(val)
}
return nil
}
}
err := proto.Unmarshal(buf, pb)
if err != nil {
if d.unmarshalCache != nil {
d.unmarshalCache.Add(cacheKey, err.Error())
}
return err
}
if d.unmarshalCache != nil {
d.unmarshalCache.Add(cacheKey, pb)
}
return err
}