diff options
| author | Alexandru Scvortov <alexandru@rabbitmq.com> | 2010-08-24 16:03:40 +0100 |
|---|---|---|
| committer | Alexandru Scvortov <alexandru@rabbitmq.com> | 2010-08-24 16:03:40 +0100 |
| commit | d08b31fa31f2136ef2f4d0f25eb9fcba44ba968c (patch) | |
| tree | da45da3dac21c7ad7b26f6a0aad21bcefd721880 | |
| parent | 0e1ab7095752ea0a50bb637603fa99b582f8f945 (diff) | |
| download | rabbitmq-server-git-d08b31fa31f2136ef2f4d0f25eb9fcba44ba968c.tar.gz | |
migrated rabbit_limiter to prioritizers
| -rw-r--r-- | src/rabbit_limiter.erl | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/rabbit_limiter.erl b/src/rabbit_limiter.erl index 9894a85005..d3cc5f6002 100644 --- a/src/rabbit_limiter.erl +++ b/src/rabbit_limiter.erl @@ -34,7 +34,7 @@ -behaviour(gen_server2). -export([init/1, terminate/2, code_change/3, handle_call/3, handle_cast/2, - handle_info/2]). + handle_info/2, prioritise_call/3]). -export([start_link/2, shutdown/1]). -export([limit/2, can_send/3, ack/2, register/2, unregister/2]). -export([get_limit/1, block/1, unblock/1]). @@ -115,7 +115,7 @@ get_limit(undefined) -> get_limit(Pid) -> rabbit_misc:with_exit_handler( fun () -> 0 end, - fun () -> gen_server2:pcall(Pid, 9, get_limit, infinity) end). + fun () -> gen_server2:call(Pid, get_limit, infinity) end). block(undefined) -> ok; @@ -135,6 +135,9 @@ unblock(LimiterPid) -> init([ChPid, UnackedMsgCount]) -> {ok, #lim{ch_pid = ChPid, volume = UnackedMsgCount}}. +prioritise_call(get_limit, _From, _State) -> 9; +prioritise_call(_Msg, _From, _State) -> 0. + handle_call({can_send, _QPid, _AckRequired}, _From, State = #lim{blocked = true}) -> {reply, false, State}; |