Skip to content

Commit 9166ab1

Browse files
committed
refactor
1 parent 9aead0d commit 9166ab1

File tree

3 files changed

+20
-30
lines changed

3 files changed

+20
-30
lines changed

deps/rabbit/src/rabbit_amqp_session.erl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3980,7 +3980,7 @@ is_valid_max(Val) ->
39803980
list_local() ->
39813981
pg:which_groups(pg_scope()).
39823982

3983-
%% Returns 'true' if Pid is a local AMQP session process.
3983+
%% Returns true if Pid is a local AMQP session process.
39843984
-spec is_local(pid()) -> boolean().
39853985
is_local(Pid) ->
39863986
pg:get_local_members(pg_scope(), Pid) =/= [].

deps/rabbit/src/rabbit_channel.erl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -557,14 +557,6 @@ handle_call({{info, Items}, Deadline}, _From, State) ->
557557
reply({error, Error}, State)
558558
end;
559559

560-
handle_call(refresh_config, _From,
561-
State = #ch{cfg = #conf{virtual_host = VHost} = Cfg}) ->
562-
reply(ok, State#ch{cfg = Cfg#conf{trace_state = rabbit_trace:init(VHost)}});
563-
564-
handle_call(refresh_interceptors, _From, State) ->
565-
IState = rabbit_channel_interceptor:init(State),
566-
reply(ok, State#ch{interceptor_state = IState});
567-
568560
handle_call({has_state, #resource{virtual_host = Vhost,
569561
name = Name}, rabbit_volatile_queue},
570562
_From,
@@ -588,6 +580,14 @@ handle_call({declare_fast_reply_to, Key}, _From, State = #ch{direct_reply = Repl
588580
end,
589581
reply(Result, State);
590582

583+
handle_call(refresh_config, _From,
584+
State = #ch{cfg = #conf{virtual_host = VHost} = Cfg}) ->
585+
reply(ok, State#ch{cfg = Cfg#conf{trace_state = rabbit_trace:init(VHost)}});
586+
587+
handle_call(refresh_interceptors, _From, State) ->
588+
IState = rabbit_channel_interceptor:init(State),
589+
reply(ok, State#ch{interceptor_state = IState});
590+
591591
handle_call(list_queue_states, _From, State = #ch{queue_states = QueueStates}) ->
592592
%% For testing of cleanup only
593593
%% HACK

deps/rabbit/src/rabbit_volatile_queue.erl

Lines changed: 11 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -202,42 +202,32 @@ deliver0(Q, Msg) ->
202202

203203
-spec deliver_local(pid(), rabbit_amqqueue:name(), mc:state()) -> ok.
204204
deliver_local(Pid, QName, Msg) ->
205-
Evt = {queue_event, QName, {deliver, Msg}},
206205
%% We ensure clients can't use this mechanism to send a message
207206
%% to an arbitrary process and kill it.
208207
case rabbit_amqp_session:is_local(Pid) orelse
209208
pg_local:in_group(rabbit_channels, Pid) of
210209
true ->
211-
gen_server:cast(Pid, Evt);
210+
gen_server:cast(Pid, {queue_event, QName, {deliver, Msg}});
212211
false ->
213212
ok
214213
end.
215214

216215
handle_event(QName, {deliver, Msg}, #?STATE{name = QName,
217216
ctag = Ctag,
218217
credit = undefined} = State) ->
219-
Actions = deliver_actions(QName, Ctag, Msg),
220-
{ok, State, Actions};
218+
{ok, State, deliver_actions(QName, Ctag, Msg)};
221219
handle_event(QName, {deliver, Msg}, #?STATE{name = QName,
222220
ctag = Ctag,
223221
delivery_count = DeliveryCount,
224-
credit = Credit,
225-
dropped = Dropped} = State0) ->
226-
{DC, C, D, Actions} = case Credit > 0 of
227-
true ->
228-
{serial_number:add(DeliveryCount, 1),
229-
Credit - 1,
230-
Dropped,
231-
deliver_actions(QName, Ctag, Msg)};
232-
false ->
233-
rabbit_global_counters:messages_dead_lettered(
234-
maxlen, ?MODULE, disabled, 1),
235-
{DeliveryCount, Credit, Dropped + 1, []}
236-
end,
237-
State = State0#?STATE{delivery_count = DC,
238-
credit = C,
239-
dropped = D},
240-
{ok, State, Actions}.
222+
credit = Credit} = State0)
223+
when Credit > 0 ->
224+
State = State0#?STATE{delivery_count = serial_number:add(DeliveryCount, 1),
225+
credit = Credit - 1},
226+
{ok, State, deliver_actions(QName, Ctag, Msg)};
227+
handle_event(QName, {deliver, _Msg}, #?STATE{name = QName,
228+
dropped = Dropped} = State) ->
229+
rabbit_global_counters:messages_dead_lettered(maxlen, ?MODULE, disabled, 1),
230+
{ok, State#?STATE{dropped = Dropped + 1}, []}.
241231

242232
deliver_actions(QName, Ctag, Mc) ->
243233
Msgs = [{QName, self(), undefined, _Redelivered = false, Mc}],

0 commit comments

Comments
 (0)