| % 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 |
| % |
| % 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. |
| |
| -module(ioq_sup). |
| -behaviour(supervisor). |
| -vsn(1). |
| -behaviour(config_listener). |
| -export([start_link/0, init/1]). |
| -export([get_ioq2_servers/0]). |
| -export([handle_config_change/5, handle_config_terminate/3]). |
| -export([processes/1]). |
| |
| %% 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:call(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:call(Pid, update_config) |
| end, processes(ioq2)) |
| end), |
| ok. |
| |
| processes(ioq2) -> |
| filter_children("^ioq_server_.*$"); |
| processes(ioq) -> |
| filter_children("^ioq_server$"); |
| processes(config_listener_mon) -> |
| filter_children("^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 |
| end, supervisor:which_children(?MODULE)). |