diff options
| -rw-r--r-- | deps/rabbit/test/rabbit_db_topic_exchange_SUITE.erl | 85 |
1 files changed, 84 insertions, 1 deletions
diff --git a/deps/rabbit/test/rabbit_db_topic_exchange_SUITE.erl b/deps/rabbit/test/rabbit_db_topic_exchange_SUITE.erl index df41111a22..4f57c200d1 100644 --- a/deps/rabbit/test/rabbit_db_topic_exchange_SUITE.erl +++ b/deps/rabbit/test/rabbit_db_topic_exchange_SUITE.erl @@ -22,7 +22,8 @@ all() -> groups() -> [ - {all_tests, [], all_tests()} + {all_tests, [], all_tests()}, + {benchmarks, [], benchmarks()} ]. all_tests() -> @@ -37,6 +38,11 @@ all_tests() -> build_multiple_key_from_deletion_events ]. +benchmarks() -> + [ + match_benchmark + ]. + init_per_suite(Config) -> rabbit_ct_helpers:log_environment(), rabbit_ct_helpers:run_setup_steps(Config). @@ -271,6 +277,83 @@ build_multiple_key_from_deletion_events1(Config) -> lists:sort([RK || {_, RK} <- rabbit_db_topic_exchange:trie_records_to_key(Records)])), passed. +%% --------------------------------------------------------------------------- +%% Benchmarks +%% --------------------------------------------------------------------------- + +match_benchmark(Config) -> + %% run the benchmark with Mnesia first + MnesiaResults = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, match_benchmark1, [Config]), + + %% migrate to Khepri + Servers = rabbit_ct_broker_helpers:get_node_configs(Config, nodename), + {T, ok} = timer:tc(fun() -> + rabbit_ct_broker_helpers:enable_feature_flag(Config, Servers, raft_based_metadata_store_phase1) + end), + ct:pal("~p: time to migrate to Khepri: ~.2fs", [?FUNCTION_NAME, T/1000000]), + + %% run the same same benchmark with Khepri enabled + KhepriResults = rabbit_ct_broker_helpers:rpc(Config, 0, ?MODULE, match_benchmark1, [Config]), + + %% print all the results first + maps:foreach(fun(Test, KhepriResult) -> + MnesiaResult = maps:get(Test, MnesiaResults), + ct:pal("~p: Test: ~p, Mnesia: ~.2fus, Khepri: ~.2fus", [?FUNCTION_NAME, Test, MnesiaResult, KhepriResult]) + end, KhepriResults), + + %% fail the test if needed + maps:foreach(fun(Test, KhepriResult) -> + MnesiaResult = maps:get(Test, MnesiaResults), + ?assert(KhepriResult < MnesiaResult * 1.5, "Khepri can't be significantly slower than Mnesia") + end, KhepriResults). + +match_benchmark1(_Config) -> + Src = rabbit_misc:r(?VHOST, exchange, <<"test-exchange">>), + Dst1 = rabbit_misc:r(?VHOST, queue, <<"test-queue1">>), + Dst2 = rabbit_misc:r(?VHOST, queue, <<"test-queue2">>), + Dst3 = rabbit_misc:r(?VHOST, queue, <<"test-queue3">>), + Dst4 = rabbit_misc:r(?VHOST, queue, <<"test-queue4">>), + Dst5 = rabbit_misc:r(?VHOST, queue, <<"test-queue5">>), + Dst6 = rabbit_misc:r(?VHOST, queue, <<"test-queue6">>), + + SimpleTopics = [list_to_binary("a.b." ++ integer_to_list(N)) || N <- lists:seq(1,1000)], + Bindings = [#binding{source = Src, key = RoutingKey, destination = Dst1, args = #{}} || RoutingKey <- SimpleTopics], + BindingRes = [rabbit_db_topic_exchange:set(Binding) || Binding <- Bindings], + ?assertMatch([ok], lists:uniq(BindingRes)), + ok = rabbit_db_topic_exchange:set(#binding{source = Src, key = <<"a.b.*">>, destination = Dst2, args = #{}}), + ok = rabbit_db_topic_exchange:set(#binding{source = Src, key = <<"a.b.*">>, destination = Dst3, args = #{}}), + ok = rabbit_db_topic_exchange:set(#binding{source = Src, key = <<"a.#">>, destination = Dst4, args = #{}}), + ok = rabbit_db_topic_exchange:set(#binding{source = Src, key = <<"*.b.42">>, destination = Dst5, args = #{}}), + ok = rabbit_db_topic_exchange:set(#binding{source = Src, key = <<"#">>, destination = Dst6, args = #{}}), + + {Tany, _} = timer:tc(fun() -> + [rabbit_db_topic_exchange:match(Src, <<"foo">>) || _ <- lists:seq(1, 100)] + end), + ?assertMatch([Dst6], rabbit_db_topic_exchange:match(Src, <<"foo">>)), + + {Tbar, _} = timer:tc(fun() -> + [rabbit_db_topic_exchange:match(Src, <<"a.b.bar">>) || _ <- lists:seq(1, 100)] + end), + ?assertEqual(lists:sort([Dst2,Dst3,Dst4,Dst6]), lists:sort(rabbit_db_topic_exchange:match(Src, <<"a.b.bar">>))), + + {Tbaz, _} = timer:tc(fun() -> + [rabbit_db_topic_exchange:match(Src, <<"baz.b.42">>) || _ <- lists:seq(1, 100)] + end), + ?assertEqual(lists:sort([Dst5,Dst6]), lists:sort(rabbit_db_topic_exchange:match(Src, <<"baz.b.42">>))), + + {Tsimple, Rsimple} = timer:tc(fun() -> + [rabbit_db_topic_exchange:match(Src, RoutingKey) + || RoutingKey <- SimpleTopics, RoutingKey =/= <<"a.b.123">>] + end), + ?assertEqual([Dst1,Dst2,Dst3,Dst4,Dst6], lists:sort(lists:uniq(hd(Rsimple)))), + + #{ + "average time to match `foo`" => Tany/100, + "average time to match `a.b.bar`" => Tbar/100, + "average time to match `baz.b.42`" => Tbaz/100, + "average time to match a simple topic" => Tsimple/length(SimpleTopics) + }. + subscribe_to_mnesia_changes([Table | Rest]) -> case mnesia:subscribe({table, Table, detailed}) of {ok, _} -> subscribe_to_mnesia_changes(Rest); |