mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 12:20:24 +00:00
293cf33702
Improves the readability of websocket handler code by having two functions: websocket_handle/3 handles the packets received from the socket, removing the tuple construct that was otherwise needed, so only websocket_handle(Data, Req, State) is needed now; websocket_info/3 handles the messages that the websocket handler process received, as websocket_info(Info, Req, State). Both functions return values are handled identically by Cowboy so nothing changes on that end.
33 lines
895 B
Erlang
33 lines
895 B
Erlang
%% Feel free to use, reuse and abuse the code in this file.
|
|
|
|
-module(websocket_handler).
|
|
-behaviour(cowboy_http_handler).
|
|
-behaviour(cowboy_http_websocket_handler).
|
|
-export([init/3, handle/2, terminate/2]).
|
|
-export([websocket_init/3, websocket_handle/3,
|
|
websocket_info/3, websocket_terminate/3]).
|
|
|
|
init(_Any, _Req, _Opts) ->
|
|
{upgrade, protocol, cowboy_http_websocket}.
|
|
|
|
handle(_Req, _State) ->
|
|
exit(badarg).
|
|
|
|
terminate(_Req, _State) ->
|
|
exit(badarg).
|
|
|
|
websocket_init(_TransportName, Req, _Opts) ->
|
|
erlang:start_timer(1000, self(), <<"websocket_init">>),
|
|
{ok, Req, undefined}.
|
|
|
|
websocket_handle(Data, Req, State) ->
|
|
{reply, Data, Req, State}.
|
|
|
|
websocket_info({timeout, _Ref, Msg}, Req, State) ->
|
|
erlang:start_timer(1000, self(), <<"websocket_handle">>),
|
|
{reply, Msg, Req, State};
|
|
websocket_info(_Info, Req, State) ->
|
|
{ok, Req, State}.
|
|
|
|
websocket_terminate(_Reason, _Req, _State) ->
|
|
ok.
|