-
Notifications
You must be signed in to change notification settings - Fork 881
/
proc.go
90 lines (69 loc) · 2.02 KB
/
proc.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package rpc
import (
"github.com/davyxu/cellnet"
"github.com/davyxu/cellnet/codec"
"github.com/davyxu/cellnet/msglog"
)
type RemoteCallMsg interface {
GetMsgID() uint16
GetMsgData() []byte
GetCallID() int64
}
func ResolveInboundEvent(inputEvent cellnet.Event) (ouputEvent cellnet.Event, handled bool, err error) {
if _, ok := inputEvent.(*RecvMsgEvent); ok {
return inputEvent, false, nil
}
rpcMsg, ok := inputEvent.Message().(RemoteCallMsg)
if !ok {
return inputEvent, false, nil
}
userMsg, _, err := codec.DecodeMessage(int(rpcMsg.GetMsgID()), rpcMsg.GetMsgData())
if err != nil {
return inputEvent, false, err
}
if msglog.IsMsgLogValid(int(rpcMsg.GetMsgID())) {
peerInfo := inputEvent.Session().Peer().(cellnet.PeerProperty)
log.Debugf("#rpc.recv(%s)@%d len: %d %s | %s",
peerInfo.Name(),
inputEvent.Session().ID(),
cellnet.MessageSize(userMsg),
cellnet.MessageToName(userMsg),
cellnet.MessageToString(userMsg))
}
switch inputEvent.Message().(type) {
case *RemoteCallREQ: // 服务端收到客户端的请求
return &RecvMsgEvent{
inputEvent.Session(),
userMsg,
rpcMsg.GetCallID(),
}, true, nil
case *RemoteCallACK: // 客户端收到服务器的回应
request := getRequest(rpcMsg.GetCallID())
if request != nil {
request.RecvFeedback(userMsg)
}
return inputEvent, true, nil
}
return inputEvent, false, nil
}
func ResolveOutboundEvent(inputEvent cellnet.Event) (handled bool, err error) {
rpcMsg, ok := inputEvent.Message().(RemoteCallMsg)
if !ok {
return false, nil
}
userMsg, _, err := codec.DecodeMessage(int(rpcMsg.GetMsgID()), rpcMsg.GetMsgData())
if err != nil {
return false, err
}
if msglog.IsMsgLogValid(int(rpcMsg.GetMsgID())) {
peerInfo := inputEvent.Session().Peer().(cellnet.PeerProperty)
log.Debugf("#rpc.send(%s)@%d len: %d %s | %s",
peerInfo.Name(),
inputEvent.Session().ID(),
cellnet.MessageSize(userMsg),
cellnet.MessageToName(userMsg),
cellnet.MessageToString(userMsg))
}
// 避免后续环节处理
return true, nil
}