Skip to content

Commit 78a43dc

Browse files
committed
Store tags in global parameters
1 parent f5ef64a commit 78a43dc

File tree

4 files changed

+41
-10
lines changed

4 files changed

+41
-10
lines changed

deps/rabbit/src/rabbit.erl

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838

3939
%%---------------------------------------------------------------------------
4040
%% Boot steps.
41-
-export([maybe_insert_default_data/0, boot_delegate/0, recover/0,
41+
-export([maybe_set_cluster_tags/0, maybe_insert_default_data/0, boot_delegate/0, recover/0,
4242
pg_local_amqp_session/0,
4343
pg_local_amqp_connection/0]).
4444

@@ -208,6 +208,12 @@
208208
{requires, recovery},
209209
{enables, routing_ready}]}).
210210

211+
212+
-rabbit_boot_step({cluster_tags,
213+
[{description, "Set cluster tags"},
214+
{mfa, {?MODULE, maybe_set_cluster_tags, []}},
215+
{requires, core_initialized}]}).
216+
211217
-rabbit_boot_step({routing_ready,
212218
[{description, "message delivery logic ready"},
213219
{requires, [core_initialized, recovery]}]}).
@@ -1138,6 +1144,24 @@ pg_local_amqp_connection() ->
11381144
pg_local_scope(Prefix) ->
11391145
list_to_atom(io_lib:format("~s_~s", [Prefix, node()])).
11401146

1147+
1148+
-spec maybe_set_cluster_tags() -> 'ok'.
1149+
1150+
maybe_set_cluster_tags() ->
1151+
maybe
1152+
not_found ?= rabbit_runtime_parameters:lookup_global(cluster_tags),
1153+
Tags = application:get_env(rabbit, cluster_tags, []),
1154+
false ?= Tags == [],
1155+
?LOG_INFO("Setting cluster tags...",
1156+
#{domain => ?RMQLOG_DOMAIN_GLOBAL}),
1157+
rabbit_runtime_parameters:set_global(cluster_tags, Tags, <<"internal_user">>)
1158+
else
1159+
_ ->
1160+
% Cluster tags are either already set (Other node, earlier start, CLI)
1161+
% Do nothing?
1162+
ok
1163+
end.
1164+
11411165
-spec maybe_insert_default_data() -> 'ok'.
11421166

11431167
maybe_insert_default_data() ->

deps/rabbitmq_cli/lib/rabbitmq/cli/ctl/commands/cluster_status_command.ex

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -398,12 +398,12 @@ defmodule RabbitMQ.CLI.Ctl.Commands.ClusterStatusCommand do
398398
defp cluster_tags(node, timeout) do
399399
case :rabbit_misc.rpc_call(
400400
node,
401-
:application,
402-
:get_env,
403-
[:rabbit, :cluster_tags],
401+
:rabbit_runtime_parameters,
402+
:value_global,
403+
[:cluster_tags],
404404
timeout) do
405-
{:ok, tags} -> tags
406-
_ -> []
405+
:not_found -> []
406+
tags -> tags
407407
end
408408
end
409409

deps/rabbitmq_management/src/rabbit_mgmt_wm_overview.erl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,4 +185,8 @@ transform_retention_intervals([{MaxAgeInSeconds, _}|Rest], Acc) ->
185185
transform_retention_intervals(Rest, [AccVal|Acc]).
186186

187187
cluster_tags() ->
188-
application:get_env(rabbit, cluster_tags, []).
188+
case rabbit_runtime_parameters:value_global(cluster_tags) of
189+
not_found ->
190+
[];
191+
Tags -> Tags
192+
end.

deps/rabbitmq_management/test/rabbit_mgmt_http_SUITE.erl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -288,8 +288,9 @@ init_per_testcase(Testcase = disabled_qq_replica_opers_test, Config) ->
288288
rabbit_ct_helpers:testcase_started(Config, Testcase);
289289
init_per_testcase(Testcase = cluster_tags_test, Config) ->
290290
Tags = [{<<"az">>, <<"us-east-3">>}, {<<"region">>,<<"us-east">>}, {<<"environment">>,<<"production">>}],
291-
rabbit_ct_broker_helpers:rpc_all(Config,
292-
application, set_env, [rabbit, cluster_tags, Tags]),
291+
rpc(
292+
Config, rabbit_runtime_parameters, set_global,
293+
[cluster_tags, Tags, none]),
293294
rabbit_ct_helpers:testcase_started(Config, Testcase);
294295
init_per_testcase(queues_detailed_test, Config) ->
295296
IsEnabled = rabbit_ct_broker_helpers:is_feature_flag_enabled(
@@ -358,7 +359,9 @@ end_per_testcase0(disabled_qq_replica_opers_test, Config) ->
358359
rpc(Config, application, unset_env, [rabbitmq_management, restrictions]),
359360
Config;
360361
end_per_testcase0(cluster_tags_test, Config) ->
361-
rpc(Config, application, unset_env, [rabbit, cluster_tags]),
362+
rpc(
363+
Config, rabbit_runtime_parameters, clear_global,
364+
[cluster_tags, none]),
362365
Config;
363366
end_per_testcase0(Testcase, Config)
364367
when Testcase == list_deprecated_features_test;

0 commit comments

Comments
 (0)