root
/
rabbitmq-server
mirror of https://github.com/rabbitmq/rabbitmq-server.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #14480 from rabbitmq/mk-rabbitmq-shovel-prometheus-should-handle-single-atom-shovel-states 

rabbitmq_shovel_prometheus: handle single atom shovel states
This commit is contained in:
Michael Klishin 2025-09-02 03:04:52 -04:00 committed by GitHub
parent 9853bdae7f 323ce94fd8
commit 232d96840a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
deps/rabbitmq_shovel_prometheus/src/rabbit_shovel_prometheus_collector.erl vendored
View File

@ -6,6 +6,9 @@
%%
-module(rabbit_shovel_prometheus_collector).
-include_lib("kernel/include/logger.hrl").
-include_lib("rabbit_common/include/logging.hrl").
-behaviour(prometheus_collector).
-export([start/0, stop/0]).
@ -29,10 +32,15 @@ deregister_cleanup(_) -> ok.
collect_mf(_Registry, Callback) ->
Status = rabbit_shovel_status:status(500),
{StaticStatusGroups, DynamicStatusGroups} = lists:foldl(fun({_,static,{S, _}, _, _}, {SMap, DMap}) ->
{maps:update_with(S, fun(C) -> C + 1 end, 1, SMap), DMap};
({_,dynamic,{S, _}, _, _}, {SMap, DMap}) ->
{SMap, maps:update_with(S, fun(C) -> C + 1 end, 1, DMap)}
%% Shovel status can be an atom or a tuple of {string(), proplists:proplist()}
{StaticStatusGroups, DynamicStatusGroups} = lists:foldl(fun({_, static, S, _, _}, {SMap, DMap}) when is_atom(S) ->
{maps:update_with(S, fun(C) -> C + 1 end, 1, SMap), DMap};
({_, static, {S, _}, _, _}, {SMap, DMap}) ->
{maps:update_with(S, fun(C) -> C + 1 end, 1, SMap), DMap};
({_, dynamic, S, _, _}, {SMap, DMap}) when is_atom(S) ->
{SMap, maps:update_with(S, fun(C) -> C + 1 end, 1, DMap)};
({_, dynamic, {S, _}, _, _}, {SMap, DMap}) ->
{SMap, maps:update_with(S, fun(C) -> C + 1 end, 1, DMap)}
end, {#{}, #{}}, Status),
Metrics = [{rabbitmq_shovel_dynamic, gauge, "Number of dynamic shovels",
@ -47,5 +55,5 @@ add_metric_family({Name, Type, Help, Metrics}, Callback) ->
Callback(create_mf(Name, Help, Type, Metrics)).
%%====================================================================
%% Private Parts
%% Implementation
%%====================================================================