From b4c7749176e0a55b5763f3e04bf9312adff7ea82 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Sun, 2 Jun 2019 17:26:37 +0200
Subject: Add request_start, request_headers and request_end events

---
 src/gun_ws.erl | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/gun_ws.erl')

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};
-- 
cgit v1.2.3