aboutsummaryrefslogblamecommitdiffstats
path: root/test/remove_conn_and_wait_protocol.erl
blob: db688ffc81ce208ccc27e384d002c2f9a388c343 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
                                       
                           
 
                        
                  
 

                                                                     

                  
                                  
                              

                           
                                                     


                          
                                        
-module(remove_conn_and_wait_protocol).
-behaviour(ranch_protocol).

-export([start_link/4]).
-export([init/3]).

start_link(Ref, _, _, [{remove, MaybeRemove, Timeout}]) ->
	Pid = spawn_link(?MODULE, init, [Ref, MaybeRemove, Timeout]),
	{ok, Pid}.

init(Ref, MaybeRemove, Timeout) ->
	ranch:accept_ack(Ref),
	case MaybeRemove of
		true ->
			ranch:remove_connection(Ref);
		false ->
			ok
	end,
	receive after Timeout -> ok end.