diff options
| -rw-r--r-- | src/ioq.erl | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/src/ioq.erl b/src/ioq.erl index b761a0b41..9bfb1f8e3 100644 --- a/src/ioq.erl +++ b/src/ioq.erl @@ -12,10 +12,14 @@ -module(ioq). -behaviour(gen_server). +-behaviour(config_listener). -export([start_link/0, call/3]). -export([init/1, handle_call/3, handle_cast/2, handle_info/2, code_change/3, terminate/2]). +% config_listener api +-export([handle_config_change/5]). + -record(state, { concurrency=10, ratio, @@ -45,12 +49,19 @@ call(Fd, Msg, Priority) -> end. init(_) -> + ok = config:listen_for_changes(?MODULE, nil), + State = #state{}, + {ok, read_config(State)}. + +read_config(State) -> Ratio = list_to_float(config:get("ioq", "ratio", "0.01")), - {ok, #state{ratio=Ratio}}. + State#state{ratio=Ratio}. handle_call(#request{}=Request, From, State) -> {noreply, enqueue_request(Request#request{from=From}, State), 0}. +handle_cast(change, State) -> + {noreply, read_config(State)}; handle_cast(_Msg, State) -> {noreply, State}. @@ -76,6 +87,11 @@ handle_info({'DOWN', Ref, _, _, Reason}, State) -> handle_info(timeout, State) -> {noreply, maybe_submit_request(State)}. +handle_config_change("ioq", _, _, _, _) -> + {ok, gen_server:cast(?MODULE, change)}; +handle_config_change(_, _, _, _, _) -> + {ok, nil}. + code_change(_Vsn, State, _Extra) -> {ok, State}. |