/
Elixir.RabbitMQ.CLI.Ctl.Commands.ListStreamConsumerGroupsCommand.erl
115 lines (96 loc) · 3.22 KB
/
Elixir.RabbitMQ.CLI.Ctl.Commands.ListStreamConsumerGroupsCommand.erl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
%% The contents of this file are subject to the Mozilla Public 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 https://www.mozilla.org/MPL/
%%
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and
%% limitations under the License.
%%
%% The Original Code is RabbitMQ.
%%
%% The Initial Developer of the Original Code is GoPivotal, Inc.
%% Copyright (c) 2007-2024 Broadcom. All Rights Reserved. The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries. All rights reserved.
-module('Elixir.RabbitMQ.CLI.Ctl.Commands.ListStreamConsumerGroupsCommand').
-include_lib("rabbitmq_stream_common/include/rabbit_stream.hrl").
-behaviour('Elixir.RabbitMQ.CLI.CommandBehaviour').
-export([formatter/0,
scopes/0,
switches/0,
aliases/0,
usage/0,
usage_additional/0,
usage_doc_guides/0,
banner/2,
validate/2,
merge_defaults/2,
run/2,
output/2,
description/0,
help_section/0]).
formatter() ->
'Elixir.RabbitMQ.CLI.Formatters.PrettyTable'.
scopes() ->
[ctl, diagnostics, streams].
switches() ->
[{verbose, boolean}].
aliases() ->
[{'V', verbose}].
description() ->
<<"Lists groups of stream single active consumers "
"for a vhost">>.
help_section() ->
{plugin, stream}.
validate(Args, _) ->
ValidKeys = lists:map(fun atom_to_list/1, ?CONSUMER_GROUP_INFO_ITEMS),
case 'Elixir.RabbitMQ.CLI.Ctl.InfoKeys':validate_info_keys(Args,
ValidKeys)
of
{ok, _} ->
ok;
Error ->
Error
end.
merge_defaults([], Opts) ->
merge_defaults([rabbit_data_coercion:to_binary(Item)
|| Item <- ?CONSUMER_GROUP_INFO_ITEMS],
Opts);
merge_defaults(Args, Opts) ->
{Args, maps:merge(#{verbose => false, vhost => <<"/">>}, Opts)}.
usage() ->
<<"list_stream_consumer_groups [--vhost <vhost>] "
"[<column> ...]">>.
usage_additional() ->
Prefix = <<" must be one of ">>,
InfoItems =
'Elixir.Enum':join(
lists:usort(?CONSUMER_GROUP_INFO_ITEMS), <<", ">>),
[{<<"<column>">>, <<Prefix/binary, InfoItems/binary>>}].
usage_doc_guides() ->
[?STREAMS_GUIDE_URL].
run(Args,
#{node := NodeName,
vhost := VHost,
timeout := Timeout,
verbose := Verbose}) ->
InfoKeys =
case Verbose of
true ->
?CONSUMER_GROUP_INFO_ITEMS;
false ->
'Elixir.RabbitMQ.CLI.Ctl.InfoKeys':prepare_info_keys(Args)
end,
rabbit_misc:rpc_call(NodeName,
rabbit_stream_sac_coordinator,
consumer_groups,
[VHost, InfoKeys],
Timeout).
banner(_, _) ->
<<"Listing stream consumer groups ...">>.
output({ok, []}, _Opts) ->
ok;
output([], _Opts) ->
ok;
output(Result, _Opts) ->
'Elixir.RabbitMQ.CLI.DefaultOutput':output(Result).