iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

replace SetConnected function.

Browse Source
This commit is contained in:
Miroslav Šedivý
2020-11-25 20:26:11 +01:00
parent fa54823098
commit f9b2460c3b
5 changed files with 27 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/websocket/manager.go
View File

@ -169,12 +169,16 @@ func (ws *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Request) e
Str("address", connection.RemoteAddr().String()).
Msg("connection started")
session.SetWebSocketConnected(true)
defer func() {
ws.logger.
Debug().
Str("session", session.ID()).
Str("address", connection.RemoteAddr().String()).
Msg("connection ended")
session.SetWebSocketConnected(false)
}()
ws.handle(connection, session)
@ -186,17 +190,11 @@ func (ws *WebSocketManagerCtx) handle(connection *websocket.Conn, session types.
cancel := make(chan struct{})
ticker := time.NewTicker(pingPeriod)
go func() {
// TODO: Change WebSocket connection state.
//session.SetConnected(true)
defer func() {
ticker.Stop()
ws.logger.Debug().Str("address", connection.RemoteAddr().String()).Msg("handle socket ending")
// TODO: Change WebSocket connection state.
session.SetConnected(false)
}()
defer func() {
ticker.Stop()
}()
go func() {
for {
_, raw, err := connection.ReadMessage()
if err != nil {