diff options
Diffstat (limited to 'test/websocket_handler.erl')
| -rw-r--r-- | test/websocket_handler.erl | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/test/websocket_handler.erl b/test/websocket_handler.erl index 8e6915b..d06bfee 100644 --- a/test/websocket_handler.erl +++ b/test/websocket_handler.erl @@ -4,7 +4,8 @@ -behaviour(cowboy_http_handler). -behaviour(cowboy_http_websocket_handler). -export([init/3, handle/2, terminate/2]). --export([websocket_init/3, websocket_handle/3, websocket_terminate/3]). +-export([websocket_init/3, websocket_handle/3, + websocket_info/3, websocket_terminate/3]). init(_Any, _Req, _Opts) -> {upgrade, protocol, cowboy_http_websocket}. @@ -19,11 +20,14 @@ websocket_init(_TransportName, Req, _Opts) -> erlang:start_timer(1000, self(), <<"websocket_init">>), {ok, Req, undefined}. -websocket_handle({timeout, _Ref, Msg}, Req, State) -> +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_handle({websocket, Data}, Req, State) -> - {reply, Data, Req, State}. +websocket_info(_Info, Req, State) -> + {ok, Req, State}. websocket_terminate(_Reason, _Req, _State) -> ok. |