refactor(emqx_topic_metrics): Decorate remote procedure calls

This commit is contained in:
k32 2022-01-12 23:52:32 +01:00
parent 64d594d1df
commit e513583e70
2 changed files with 56 additions and 7 deletions

View File

@ -183,7 +183,8 @@ operate_topic_metrics(delete, #{bindings := #{topic := Topic0}}) ->
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
cluster_accumulation_metrics() -> cluster_accumulation_metrics() ->
case multicall(emqx_topic_metrics, metrics, []) of Nodes = mria_mnesia:running_nodes(),
case emqx_topic_metrics_proto_v1:metrics(Nodes) of
{SuccResList, []} -> {SuccResList, []} ->
{ok, accumulate_nodes_metrics(SuccResList)}; {ok, accumulate_nodes_metrics(SuccResList)};
{_, FailedNodes} -> {_, FailedNodes} ->
@ -191,7 +192,8 @@ cluster_accumulation_metrics() ->
end. end.
cluster_accumulation_metrics(Topic) -> cluster_accumulation_metrics(Topic) ->
case multicall(emqx_topic_metrics, metrics, [Topic]) of Nodes = mria_mnesia:running_nodes(),
case emqx_topic_metrics_proto_v1:metrics(Nodes, Topic) of
{SuccResList, []} -> {SuccResList, []} ->
case lists:filter(fun({error, _}) -> false; (_) -> true case lists:filter(fun({error, _}) -> false; (_) -> true
end, SuccResList) of end, SuccResList) of
@ -244,11 +246,13 @@ do_accumulation_metrics(MetricsIn, {MetricsAcc, _}) ->
end, #{}, Keys). end, #{}, Keys).
reset() -> reset() ->
_ = multicall(emqx_topic_metrics, reset, []), Nodes = mria_mnesia:running_nodes(),
_ = emqx_topic_metrics_proto_v1:reset(Nodes),
ok. ok.
reset(Topic) -> reset(Topic) ->
case multicall(emqx_topic_metrics, reset, [Topic]) of Nodes = mria_mnesia:running_nodes(),
case emqx_topic_metrics_proto_v1:reset(Nodes, Topic) of
{SuccResList, []} -> {SuccResList, []} ->
case lists:filter(fun({error, _}) -> true; (_) -> false case lists:filter(fun({error, _}) -> true; (_) -> false
end, SuccResList) of end, SuccResList) of
@ -262,9 +266,6 @@ reset(Topic) ->
%%-------------------------------------------------------------------- %%--------------------------------------------------------------------
%% utils %% utils
multicall(M, F, A) ->
emqx_rpc:multicall(mria_mnesia:running_nodes(), M, F, A).
reason2httpresp(quota_exceeded) -> reason2httpresp(quota_exceeded) ->
Msg = list_to_binary( Msg = list_to_binary(
io_lib:format("Max topic metrics count is ~p", io_lib:format("Max topic metrics count is ~p",

View File

@ -0,0 +1,48 @@
%%--------------------------------------------------------------------
%% Copyright (c) 2022 EMQ Technologies Co., Ltd. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%--------------------------------------------------------------------
-module(emqx_topic_metrics_proto_v1).
-behaviour(emqx_bpapi).
-export([ introduced_in/0
, metrics/1
, metrics/2
, reset/1
, reset/2
]).
-include_lib("emqx/include/bpapi.hrl").
introduced_in() ->
"5.0.0".
-spec metrics([node()]) -> emqx_rpc:multicall_result().
metrics(Nodes) ->
emqx_rpc:multicall(Nodes, emqx_topic_metrics, metrics, []).
-spec metrics([node()], emqx_types:topic()) -> emqx_rpc:multicall_result().
metrics(Nodes, Topic) ->
emqx_rpc:multicall(Nodes, emqx_topic_metrics, metrics, [Topic]).
-spec reset([node()]) -> emqx_rpc:multicall_result().
reset(Nodes) ->
emqx_rpc:multicall(Nodes, emqx_topic_metrics, reset, []).
-spec reset([node()], emqx_types:topic()) -> emqx_rpc:multicall_result().
reset(Nodes, Topic) ->
emqx_rpc:multicall(Nodes, emqx_topic_metrics, reset, [Topic]).