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

update heartbeat logic.

Browse Source
This commit is contained in:
Monet Lee 2024-07-10 10:44:06 +08:00
parent afd3421773
commit a91a8b9c8b
1 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
internal/msggateway/client.go
View File

@ -90,7 +90,7 @@ func (c *Client) ResetClient(ctx *UserConnContext, conn LongConn, longConnServer
c.closed.Store(false) c.closed.Store(false)
c.closedErr = nil c.closedErr = nil
c.token = ctx.GetToken() c.token = ctx.GetToken()
c.hbCtx, _ = context.WithTimeout(c.ctx, pongWait*2) c.hbCtx, _ = context.WithCancel(c.ctx)
} }
func (c *Client) pingHandler(_ string) error { func (c *Client) pingHandler(_ string) error {
@ -114,10 +114,7 @@ func (c *Client) readMessage() {
c.conn.SetReadLimit(maxMessageSize) c.conn.SetReadLimit(maxMessageSize)
_ = c.conn.SetReadDeadline(pongWait) _ = c.conn.SetReadDeadline(pongWait)
c.conn.SetPingHandler(c.pingHandler) c.conn.SetPingHandler(c.pingHandler)
if c.PlatformID == constant.WebPlatformID {
go c.heartbeat(c.hbCtx) go c.heartbeat(c.hbCtx)
}
for { for {
log.ZDebug(c.ctx, "readMessage") log.ZDebug(c.ctx, "readMessage")
@ -331,6 +328,7 @@ func (c *Client) writeBinaryMsg(resp Resp) error {
} }
func (c *Client) heartbeat(ctx context.Context) { func (c *Client) heartbeat(ctx context.Context) {
if c.PlatformID == constant.WebPlatformID {
log.ZDebug(ctx, "server initiative send heartbeat start.") log.ZDebug(ctx, "server initiative send heartbeat start.")
ticker := time.NewTicker(pingPeriod) ticker := time.NewTicker(pingPeriod)
defer ticker.Stop() defer ticker.Stop()
@ -346,6 +344,7 @@ func (c *Client) heartbeat(ctx context.Context) {
return return
} }
} }
}
} }
func (c *Client) writePongMsg() error { func (c *Client) writePongMsg() error {