Split out direct and topic routing

This commit is contained in:
Ben Hood 2008-09-04 22:32:32 +01:00
parent 10c0fc56ad
commit 4391947d60
1 changed files with 24 additions and 29 deletions

View File

@ -185,48 +185,43 @@ simple_publish(Mandatory, Immediate,
route(#exchange{name = Name, type = topic}, RoutingKey) ->
route_internal(Name, RoutingKey, fun topic_matches/2);
% This matches for the direct exchanges and tries to short-cut the routing table
% for the default queue if that is what the user supplied
% route(#exchange{name = Name = #resource{name = <<>>}, type = direct}, RoutingKey) ->
% case route_internal(Name, RoutingKey) of
% [] -> route_internal(Name, RoutingKey, fun(X,Y) -> X == Y end);
% Other -> Other
% end;
route(#exchange{name = Name, type = Type}, RoutingKey) ->
route_internal(Name, RoutingKey, fun(X,Y) -> X == Y end).
route_internal(Name, RoutingKey).
% This returns a list of QPids to route to.
% Maybe this should be handled by a cursor instead.
% This routes directly to queues, avoiding any lookup of routes
% route_internal(#resource{name = <<>>, virtual_host = VHostPath}, RoutingKey) ->
% Query = qlc:q([QPid || #amqqueue{name = Queue, pid = QPid} <- mnesia:table(amqqueue),
% Queue == rabbit_misc:r(VHostPath, queue, RoutingKey)]),
% mnesia:activity(async_dirty, fun() -> qlc:e(Query) end).
route_internal(#resource{name = Name, virtual_host = VHostPath}, RoutingKey) ->
MatchHead = #route{binding = #binding{exchange_name = '$1',
queue_name = '$2',
key = '$3'}},
%MatchHead = #person{name='$1', sex=male, age='$2', _='_'},
Guards = [{'==', '$1', Name}, {'==', '$3', RoutingKey}],
lookup_qpids(
mnesia:activity(async_dirty,
fun() -> mnesia:select(route,[{MatchHead, Guards, ['$2']}])
end)).
% This returns a list of QPids to route to.
% Maybe this should be handled by a cursor instead.
route_internal(Exchange, RoutingKey, MatchFun) ->
Query = qlc:q([QName || #route{binding = #binding{exchange_name = ExchangeName,
queue_name = QName,
key = BindingKey}} <- mnesia:table(route),
ExchangeName == Exchange,
MatchFun(BindingKey, RoutingKey)]),
Fun = fun() -> qlc:e(Query) end,
{Time1,L} = timer:tc(mnesia,activity,[async_dirty,Fun]),
Set = sets:from_list(L),
Fun2 = fun() ->
queue_name = QName,
key = BindingKey}} <- mnesia:table(route),
ExchangeName == Exchange,
% This causes a full table scan (see bug 19336)
MatchFun(BindingKey, RoutingKey)]),
lookup_qpids(mnesia:activity(async_dirty, fun() -> qlc:e(Query) end)).
lookup_qpids(Queues) ->
Set = sets:from_list(Queues),
Fun = fun() ->
sets:fold(
fun(Key, Acc) -> [#amqqueue{pid = QPid}] = mnesia:read({amqqueue, Key}),
[QPid] ++ Acc end,
[], Set) end,
{Time2, QPids} = timer:tc(mnesia, activity, [async_dirty,Fun2]),
io:format("Time 1 -> ~p~n",[Time1]),
io:format("Time 2 -> ~p~n",[Time2]),
QPids.
mnesia:activity(async_dirty,Fun).
% Should all of the route and binding management not be refactored to it's own module
% Especially seeing as unbind will have to be implemented for 0.91 ?
delete_routes(Q = #amqqueue{name = Name}) ->