Listen to disconnects with @websocket_stream

[borisnieuwenhuis]

I need to know when a user disconnects to do cleanup on the stream.
But I cannot find anyway to get notified when a client disconnects.

With a class Handler(WebsocketListener) it is very obvious and you get a callback, but I cannot find any way to get on_disconnect
notification when using the @websocket_stream decorator.

Any ideas?

[borisnieuwenhuis]

Nevermind, I managed to catch the disconnect by wrapping the generator with a try finally block.
So obvious in hindsight.