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

Add the {active, boolean()} Websocket command

Browse source 
This command is currently not documented. It allows disabling
the reading of incoming data from the socket, and can be used
as a poor man's flow control.
This commit is contained in:
Loïc Hoguin 2018-09-21 14:04:20 +02:00
parent 6e784f1a45
commit f810d8dd64
No known key found for this signature in database
GPG key ID: 8A9DF795F6FED764
4 changed files with 52 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/cowboy_websocket.erl
View file

@ -77,6 +77,7 @@
ref :: ranch:ref(),
socket = undefined :: inet:socket() | {pid(), cowboy_stream:streamid()} | undefined,
transport = undefined :: module() | undefined,
active = true :: boolean(),
handler :: module(),
key = undefined :: undefined | binary(),
timeout = infinity :: timeout(),
@ -295,6 +296,8 @@ takeover(Parent, Ref, Socket, Transport, _Opts, Buffer,
false -> before_loop(State, HandlerState, #ps_header{buffer=Buffer})
end.
before_loop(State=#state{active=false}, HandlerState, ParseState) ->
loop(State, HandlerState, ParseState);
%% @todo We probably shouldn't do the setopts if we have not received a socket message.
%% @todo We need to hibernate when HTTP/2 is used too.
before_loop(State=#state{socket=Stream={Pid, _}, transport=undefined},
@ -516,6 +519,8 @@ commands([], State, []) ->
commands([], State, Data) ->
Result = transport_send(State, nofin, lists:reverse(Data)),
{Result, State};
commands([{active, Active}|Tail], State, Data) when is_boolean(Active) ->
commands(Tail, State#state{active=Active}, Data);
commands([Frame|Tail], State=#state{extensions=Extensions}, Data0) ->
Data = [cow_ws:frame(Frame, Extensions)|Data0],
case is_close_frame(Frame) of