umgeher/cowboy
0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-16 05:00:24 +00:00
Code Issues Wiki Activity

Remove duplicate code for ensure_response

Browse source
This commit is contained in:
Loïc Hoguin 2012-09-15 22:19:39 +02:00
parent 27d591180c
commit 0aaa717575
3 changed files with 27 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/cowboy_req.erl
View file

@ -76,6 +76,7 @@
-export([chunked_reply/3]).
-export([chunk/2]).
-export([upgrade_reply/3]).
-export([ensure_response/2]).
%% Misc API.
-export([compact/1]).
@ -826,6 +827,27 @@ upgrade_reply(Status, Headers, Req=#http_req{
], Req),
{ok, Req2#http_req{resp_state=done, resp_headers=[], resp_body= <<>>}}.
%% @doc Ensure the response has been sent fully.
%% @private
-spec ensure_response(req(), cowboy_http:status()) -> ok.
%% The response has already been fully sent to the client.
ensure_response(#http_req{resp_state=done}, _) ->
ok;
%% No response has been sent but everything apparently went fine.
%% Reply with the status code found in the second argument.
ensure_response(Req=#http_req{resp_state=waiting}, Status) ->
_ = reply(Status, [], [], Req),
ok;
%% Terminate the chunked body for HTTP/1.1 only.
ensure_response(#http_req{method='HEAD', resp_state=chunks}, _) ->
ok;
ensure_response(#http_req{version={1, 0}, resp_state=chunks}, _) ->
ok;
ensure_response(#http_req{socket=Socket, transport=Transport,
resp_state=chunks}, _) ->
Transport:send(Socket, <<"0\r\n\r\n">>),
ok.
%% Misc API.
%% @doc Compact the request data by removing all non-system information.