3pp-mirror/neko
1
0
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2025-05-31 09:57:08 +02:00
Code Issues Projects Releases Packages Wiki Activity

websocket connection change.

Browse source
This commit is contained in:
Miroslav Šedivý 2021-03-23 15:38:37 +01:00
parent 89081d9887
commit da1d073846
3 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
internal/websocket/manager.go
View file

@ -207,11 +207,13 @@ func (manager *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Reque
return
}
session.SetWebSocketPeer(&WebSocketPeerCtx{
peer := &WebSocketPeerCtx{
session: session,
manager: manager,
connection: connection,
})
}
session.SetWebSocketPeer(peer)
manager.logger.
Debug().
@ -219,7 +221,7 @@ func (manager *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Reque
Str("address", connection.RemoteAddr().String()).
Msg("connection started")
session.SetWebSocketConnected(true)
session.SetWebSocketConnected(peer, true)
defer func() {
manager.logger.
@ -228,7 +230,7 @@ func (manager *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Reque
Str("address", connection.RemoteAddr().String()).
Msg("connection ended")
session.SetWebSocketConnected(false)
session.SetWebSocketConnected(peer, false)
}()
manager.handle(connection, session)