Connected / Disconnected refactor.

This commit is contained in:
Miroslav Šedivý 2020-11-01 21:03:25 +01:00
parent e51541fe01
commit 51c6e0091f
5 changed files with 17 additions and 18 deletions

View file

@ -69,18 +69,18 @@ func (session *SessionCtx) SetPeer(peer types.Peer) {
session.peer = peer
}
func (session *SessionCtx) SetConnected() {
session.connected = true
session.manager.emmiter.Emit("connected", session)
}
func (session *SessionCtx) SetConnected(connected bool) {
session.connected = connected
func (session *SessionCtx) SetDisconnected() {
session.connected = false
session.manager.emmiter.Emit("disconnected", session)
session.socket = nil
// TODO: Refactor.
session.manager.Destroy(session.id)
if connected {
session.manager.emmiter.Emit("connected", session)
} else {
session.manager.emmiter.Emit("disconnected", session)
session.socket = nil
// TODO: Refactor.
session.manager.Destroy(session.id)
}
}
func (session *SessionCtx) Disconnect(reason string) error {
@ -96,7 +96,7 @@ func (session *SessionCtx) Disconnect(reason string) error {
return err
}
session.SetDisconnected()
session.SetConnected(false)
return nil
}