package/go_process_manager
1
0
Fork 0
mirror of https://github.com/lzh-1625/go_process_manager.git synced 2025-10-05 07:56:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimal ws health check

Browse Source
This commit is contained in:
akrike
2025-02-16 12:47:39 +08:00
parent 7343f45dda
commit de296ee9cf
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/app/api/ws.go
View File

@@ -113,7 +113,7 @@ func (w *wsApi) startWsConnect(wci *WsConnetInstance, cancel context.CancelFunc,
pongChan <- struct{}{} pongChan <- struct{}{}
return nil return nil
}) })
timer := time.NewTicker(time.Second * time.Duration(config.CF.WsHealthCheckInterval)) timer := time.NewTicker(time.Second)
go func() { go func() {
defer timer.Stop() defer timer.Stop()
for { for {
@@ -121,13 +121,13 @@ func (w *wsApi) startWsConnect(wci *WsConnetInstance, cancel context.CancelFunc,
wci.WsConnect.WriteMessage(websocket.PingMessage, nil) wci.WsConnect.WriteMessage(websocket.PingMessage, nil)
wci.wsLock.Unlock() wci.wsLock.Unlock()
select { select {
case <-pongChan:
timer.Reset(time.Second * time.Duration(config.CF.WsHealthCheckInterval))
case <-timer.C: case <-timer.C:
log.Logger.Debugw("pong报文超时,结束ws连接")
cancel() cancel()
return return
case <-pongChan:
} }
time.Sleep(time.Second * time.Duration(config.CF.WsHealthCheckInterval))
timer.Reset(time.Second)
} }
}() }()