yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: abstract function call

This commit is contained in:
JimMoen 2024-01-15 15:32:29 +08:00
parent 8f7964f435
commit a18c4d193a
No known key found for this signature in database
1 changed files with 9 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
apps/emqx_prometheus/src/emqx_prometheus_data_integration.erl
View File

@ -103,7 +103,7 @@ collect(<<"json">>) ->
Rules = emqx_rule_engine:get_rules(), Rules = emqx_rule_engine:get_rules(),
Bridges = emqx_bridge:list(), Bridges = emqx_bridge:list(),
#{ #{
data_integration_overview => collect_data_integration(overview, {Rules, Bridges}), data_integration_overview => collect_data_integration_overview(Rules, Bridges),
rules => collect_data_integration(rules, Rules), rules => collect_data_integration(rules, Rules),
actions => collect_data_integration(actions, Bridges), actions => collect_data_integration(actions, Bridges),
connectors => collect_data_integration(connectors, Bridges) connectors => collect_data_integration(connectors, Bridges)
@ -111,15 +111,17 @@ collect(<<"json">>) ->
collect(<<"prometheus">>) -> collect(<<"prometheus">>) ->
prometheus_text_format:format(?PROMETHEUS_DATA_INTEGRATION_REGISTRY). prometheus_text_format:format(?PROMETHEUS_DATA_INTEGRATION_REGISTRY).
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% %%====================
%% API Helpers
add_collect_family(Name, Data, Callback, Type) -> add_collect_family(Name, Data, Callback, Type) ->
%% TODO: help document from Name
Callback(create_mf(Name, _Help = <<"">>, Type, ?MODULE, Data)). Callback(create_mf(Name, _Help = <<"">>, Type, ?MODULE, Data)).
collect_metrics(Name, Metrics) -> collect_metrics(Name, Metrics) ->
collect_di(Name, Metrics). collect_di(Name, Metrics).
collect_data_integration(overview, {Rules, Bridges}) -> collect_data_integration_overview(Rules, Bridges) ->
RulesD = rules_data(Rules), RulesD = rules_data(Rules),
ActionsD = actions_data(Rules), ActionsD = actions_data(Rules),
ConnectorsD = connectors_data(Bridges), ConnectorsD = connectors_data(Bridges),
@ -129,30 +131,15 @@ collect_data_integration(overview, {Rules, Bridges}) ->
M3 = lists:foldl(fun(K, AccIn) -> AccIn#{K => ?MG(K, ConnectorsD)} end, #{}, connectors()), M3 = lists:foldl(fun(K, AccIn) -> AccIn#{K => ?MG(K, ConnectorsD)} end, #{}, connectors()),
M4 = maybe_collect_schema_registry(), M4 = maybe_collect_schema_registry(),
lists:foldl(fun(M, AccIn) -> maps:merge(M, AccIn) end, #{}, [M1, M2, M3, M4]); lists:foldl(fun(M, AccIn) -> maps:merge(M, AccIn) end, #{}, [M1, M2, M3, M4]).
collect_data_integration(Type = rules, Rules) ->
collect_data_integration(Type, DataSeed) ->
maps:fold( maps:fold(
fun(K, V, Acc) -> fun(K, V, Acc) ->
zip_metrics(Type, K, V, Acc) zip_metrics(Type, K, V, Acc)
end, end,
[], [],
di_data(Type, Rules) di_data(Type, DataSeed)
);
collect_data_integration(Type = actions, Rules) ->
maps:fold(
fun(K, V, Acc) ->
zip_metrics(Type, K, V, Acc)
end,
[],
di_data(Type, Rules)
);
collect_data_integration(Type = connectors, Bridges) ->
maps:fold(
fun(K, V, Acc) ->
zip_metrics(Type, K, V, Acc)
end,
[],
di_data(Type, Bridges)
). ).
-if(?EMQX_RELEASE_EDITION == ee). -if(?EMQX_RELEASE_EDITION == ee).