3pp-mirror/neko
1
0
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2025-06-05 12:22:52 +02:00
Code Issues Projects Releases Packages Wiki Activity

move websocket Ping to peer.

Browse source
This commit is contained in:
Miroslav Šedivý 2021-09-24 15:15:59 +02:00
parent 41d8fa7e22
commit d91c0a9f7f
3 changed files with 17 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
internal/websocket/manager.go
View file

@ -215,10 +215,10 @@ func (manager *WebSocketManagerCtx) connect(connection *websocket.Conn, r *http.
session.SetWebSocketConnected(peer, false)
}()
manager.handle(connection, session)
manager.handle(connection, peer, session)
}
func (manager *WebSocketManagerCtx) handle(connection *websocket.Conn, session types.Session) {
func (manager *WebSocketManagerCtx) handle(connection *websocket.Conn, peer types.WebSocketPeer, session types.Session) {
// add session id to logger context
logger := manager.logger.With().Str("session_id", session.ID()).Logger()
@ -281,11 +281,10 @@ func (manager *WebSocketManagerCtx) handle(connection *websocket.Conn, session t
case <-cancel:
return
case <-manager.shutdown:
err := connection.Close()
manager.logger.Err(err).Msg("connection shutdown")
peer.Destroy("connection shutdown")
return
case <-ticker.C:
if err := connection.WriteMessage(websocket.PingMessage, nil); err != nil {
if err := peer.Ping(); err != nil {
logger.Err(err).Msg("ping message has failed")
return
}