From 7194df4568e66c1f2fee86816ace3308ec9eb302 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Sat, 7 Dec 2013 13:40:37 +0100 Subject: Gracefully shutdown when stop_listener/1 is called Implements the `shutdown` option as documented previously. --- src/ranch.erl | 2 +- src/ranch_conns_sup.erl | 77 +++++++++++++++++++++++++++++++++++++--------- src/ranch_listener_sup.erl | 3 +- 3 files changed, 66 insertions(+), 16 deletions(-) (limited to 'src') diff --git a/src/ranch.erl b/src/ranch.erl index 9bcd328..641fc4d 100644 --- a/src/ranch.erl +++ b/src/ranch.erl @@ -120,7 +120,7 @@ child_spec(Ref, NbAcceptors, Transport, TransOpts, Protocol, ProtoOpts) andalso is_atom(Protocol) -> {{ranch_listener_sup, Ref}, {ranch_listener_sup, start_link, [ Ref, NbAcceptors, Transport, TransOpts, Protocol, ProtoOpts - ]}, permanent, 5000, supervisor, [ranch_listener_sup]}. + ]}, permanent, infinity, supervisor, [ranch_listener_sup]}. %% @doc Acknowledge the accepted connection. %% diff --git a/src/ranch_conns_sup.erl b/src/ranch_conns_sup.erl index f920919..e1ddbca 100644 --- a/src/ranch_conns_sup.erl +++ b/src/ranch_conns_sup.erl @@ -20,22 +20,24 @@ -module(ranch_conns_sup). %% API. --export([start_link/5]). +-export([start_link/6]). -export([start_protocol/2]). -export([active_connections/1]). %% Supervisor internals. --export([init/6]). +-export([init/7]). -export([system_continue/3]). -export([system_terminate/4]). -export([system_code_change/4]). -type conn_type() :: worker | supervisor. +-type shutdown() :: brutal_kill | timeout(). -record(state, { parent = undefined :: pid(), ref :: ranch:ref(), conn_type :: conn_type(), + shutdown :: shutdown(), transport = undefined :: module(), protocol = undefined :: module(), opts :: any(), @@ -45,11 +47,11 @@ %% API. --spec start_link(ranch:ref(), conn_type(), module(), timeout(), module()) - -> {ok, pid()}. -start_link(Ref, ConnType, Transport, AckTimeout, Protocol) -> +-spec start_link(ranch:ref(), conn_type(), shutdown(), module(), + timeout(), module()) -> {ok, pid()}. +start_link(Ref, ConnType, Shutdown, Transport, AckTimeout, Protocol) -> proc_lib:start_link(?MODULE, init, - [self(), Ref, ConnType, Transport, AckTimeout, Protocol]). + [self(), Ref, ConnType, Shutdown, Transport, AckTimeout, Protocol]). %% We can safely assume we are on the same node as the supervisor. %% @@ -94,17 +96,17 @@ active_connections(SupPid) -> %% Supervisor internals. --spec init(pid(), ranch:ref(), conn_type(), module(), timeout(), module()) - -> no_return(). -init(Parent, Ref, ConnType, Transport, AckTimeout, Protocol) -> +-spec init(pid(), ranch:ref(), conn_type(), shutdown(), + module(), timeout(), module()) -> no_return(). +init(Parent, Ref, ConnType, Shutdown, Transport, AckTimeout, Protocol) -> process_flag(trap_exit, true), ok = ranch_server:set_connections_sup(Ref, self()), MaxConns = ranch_server:get_max_connections(Ref), Opts = ranch_server:get_protocol_options(Ref), ok = proc_lib:init_ack(Parent, {ok, self()}), loop(#state{parent=Parent, ref=Ref, conn_type=ConnType, - transport=Transport, protocol=Protocol, opts=Opts, - ack_timeout=AckTimeout, max_conns=MaxConns}, 0, 0, []). + shutdown=Shutdown, transport=Transport, protocol=Protocol, + opts=Opts, ack_timeout=AckTimeout, max_conns=MaxConns}, 0, 0, []). loop(State=#state{parent=Parent, ref=Ref, conn_type=ConnType, transport=Transport, protocol=Protocol, opts=Opts, @@ -151,7 +153,7 @@ loop(State=#state{parent=Parent, ref=Ref, conn_type=ConnType, loop(State#state{opts=Opts2}, CurConns, NbChildren, Sleepers); {'EXIT', Parent, Reason} -> - exit(Reason); + terminate(State, Reason, NbChildren); {'EXIT', Pid, Reason} when Sleepers =:= [] -> report_error(Ref, Protocol, Pid, Reason), erase(Pid), @@ -190,12 +192,59 @@ loop(State=#state{parent=Parent, ref=Ref, conn_type=ConnType, [Ref, Msg]) end. +-spec terminate(#state{}, any(), non_neg_integer()) -> no_return(). +%% Kill all children and then exit. We unlink first to avoid +%% getting a message for each child getting killed. +terminate(#state{shutdown=brutal_kill}, Reason, _) -> + Pids = get_keys(true), + _ = [begin + unlink(P), + exit(P, kill) + end || P <- Pids], + exit(Reason); +%% Attempt to gracefully shutdown all children. +terminate(#state{shutdown=Shutdown}, Reason, NbChildren) -> + shutdown_children(), + _ = if + Shutdown =:= infinity -> + ok; + true -> + erlang:send_after(Shutdown, self(), kill) + end, + wait_children(NbChildren), + exit(Reason). + +%% Monitor processes so we can know which ones have shutdown +%% before the timeout. Unlink so we avoid receiving an extra +%% message. Then send a shutdown exit signal. +shutdown_children() -> + Pids = get_keys(true), + _ = [begin + monitor(process, P), + unlink(P), + exit(P, shutdown) + end || P <- Pids], + ok. + +wait_children(0) -> + ok; +wait_children(NbChildren) -> + receive + {'DOWN', _, process, Pid, _} -> + _ = erase(Pid), + wait_children(NbChildren - 1); + kill -> + Pids = get_keys(true), + _ = [exit(P, kill) || P <- Pids], + ok + end. + system_continue(_, _, {State, CurConns, NbChildren, Sleepers}) -> loop(State, CurConns, NbChildren, Sleepers). -spec system_terminate(any(), _, _, _) -> no_return(). -system_terminate(Reason, _, _, _) -> - exit(Reason). +system_terminate(Reason, _, _, {State, _, NbChildren, _}) -> + terminate(State, Reason, NbChildren). system_code_change(Misc, _, _, _) -> {ok, Misc}. diff --git a/src/ranch_listener_sup.erl b/src/ranch_listener_sup.erl index b0a6bd5..30017d0 100644 --- a/src/ranch_listener_sup.erl +++ b/src/ranch_listener_sup.erl @@ -38,9 +38,10 @@ start_link(Ref, NbAcceptors, Transport, TransOpts, Protocol, ProtoOpts) -> init({Ref, NbAcceptors, Transport, TransOpts, Protocol}) -> AckTimeout = proplists:get_value(ack_timeout, TransOpts, 5000), ConnType = proplists:get_value(connection_type, TransOpts, worker), + Shutdown = proplists:get_value(shutdown, TransOpts, 5000), ChildSpecs = [ {ranch_conns_sup, {ranch_conns_sup, start_link, - [Ref, ConnType, Transport, AckTimeout, Protocol]}, + [Ref, ConnType, Shutdown, Transport, AckTimeout, Protocol]}, permanent, infinity, supervisor, [ranch_conns_sup]}, {ranch_acceptors_sup, {ranch_acceptors_sup, start_link, [Ref, NbAcceptors, Transport, TransOpts]}, -- cgit v1.2.3