riak-0.9.2 / apps / riak / src / riak_sup.erl

%% -------------------------------------------------------------------
%%
%% riak_sup: supervise the core Riak services
%%
%% Copyright (c) 2007-2010 Basho Technologies, Inc.  All Rights Reserved.
%%
%% This file is provided to you under the Apache License,
%% Version 2.0 (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.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing,
%% software distributed under the License is distributed on an
%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
%% KIND, either express or implied.  See the License for the
%% specific language governing permissions and limitations
%% under the License.
%%
%% -------------------------------------------------------------------

%% @doc supervise the core Riak services

-module(riak_sup).

-behaviour(supervisor).

-export([start_link/0]).
-export([init/1]).

-define (IF (Bool, A, B), if Bool -> A; true -> B end).

%% @spec start_link() -> ServerRet
%% @doc API for starting the supervisor.
start_link() ->
    supervisor:start_link({local, ?MODULE}, ?MODULE, []).

%% @spec init([]) -> SupervisorTree
%% @doc supervisor callback.
init([]) ->
    Eventer = {riak_eventer,
               {riak_eventer, start_link, []},
               permanent, 5000, worker, [riak_eventer]},
    VSup = {riak_vnode_sup,
            {riak_vnode_sup, start_link, []},
            permanent, infinity, supervisor, [riak_vnode_sup]},
    VMaster = {riak_vnode_master,
               {riak_vnode_master, start_link, []},
               permanent, 5000, worker, [riak_vnode_master]},
    HandoffListen = {riak_handoff_listener,
               {riak_handoff_listener, start_link, []},
               permanent, 5000, worker, [riak_handoff_listener]},
    RingMgr = {riak_ring_manager,
             {riak_ring_manager, start_link, []},
             permanent, 5000, worker, [riak_ring_manager]},
    Connect = {riak_connect,
             {riak_connect, start_link, []},
             permanent, 5000, worker, [riak_connect]},
    LocalLogger = {riak_local_logger,
                   {riak_local_logger, start_link, []},
                   permanent, 5000, worker, [riak_local_logger]},
    RiakPb  = [{riak_pb_listener,
                {riak_pb_listener, start_link, []},
                permanent, 5000, worker, [riak_pb_listener]},
               {riak_pb_socket_sup, 
                {riak_pb_socket_sup, start_link, []},
                permanent, infinity, supervisor, [riak_pb_socket_sup]}
               ],
    RiakWeb = {webmachine_mochiweb,
                 {webmachine_mochiweb, start, [riak_web:config()]},
                  permanent, 5000, worker, dynamic},
    RiakStat = {riak_stat,
                {riak_stat, start_link, []},
                permanent, 5000, worker, [riak_stat]},
    RiakJsMgr = {riak_js_manager,
                 {riak_js_manager, start_link, [riak:get_app_env(js_vm_count, 0)]},
                 permanent, 30000, worker, [riak_js_manager]},
    RiakJsSup = {riak_js_sup,
                 {riak_js_sup, start_link, []},
                 permanent, infinity, supervisor, [riak_js_sup]},
    % Figure out which processes we should run...
    IsWebConfigured = (riak:get_app_env(riak_web_ip) /= undefined) andalso (riak:get_app_env(riak_web_ip) /= "undefined"),
    IsPbConfigured = (riak:get_app_env(pb_ip) /= undefined)
        andalso (riak:get_app_env(pb_port) /= undefined),
    HasStorageBackend = (riak:get_app_env(storage_backend) /= undefined) andalso (riak:get_app_env(storage_backend) /= "undefined"),
    IsStatEnabled = (riak:get_app_env(riak_stat) == true),

    % Build the process list...
    Processes = lists:flatten([
        Eventer,
        VSup,
        ?IF(HasStorageBackend, VMaster, []),
        HandoffListen,
        RingMgr,
        Connect,
        LocalLogger,
        ?IF(IsWebConfigured, RiakWeb, []),
        ?IF(IsPbConfigured, RiakPb, []),
        ?IF(IsStatEnabled, RiakStat, []),
        RiakJsSup,
        RiakJsMgr
    ]),

    % Run the proesses...
    {ok, {{one_for_one, 10, 10}, Processes}}.
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.