session id to session.

This commit is contained in:
Miroslav Šedivý 2020-11-14 16:03:12 +01:00
parent fdf5839547
commit e5eaf5c60c
3 changed files with 12 additions and 18 deletions

View file

@ -149,13 +149,13 @@ func (ws *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Request) e
// }
socket := &WebSocketCtx{
id: session.ID(),
session: session,
ws: ws,
address: ip,
connection: connection,
}
ok, reason := ws.handler.Connected(session.ID(), socket)
ok, reason := ws.handler.Connected(session, socket)
if !ok {
// TODO: Refactor
if err = connection.WriteJSON(message.Disconnect{
@ -226,7 +226,7 @@ func (ws *WebSocketManagerCtx) handle(connection *websocket.Conn, session types.
Str("raw", string(raw)).
Msg("received message from client")
if err := ws.handler.Message(session.ID(), raw); err != nil {
if err := ws.handler.Message(session, raw); err != nil {
ws.logger.Error().Err(err).Msg("message handler has failed")
}
case <-cancel: