-
Notifications
You must be signed in to change notification settings - Fork 9
/
proto_decoders.go
60 lines (48 loc) · 1.12 KB
/
proto_decoders.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package isolate
import (
"encoding/json"
"io"
"github.com/ugorji/go/codec"
)
type jsonArgsDecoder struct{}
func (j jsonArgsDecoder) Unpack(in interface{}, out ...interface{}) error {
data, err := json.Marshal(in)
if err != nil {
return err
}
err = json.Unmarshal(data, &out)
return err
}
type msgpackArgsDecoder struct{}
var (
mPayloadHandler codec.MsgpackHandle
payloadHandler = &mPayloadHandler
mhAsocket = codec.MsgpackHandle{
BasicHandle: codec.BasicHandle{
EncodeOptions: codec.EncodeOptions{
StructToArray: true,
},
},
}
hAsocket = &mhAsocket
)
func newMsgpackDecoder(r io.Reader) Decoder {
return codec.NewDecoder(r, hAsocket)
}
func newMsgpackEncoder(w io.Writer) Encoder {
return codec.NewEncoder(w, hAsocket)
}
func (m msgpackArgsDecoder) Unpack(in interface{}, out ...interface{}) error {
var buf []byte
if err := codec.NewEncoderBytes(&buf, payloadHandler).Encode(in); err != nil {
return err
}
if err := codec.NewDecoderBytes(buf, payloadHandler).Decode(out); err != nil {
return err
}
return nil
}
var (
_ ArgsUnpacker = jsonArgsDecoder{}
_ ArgsUnpacker = msgpackArgsDecoder{}
)