Add @behaviour to websocket_handler.ex

[ci skip]