-
Notifications
You must be signed in to change notification settings - Fork 3.9k
/
rabbit_federation_exchange_link_sup_sup.erl
79 lines (66 loc) · 3.08 KB
/
rabbit_federation_exchange_link_sup_sup.erl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
%% This Source Code Form is subject to the terms of the Mozilla Public
%% License, v. 2.0. If a copy of the MPL was not distributed with this
%% file, You can obtain one at https://mozilla.org/MPL/2.0/.
%%
%% Copyright (c) 2007-2023 VMware, Inc. or its affiliates. All rights reserved.
%%
-module(rabbit_federation_exchange_link_sup_sup).
-behaviour(mirrored_supervisor).
-include_lib("rabbit_common/include/rabbit.hrl").
-define(SUPERVISOR, ?MODULE).
%% Supervises the upstream links for all exchanges (but not queues). We need
%% different handling here since exchanges want a mirrored sup.
-export([start_link/0, start_child/1, adjust/1, stop_child/1]).
-export([init/1]).
%%----------------------------------------------------------------------------
start_link() ->
_ = pg:start_link(),
%% This scope is used by concurrently starting exchange and queue links,
%% and other places, so we have to start it very early outside of the supervision tree.
%% The scope is stopped in stop/1.
rabbit_federation_pg:start_scope(),
mirrored_supervisor:start_link({local, ?SUPERVISOR}, ?SUPERVISOR,
?MODULE, []).
%% Note that the next supervisor down, rabbit_federation_link_sup, is common
%% between exchanges and queues.
start_child(X) ->
case mirrored_supervisor:start_child(
?SUPERVISOR,
{id(X), {rabbit_federation_link_sup, start_link, [X]},
transient, ?SUPERVISOR_WAIT, supervisor,
[rabbit_federation_link_sup]}) of
{ok, _Pid} -> ok;
{error, {already_started, _Pid}} ->
#exchange{name = ExchangeName} = X,
rabbit_log_federation:debug("Federation link for exchange ~tp was already started",
[rabbit_misc:rs(ExchangeName)]),
ok;
%% A link returned {stop, gone}, the link_sup shut down, that's OK.
{error, {shutdown, _}} -> ok
end.
adjust({clear_upstream, VHost, UpstreamName}) ->
[rabbit_federation_link_sup:adjust(Pid, X, {clear_upstream, UpstreamName}) ||
{#exchange{name = Name} = X, Pid, _, _} <- mirrored_supervisor:which_children(?SUPERVISOR),
Name#resource.virtual_host == VHost],
ok;
adjust(Reason) ->
[rabbit_federation_link_sup:adjust(Pid, X, Reason) ||
{X, Pid, _, _} <- mirrored_supervisor:which_children(?SUPERVISOR)],
ok.
stop_child(X) ->
case mirrored_supervisor:terminate_child(?SUPERVISOR, id(X)) of
ok -> ok;
{error, Err} ->
#exchange{name = ExchangeName} = X,
rabbit_log_federation:warning(
"Attempt to stop a federation link for exchange ~tp failed: ~tp",
[rabbit_misc:rs(ExchangeName), Err]),
ok
end,
ok = mirrored_supervisor:delete_child(?SUPERVISOR, id(X)).
%%----------------------------------------------------------------------------
init([]) ->
{ok, {{one_for_one, 1200, 60}, []}}.
%% See comment in rabbit_federation_queue_link_sup_sup:id/1
id(X = #exchange{policy = Policy}) -> X1 = rabbit_exchange:immutable(X),
X1#exchange{policy = Policy}.