blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-04 11:22:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

update writePingMsg logic

Browse Source
This commit is contained in:
Monet Lee 2024-07-10 11:40:46 +08:00
parent 32457ab8b5
commit b454aacd36
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/msggateway/client.go
View File

@ -347,7 +347,7 @@ func (c *Client) activeHeartbeat(ctx context.Context) {
for { for {
select { select {
case <-ticker.C: case <-ticker.C:
if err := c.conn.WriteMessage(PingMessage, nil); err != nil { if err := c.writePingMsg(); err != nil {
log.ZError(c.ctx, "send Ping Message error.", err) log.ZError(c.ctx, "send Ping Message error.", err)
return return
} }
@ -358,6 +358,21 @@ func (c *Client) activeHeartbeat(ctx context.Context) {
}() }()
} }
} }
func (c *Client) writePingMsg() error {
if c.closed.Load() {
return nil
}
c.w.Lock()
defer c.w.Unlock()
err := c.conn.SetWriteDeadline(writeWait)
if err != nil {
return err
}
return c.conn.WriteMessage(PingMessage, nil)
}
func (c *Client) writePongMsg() error { func (c *Client) writePongMsg() error {
if c.closed.Load() { if c.closed.Load() {