Merge branch 'masaki-furuta-toggle_disk_monitor'

(cherry picked from commit 6e54d77)

Author: michaelklishin

src/rabbit_disk_monitor.erl

@@ -39,7 +39,7 @@
 -export([get_disk_free_limit/0, set_disk_free_limit/1,
          get_min_check_interval/0, set_min_check_interval/1,
          get_max_check_interval/0, set_max_check_interval/1,
-         get_disk_free/0]).
+         get_disk_free/0, set_enabled/1]).
 
 -define(SERVER, ?MODULE).
 -define(DEFAULT_MIN_DISK_CHECK_INTERVAL, 100).
@@ -110,6 +110,10 @@ set_max_check_interval(Interval) ->
 get_disk_free() ->
     gen_server:call(?MODULE, get_disk_free, infinity).
 
+-spec set_enabled(string()) -> 'ok'.
+set_enabled(Enabled) ->
+    gen_server:call(?MODULE, {set_enabled, Enabled}, infinity).
+
 %%----------------------------------------------------------------------------
 %% gen_server callbacks
 %%----------------------------------------------------------------------------
@@ -157,6 +161,15 @@ handle_call({set_max_check_interval, MaxInterval}, _From, State) ->
 handle_call(get_disk_free, _From, State = #state { actual = Actual }) ->
     {reply, Actual, State};
 
+handle_call({set_enabled, _Enabled = true}, _From, State) ->
+    start_timer(set_disk_limits(State, State#state.limit)),
+    rabbit_log:info("Free disk space monitor was enabled"),
+    {reply, ok, State#state{enabled = true}};
+handle_call({set_enabled, _Enabled = false}, _From, State) ->
+    erlang:cancel_timer(State#state.timer),
+    rabbit_log:info("Free disk space monitor was manually disabled"),
+    {reply, ok, State#state{enabled = false}};
+
 handle_call(_Request, _From, State) ->
     {noreply, State}.
 
@@ -236,14 +249,14 @@ parse_free_win32(CommandResult) ->
                              [{capture, all_but_first, list}]),
     list_to_integer(lists:reverse(Free)).
 
-interpret_limit({mem_relative, Relative}) 
+interpret_limit({mem_relative, Relative})
     when is_number(Relative) ->
     round(Relative * vm_memory_monitor:get_total_memory());
-interpret_limit(Absolute) -> 
+interpret_limit(Absolute) ->
     case rabbit_resource_monitor_misc:parse_information_unit(Absolute) of
         {ok, ParsedAbsolute} -> ParsedAbsolute;
         {error, parse_error} ->
-            rabbit_log:error("Unable to parse disk_free_limit value ~p", 
+            rabbit_log:error("Unable to parse disk_free_limit value ~p",
                              [Absolute]),
             ?DEFAULT_DISK_FREE_LIMIT
     end.