riak_test/intercepts/riak_repl2_rtq_intercepts.erl
2013-08-19 11:33:36 -04:00

27 lines
852 B
Erlang

%% Intercepts functions for the riak_test in ../tests/repl_rt_heartbeat.erl
-module(riak_repl2_rtq_intercepts).
-compile(export_all).
-include("intercept.hrl").
-define(M, riak_repl2_rtq_orig).
%% @doc Drop the heartbeat messages from the rt source.
slow_trim_q(State) ->
%% ?I_INFO("slow_trim_q"),
%% This hideousness is necessary in order to have this intercept sleep only
%% on the first iteration. With hope, it causes the message queue of the
%% RTQ to spike enough to initiate overload handling, then subsequently
%% allows the queue to drain, overload to flip off, and the writes to complete.
case get(hosed) of
undefined ->
put(hosed, true);
true ->
timer:sleep(5000),
put(hosed, false);
false ->
put(hosed, false)
end,
?M:trim_q_orig(State).