summaryrefslogtreecommitdiff
path: root/src/rabbit_exchange_events.erl
blob: 776471357480356873378d2d0e999a133e95f222 (plain)
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
%%   The contents of this file are subject to the Mozilla Public License
%%   Version 1.1 (the "License"); you may not use this file except in
%%   compliance with the License. You may obtain a copy of the License at
%%   http://www.mozilla.org/MPL/
%%
%%   Software distributed under the License is distributed on an "AS IS"
%%   basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
%%   License for the specific language governing rights and limitations
%%   under the License.
%%
%%   The Original Code is RabbitMQ.
%%
%%   The Initial Developers of the Original Code are LShift Ltd,
%%   Cohesive Financial Technologies LLC, and Rabbit Technologies Ltd.
%%
%%   Portions created before 22-Nov-2008 00:00:00 GMT by LShift Ltd,
%%   Cohesive Financial Technologies LLC, or Rabbit Technologies Ltd
%%   are Copyright (C) 2007-2008 LShift Ltd, Cohesive Financial
%%   Technologies LLC, and Rabbit Technologies Ltd.
%%
%%   Portions created by LShift Ltd are Copyright (C) 2007-2009 LShift
%%   Ltd. Portions created by Cohesive Financial Technologies LLC are
%%   Copyright (C) 2007-2009 Cohesive Financial Technologies
%%   LLC. Portions created by Rabbit Technologies Ltd are Copyright
%%   (C) 2007-2009 Rabbit Technologies Ltd.
%%
%%   All Rights Reserved.
%%
%%   Contributor(s): ______________________________________.
%%

-module(rabbit_exchange_events).
-include("rabbit.hrl").

-behaviour(gen_server2).

-export([start_link/0]).
-export([init/1, handle_call/3, handle_cast/2, handle_info/2, terminate/2, code_change/3]).

-define(SERVER, ?MODULE).

-ifdef(use_specs).
-spec(start_link/0 :: () -> {'ok', pid()} | 'ignore' | {'error', any()}).
-endif.

%%----------------------------------------------------------------------------

start_link() ->
    gen_server2:start_link({local, ?SERVER}, ?MODULE, [], []).

%%---------------------------------------------------------------------------

with_exchange(#binding{exchange_name = ExchangeName}, Fun) ->
    case rabbit_exchange:lookup(ExchangeName) of
        {ok, X} ->
            Fun(X);
        not_found ->
            ok
    end.

handle_table_event({write, rabbit_exchange, X = #exchange{type = Type}, _OldRecs, _ActivityId}) ->
    %% Exchange created/recovered.
    ok = Type:init(X);
handle_table_event({delete, rabbit_exchange, {rabbit_exchange, _ExchangeName},
                    [X = #exchange{type = Type}], _ActivityId}) ->
    %% Exchange deleted.
    ok = Type:delete(X);
handle_table_event({write, rabbit_route, #route{binding = B}, _OldRecs, _ActivityId}) ->
    %% New binding.
    ok = with_exchange(B, fun (X = #exchange{type = Type}) -> Type:add_binding(X, B) end);
handle_table_event({delete, rabbit_route, #route{binding = B}, _OldRecs, _ActivityId}) ->
    %% Deleted binding.
    ok = with_exchange(B, fun (X = #exchange{type = Type}) -> Type:delete_binding(X, B) end);
handle_table_event(_Event) ->
    {error, unhandled_table_event}.

%%---------------------------------------------------------------------------

init([]) ->
    mnesia:subscribe({table, rabbit_exchange, detailed}),
    mnesia:subscribe({table, rabbit_route, detailed}),
    {ok, no_state}.

handle_call(Request, _From, State) ->
    {stop, {unhandled_call, Request}, State}.

handle_cast(Request, State) ->
    {stop, {unhandled_cast, Request}, State}.

handle_info({mnesia_table_event, Event}, State) ->
    case catch handle_table_event(Event) of
        {'EXIT', Reason} ->
            rabbit_log:error("Exchange event callback failed~n~p~n", [[{event, Event},
                                                                       {reason, Reason}]]);
        ok ->
            ok;
        {error, unhandled_table_event} ->
            rabbit_log:error("Unexpected mnesia_table_event~n~p~n", [Event])
    end,
    {noreply, State};
handle_info(Info, State) ->
    {stop, {unhandled_info, Info}, State}.

terminate(_Reason, _State) ->
    ok.

code_change(_OldVsn, State, _Extra) ->
    {ok, State}.