liwei_3d/go_dreamfactory
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

优化心跳代码

Browse Source
This commit is contained in:
liwei1dao 2023-01-29 15:02:51 +08:00
parent 705b16a362
commit c7d421e421
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
modules/gateway/agent.go
View File

@ -78,6 +78,17 @@ locp:
var code pb.ErrorCode var code pb.ErrorCode
code, err = this.secAuth(msg) code, err = this.secAuth(msg)
if err == nil { if err == nil {
if msg.MainType == string(comm.ModuleGate) { //心跳消息
data, _ := anypb.New(&pb.GatewayHeartbeatResp{
Timestamp: configure.Now().Unix(),
})
this.WriteMsg(&pb.UserMessage{
MainType: string(comm.ModuleGate),
SubType: "heartbeat",
Data: data,
})
continue
}
if err := this.messageDistribution(msg); err != nil { if err := this.messageDistribution(msg); err != nil {
go this.Close() go this.Close()
break locp break locp
@ -150,17 +161,6 @@ func (this *Agent) decodeUserData(msg *pb.UserMessage) (code pb.ErrorCode, err e
this.gateway.Errorf("last timestamp:%v more than 30s", timestamp) this.gateway.Errorf("last timestamp:%v more than 30s", timestamp)
return pb.ErrorCode_TimestampTimeout, fmt.Errorf("sec key expire") return pb.ErrorCode_TimestampTimeout, fmt.Errorf("sec key expire")
} }
if msg.MainType == string(comm.ModuleGate) { //心跳消息
data, _ := anypb.New(&pb.GatewayHeartbeatResp{
Timestamp: configure.Now().Unix(),
})
this.WriteMsg(&pb.UserMessage{
MainType: string(comm.ModuleGate),
SubType: "heartbeat",
Data: data,
})
return
}
//只有login的时候才需要设置Data //只有login的时候才需要设置Data
if msg.MainType == string(comm.ModuleUser) && msg.SubType == "login" { if msg.MainType == string(comm.ModuleUser) && msg.SubType == "login" {