/
helper.go
38 lines (33 loc) · 877 Bytes
/
helper.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
package api
import (
"github.com/golang/protobuf/proto"
"github.com/EndlessCheng/mahjong-helper/platform/majsoul/proto/lq"
)
func WrapMessage(name string, message proto.Message) (data []byte, err error) {
data, err = proto.Marshal(message)
if err != nil {
return
}
wrap := lq.Wrapper{
Name: name,
Data: data,
}
return proto.Marshal(&wrap)
}
func UnwrapData(rawData []byte) (methodName string, data []byte, err error) {
wrapper := lq.Wrapper{}
if err = proto.Unmarshal(rawData, &wrapper); err != nil {
return
}
return wrapper.GetName(), wrapper.GetData(), nil
}
// TODO: auto UnwrapMessage by methodName
func UnwrapMessage(rawData []byte, message proto.Message) error {
methodName, data, err := UnwrapData(rawData)
if err != nil {
return err
}
// TODO: assert methodName when its not empty
_ = methodName
return proto.Unmarshal(data, message)
}