agent.go
2.58 KB
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package main
import (
"github.com/golang/protobuf/proto"
"math"
"pro2d/common"
"pro2d/common/components"
"pro2d/common/logger"
"pro2d/models"
"pro2d/utils"
"sync"
"sync/atomic"
)
type Agent struct {
components.IConnection
Server components.IServer
Role *models.RoleModel
nextCheckTime int64 //下一次检查的时间
lastHeartCheckTime int64
heartTimeoutCount int //超时次数
}
var agentPool = sync.Pool{New: func() interface{} { return new(Agent)}}
func NewAgent(s components.IServer) *Agent {
a := agentPool.Get().(*Agent)
a.Server = s
a.nextCheckTime = 0
a.lastHeartCheckTime = utils.Timex()
a.heartTimeoutCount= 0
return a
}
func (c *Agent) OnConnection(conn components.IConnection) {
c.IConnection = conn
}
func (c *Agent) OnMessage(msg components.IMessage) {
atomic.StoreInt64(&c.lastHeartCheckTime, utils.Timex())
md := c.Server.GetAction(msg.GetHeader().GetMsgID())
if md == nil {
logger.Debug("cmd: %d, handler is nil", msg.GetHeader().GetMsgID())
return
}
logger.Debug("protocolID: %d", msg.GetHeader().GetMsgID())
//fmt.Printf("errCode: %d, protoMsg:%v\n", errCode, protoMsg)
f := md.(func (msg components.IMessage) (int32, interface{}))
errCode, protoMsg := f(msg)
if protoMsg == nil {
return
}
rsp, err := proto.Marshal(protoMsg.(proto.Message))
if err != nil {
conn := msg.GetSession()
if conn != nil {
conn.Send(-100, msg.GetHeader().GetMsgID(), nil)
}
return
}
conn := msg.GetSession()
if conn != nil {
conn.Send(errCode, msg.GetHeader().GetMsgID(), rsp)
return
}
logger.Error("protocol not handler: %d", msg.GetHeader().GetMsgID())
}
func (c *Agent) OnTimer() {
nextCheckTime := atomic.LoadInt64(&c.nextCheckTime)
now := utils.Timex()
if now >= nextCheckTime {
//检查心跳
c.checkHeartBeat(now)
nextCheckTime = now + common.HeartTimerInterval
atomic.StoreInt64(&c.nextCheckTime, nextCheckTime)
}
if c.Role != nil {
//role 恢复数据
c.Role.OnRecoverTimer(now)
}
}
func (c *Agent) OnClose() {
c.Close()
}
func (c *Agent) Close() {
agentPool.Put(c)
if c.Role == nil {
return
}
c.Role.OnOfflineEvent()
}
func (c *Agent) checkHeartBeat(now int64) {
lastHeartCheckTime := atomic.LoadInt64(&c.lastHeartCheckTime)
//logger.Debug("checkHeartBeat ID: %d, last: %d, now: %d", c.GetID(), lastHeartCheckTime, now)
if math.Abs(float64(lastHeartCheckTime - now)) > common.HeartTimerInterval {
c.heartTimeoutCount++
if c.heartTimeoutCount >= common.HeartTimeoutCountMax {
c.Stop()
return
}
logger.Debug("timeout count: %d", c.heartTimeoutCount)
}else {
c.heartTimeoutCount = 0
}
}