summaryrefslogtreecommitdiff
path: root/src/dynomite_sup.erl
blob: f81369343321b402be7c8189386e647734dfc0b5 (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
%%%-------------------------------------------------------------------
%%% File:      dynomite_sup.erl
%%% @author    Cliff Moon <cliff@powerset.com> []
%%% @copyright 2008 Cliff Moon
%%% @doc
%%%
%%% @end
%%%
%%% @since 2008-06-27 by Cliff Moon
%%%-------------------------------------------------------------------
-module(dynomite_sup).
-author('cliff@powerset.com').

-behaviour(supervisor).

%% API
-export([start_link/1]).

%% Supervisor callbacks
-export([init/1]).

-include("../include/config.hrl").

-define(SERVER, ?MODULE).

%%====================================================================
%% API functions
%%====================================================================
%%--------------------------------------------------------------------
%% @spec start_link() -> {ok,Pid} | ignore | {error,Error}
%% @doc Starts the supervisor
%% @end
%%--------------------------------------------------------------------
start_link(Hints) ->
    supervisor:start_link(?MODULE, [Hints]).

%%====================================================================
%% Supervisor callbacks
%%====================================================================
%%--------------------------------------------------------------------
%% @spec init(Args) -> {ok,  {SupFlags,  [ChildSpec]}} |
%%                     ignore                          |
%%                     {error, Reason}
%% @doc Whenever a supervisor is started using
%% supervisor:start_link/[2,3], this function is called by the new process
%% to find out about restart strategy, maximum restart frequency and child
%% specifications.
%% @end
%%--------------------------------------------------------------------
init(Args) ->
    Node = node(),
    Nodes = running_nodes() ++ [node()],
    Membership = {membership,
                  {membership2, start_link, [Node, Nodes, Args]},
                  permanent,
                  1000,
                  worker,
                  [membership2]},
    MemEventMgr = {mem_event_manager,
                   {gen_event, start_link, [{local, membership_events}]},
                   permanent,
                   1000,
                   worker,
                   []},
    {ok, {{one_for_one,10,1}, [Membership, MemEventMgr]}}.


%%====================================================================
%% Internal functions
%%====================================================================

%% @doc get a list of running nodes visible to this local node
running_nodes() ->
    [Node || Node <- nodes([this,visible]), running(Node)].

%% @doc monitor the membership server on Node from here
running(Node) ->
    Ref = erlang:monitor(process, {membership, Node}),
    R = receive
            {'DOWN', Ref, _, _, _} -> false
        after 1 ->
                true
        end,
    erlang:demonitor(Ref),
    R.