From d6b7ec654a876531dc0cfc7c65350d354bda2f52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Wed, 3 Jul 2019 16:53:36 +0200 Subject: Add the response_start event Thought it needed cow_http2_machine changes but everything was available. For HTTP/1.1 it is triggered when receiving data while expecting headers. For HTTP/2 it is triggered after we have received a HEADERS frame for streams in idle state. --- src/gun_http.erl | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) (limited to 'src/gun_http.erl') diff --git a/src/gun_http.erl b/src/gun_http.erl index 7edaf14..b7c5bc1 100644 --- a/src/gun_http.erl +++ b/src/gun_http.erl @@ -100,7 +100,19 @@ handle(<<>>, State, _, EvHandlerState) -> handle(_, #http_state{streams=[]}, _, EvHandlerState) -> {close, EvHandlerState}; %% Wait for the full response headers before trying to parse them. -handle(Data, State=#http_state{in=head, buffer=Buffer}, EvHandler, EvHandlerState) -> +handle(Data, State=#http_state{in=head, buffer=Buffer, + streams=[#stream{ref=StreamRef, reply_to=ReplyTo}|_]}, EvHandler, EvHandlerState0) -> + %% Send the event only if there was no data in the buffer. + %% If there is data in the buffer then we already sent the event. + EvHandlerState = case Buffer of + <<>> -> + EvHandler:response_start(#{ + stream_ref => StreamRef, + reply_to => ReplyTo + }, EvHandlerState0); + _ -> + EvHandlerState0 + end, Data2 = << Buffer/binary, Data/binary >>, case binary:match(Data2, <<"\r\n\r\n">>) of nomatch -> {{state, State#http_state{buffer=Data2}}, EvHandlerState}; -- cgit v1.2.3