blob: 8bfda77f27c339840403117ccb6023cdfb41c00c [file] [log] [blame]
% Licensed 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
% 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.
-export([start_link/0, init/1]).
-export([handle_config_change/5, handle_config_terminate/3]).
%% Helper macro for declaring children of supervisor
-define(CHILD(I, Type), {I, {I, start_link, []}, permanent, 5000, Type, [I]}).
-define(CHILD_WITH_ARGS(I, Type, Args), {I, {I, start_link, Args}, permanent, 5000, Type, [I]}).
start_link() ->
supervisor:start_link({local, ?MODULE}, ?MODULE, []).
init([]) ->
ok = ioq_config_listener:subscribe(),
IOQ2Children = ioq_server2_children(),
{ok, {
{one_for_one, 5, 10},
?CHILD_WITH_ARGS(config_listener_mon, worker, [?MODULE, nil]),
?CHILD(ioq_server, worker),
?CHILD(ioq_osq, worker)
| IOQ2Children
ioq_server2_children() ->
Bind = config:get_boolean("ioq2", "bind_to_schedulers", false),
ioq_server2_children(erlang:system_info(schedulers), Bind).
ioq_server2_children(Count, Bind) ->
lists:map(fun(I) ->
Name = list_to_atom("ioq_server_" ++ integer_to_list(I)),
{Name, {ioq_server2, start_link, [Name, I, Bind]}, permanent, 5000, worker, [Name]}
end, lists:seq(1, Count)).
get_ioq2_servers() ->
lists:map(fun(I) ->
list_to_atom("ioq_server_" ++ integer_to_list(I))
end, lists:seq(1, erlang:system_info(schedulers))).
handle_config_change("ioq", _Key, _Val, _Persist, St) ->
gen_server:cast(ioq_server, update_config),
{ok, St};
handle_config_change("ioq2" ++ _, _Key, _Val, _Persist, St) ->
lists:foreach(fun({_Id, Pid}) ->
gen_server:cast(Pid, update_config)
end, processes(ioq2)),
{ok, St};
handle_config_change(_Sec, _Key, _Val, _Persist, St) ->
{ok, St}.
handle_config_terminate(_Server, _Reason, _State) ->
gen_server:cast(ioq_server, update_config),
spawn(fun() ->
lists:foreach(fun({_Id, Pid}) ->
gen_server:cast(Pid, update_config)
end, processes(ioq2))
processes(ioq2) ->
processes(ioq) ->
processes(config_listener_mon) ->
processes(Arg) ->
{error, [
{expected_one_of, [ioq, ioq2, config_listener_mon]},
{got, Arg}]}.
filter_children(RegExp) ->
lists:filtermap(fun({Id, P, _, _}) ->
case re:run(atom_to_list(Id), RegExp) of
{match, _} -> {true, {Id, P}};
_ -> false
end, supervisor:which_children(?MODULE)).