zishuo/natpass
1
0
Fork 0
mirror of https://github.com/lwch/natpass synced 2025-10-17 10:20:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

修正keepalive包的处理逻辑问题

Browse Source
This commit is contained in:
lwch
2021-08-20 10:12:49 +08:00
parent c7f73d7d80
commit a9a2b65a83
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
code/server/handler/client.go
View File

@@ -49,6 +49,7 @@ func (c *client) run() {
logging.Error("read message from %s: %v", c.id, err) logging.Error("read message from %s: %v", c.id, err)
return return
} }
c.updated = time.Now()
c.parent.onMessage(msg) c.parent.onMessage(msg)
} }
} }

7
code/server/handler/handler.go
View File

@@ -83,6 +83,9 @@ func (h *Handler) readHandshake(c *network.Conn) (string, error) {
// onMessage forward message // onMessage forward message
func (h *Handler) onMessage(msg *network.Msg) { func (h *Handler) onMessage(msg *network.Msg) {
if msg.GetXType() == network.Msg_keepalive {
return
}
to := msg.GetTo() to := msg.GetTo()
h.RLock() h.RLock()
cli := h.clients[to] cli := h.clients[to]
@@ -91,10 +94,6 @@ func (h *Handler) onMessage(msg *network.Msg) {
logging.Error("client %s not found", to) logging.Error("client %s not found", to)
return return
} }
cli.updated = time.Now()
if msg.GetXType() == network.Msg_keepalive {
return
}
h.msgHook(msg) h.msgHook(msg)
cli.writeMessage(msg) cli.writeMessage(msg)
} }