aboutsummaryrefslogtreecommitdiffstats
path: root/src/gun_ws.erl
diff options
context:
space:
mode:
Diffstat (limited to 'src/gun_ws.erl')
-rw-r--r--src/gun_ws.erl5
1 files changed, 4 insertions, 1 deletions
diff --git a/src/gun_ws.erl b/src/gun_ws.erl
index 176ba3b..ff54ecd 100644
--- a/src/gun_ws.erl
+++ b/src/gun_ws.erl
@@ -17,7 +17,7 @@
-export([check_options/1]).
-export([name/0]).
-export([init/8]).
--export([handle/2]).
+-export([handle/4]).
-export([close/2]).
-export([send/2]).
-export([down/1]).
@@ -72,6 +72,9 @@ init(Owner, Socket, Transport, StreamRef, Headers, Extensions, Handler, Opts) ->
{switch_protocol, ?MODULE, #ws_state{owner=Owner, socket=Socket, transport=Transport,
extensions=Extensions, handler=Handler, handler_state=HandlerState}}.
+handle(Data, State, _EvHandler, EvHandlerState) ->
+ {handle(Data, State), EvHandlerState}.
+
%% Do not handle anything if we received a close frame.
handle(_, State=#ws_state{in=close}) ->
{state, State};