wsConnection: call onDisconnect

This commit is contained in:
Anton Kaliaev 2017-11-07 19:16:05 -05:00
parent 47f5e37205
commit a01c226dc4
No known key found for this signature in database
GPG Key ID: 7B6881D965918214

View File

@ -457,10 +457,13 @@ func (wsc *wsConnection) OnStart() error {
return nil return nil
} }
// OnStop is a nop. // OnStop implements cmn.Service by calling OnDisconnect callback.
func (wsc *wsConnection) OnStop() { func (wsc *wsConnection) OnStop() {
// Both read and write loops close the websocket connection when they exit their loops. // Both read and write loops close the websocket connection when they exit their loops.
// The writeChan is never closed, to allow WriteRPCResponse() to fail. // The writeChan is never closed, to allow WriteRPCResponse() to fail.
if wsc.onDisconnect != nil {
wsc.onDisconnect(wsc.remoteAddr)
}
} }
// GetRemoteAddr returns the remote address of the underlying connection. // GetRemoteAddr returns the remote address of the underlying connection.