0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-14 12:20:24 +00:00

Make Cowboy compatible with upcoming Ranch 2.0

This commit is contained in:
juhlig 2019-06-06 17:52:34 +02:00 committed by Loïc Hoguin
parent 7708fc77cd
commit 5b4e78fac4
No known key found for this signature in database
GPG key ID: 8A9DF795F6FED764
4 changed files with 32 additions and 18 deletions

View file

@ -196,24 +196,24 @@ before_loop(State=#state{socket=Socket, transport=Transport}, Buffer) ->
loop(State=#state{parent=Parent, socket=Socket, transport=Transport, opts=Opts,
timer=TimerRef, children=Children, in_streamid=InStreamID,
last_streamid=LastStreamID, streams=Streams}, Buffer) ->
{OK, Closed, Error} = Transport:messages(),
Messages = Transport:messages(),
InactivityTimeout = maps:get(inactivity_timeout, Opts, 300000),
receive
%% Discard data coming in after the last request
%% we want to process was received fully.
{OK, Socket, _} when InStreamID > LastStreamID ->
{OK, Socket, _} when OK =:= element(1, Messages), InStreamID > LastStreamID ->
before_loop(State, Buffer);
%% Socket messages.
{OK, Socket, Data} ->
{OK, Socket, Data} when OK =:= element(1, Messages) ->
%% Only reset the timeout if it is idle_timeout (active streams).
State1 = case Streams of
[] -> State;
_ -> set_timeout(State)
end,
parse(<< Buffer/binary, Data/binary >>, State1);
{Closed, Socket} ->
{Closed, Socket} when Closed =:= element(2, Messages) ->
terminate(State, {socket_error, closed, 'The socket has been closed.'});
{Error, Socket, Reason} ->
{Error, Socket, Reason} when Error =:= element(3, Messages) ->
terminate(State, {socket_error, Reason, 'An error has occurred on the socket.'});
%% Timeouts.
{timeout, Ref, {shutdown, Pid}} ->
@ -1406,17 +1406,17 @@ terminate_linger(State=#state{socket=Socket, transport=Transport, opts=Opts}) ->
end.
terminate_linger_loop(State=#state{socket=Socket, transport=Transport}, TimerRef) ->
{OK, Closed, Error} = Transport:messages(),
Messages = Transport:messages(),
%% We may already have a message in the mailbox when we do this
%% but it's OK because we are shutting down anyway.
case Transport:setopts(Socket, [{active, once}]) of
ok ->
receive
{OK, Socket, _} ->
{OK, Socket, _} when OK =:= element(1, Messages) ->
terminate_linger_loop(State, TimerRef);
{Closed, Socket} ->
{Closed, Socket} when Closed =:= element(2, Messages) ->
ok;
{Error, Socket, _} ->
{Error, Socket, _} when Error =:= element(3, Messages) ->
ok;
{timeout, TimerRef, linger_timeout} ->
ok;