forked from davyxu/cellnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
req.go
71 lines (50 loc) · 1.16 KB
/
req.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
package rpc
import (
"errors"
"github.com/davyxu/cellnet"
"github.com/davyxu/cellnet/codec"
"sync"
"sync/atomic"
)
var (
rpcIDSeq int64
requestByCallID sync.Map
)
type request struct {
id int64
onRecv func(interface{})
}
var ErrTimeout = errors.New("time out")
func (self *request) RecvFeedback(msg interface{}) {
// 异步和同步执行复杂,队列处理在具体的逻辑中手动处理
self.onRecv(msg)
}
func (self *request) Send(ses cellnet.Session, msg interface{}) {
//ctx, _ := ses.(cellnet.ContextSet)
data, meta, err := codec.EncodeMessage(msg, nil)
if err != nil {
log.Errorf("rpc request message encode error: %s", err)
return
}
ses.Send(&RemoteCallREQ{
MsgID: uint16(meta.ID),
Data: data,
CallID: self.id,
})
//codec.FreeCodecResource(meta.Codec, data, ctx)
}
func createRequest(onRecv func(interface{})) *request {
self := &request{
onRecv: onRecv,
}
self.id = atomic.AddInt64(&rpcIDSeq, 1)
requestByCallID.Store(self.id, self)
return self
}
func getRequest(callid int64) *request {
if v, ok := requestByCallID.Load(callid); ok {
requestByCallID.Delete(callid)
return v.(*request)
}
return nil
}