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
116
117
118
119
120
121
122
123
|
%% The contents of this file are subject to the Mozilla Public License
%% Version 1.1 (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) 2018 Pivotal Software, Inc. All rights reserved.
%%
-module(rabbit_core_ff).
-export([quorum_queue_migration/3,
implicit_default_bindings_migration/3,
virtual_host_metadata_migration/3]).
-rabbit_feature_flag(
{quorum_queue,
#{desc => "Support queues of type `quorum`",
doc_url => "https://www.rabbitmq.com/quorum-queues.html",
stability => stable,
migration_fun => {?MODULE, quorum_queue_migration}
}}).
-rabbit_feature_flag(
{implicit_default_bindings,
#{desc => "Default bindings are now implicit, instead of "
"being stored in the database",
stability => stable,
migration_fun => {?MODULE, implicit_default_bindings_migration}
}}).
-rabbit_feature_flag(
{virtual_host_metadata,
#{desc => "Virtual host metadata (description, tags, etc)",
stability => stable,
migration_fun => {?MODULE, virtual_host_metadata_migration}
}}).
%% -------------------------------------------------------------------
%% Quorum queues.
%% -------------------------------------------------------------------
-define(quorum_queue_tables, [rabbit_queue,
rabbit_durable_queue]).
quorum_queue_migration(FeatureName, _FeatureProps, enable) ->
Tables = ?quorum_queue_tables,
rabbit_table:wait(Tables, _Retry = true),
Fields = amqqueue:fields(amqqueue_v2),
migrate_to_amqqueue_with_type(FeatureName, Tables, Fields);
quorum_queue_migration(_FeatureName, _FeatureProps, is_enabled) ->
Tables = ?quorum_queue_tables,
rabbit_table:wait(Tables, _Retry = true),
Fields = amqqueue:fields(amqqueue_v2),
mnesia:table_info(rabbit_queue, attributes) =:= Fields andalso
mnesia:table_info(rabbit_durable_queue, attributes) =:= Fields.
migrate_to_amqqueue_with_type(FeatureName, [Table | Rest], Fields) ->
rabbit_log_feature_flags:info(
"Feature flag `~s`: migrating Mnesia table ~s...",
[FeatureName, Table]),
Fun = fun(Queue) -> amqqueue:upgrade_to(amqqueue_v2, Queue) end,
case mnesia:transform_table(Table, Fun, Fields) of
{atomic, ok} -> migrate_to_amqqueue_with_type(FeatureName,
Rest,
Fields);
{aborted, Reason} -> {error, Reason}
end;
migrate_to_amqqueue_with_type(FeatureName, [], _) ->
rabbit_log_feature_flags:info(
"Feature flag `~s`: Mnesia tables migration done",
[FeatureName]),
ok.
%% -------------------------------------------------------------------
%% Default bindings.
%% -------------------------------------------------------------------
implicit_default_bindings_migration(FeatureName, _FeatureProps,
enable) ->
%% Default exchange bindings are now implicit (not stored in the
%% route tables). It should be safe to remove them outside of a
%% transaction.
rabbit_table:wait([rabbit_queue]),
Queues = mnesia:dirty_all_keys(rabbit_queue),
remove_explicit_default_bindings(FeatureName, Queues);
implicit_default_bindings_migration(_Feature_Name, _FeatureProps,
is_enabled) ->
undefined.
remove_explicit_default_bindings(_FeatureName, []) ->
ok;
remove_explicit_default_bindings(FeatureName, Queues) ->
rabbit_log_feature_flags:info(
"Feature flag `~s`: deleting explicit default bindings "
"for ~b queues (it may take some time)...",
[FeatureName, length(Queues)]),
[rabbit_binding:remove_default_exchange_binding_rows_of(Q)
|| Q <- Queues],
ok.
%% -------------------------------------------------------------------
%% Virtual host metadata.
%% -------------------------------------------------------------------
virtual_host_metadata_migration(_FeatureName, _FeatureProps, enable) ->
Tab = rabbit_vhost,
rabbit_table:wait([Tab], _Retry = true),
Fun = fun(Row) -> vhost:upgrade_to(vhost_v2, Row) end,
case mnesia:transform_table(Tab, Fun, vhost:fields(vhost_v2)) of
{atomic, ok} -> ok;
{aborted, Reason} -> {error, Reason}
end;
virtual_host_metadata_migration(_FeatureName, _FeatureProps, is_enabled) ->
mnesia:table_info(rabbit_vhost, attributes) =:= vhost:fields(vhost_v2).
|