db

Author: dumbbell

deps/rabbit/src/rabbit_db.erl

@@ -15,6 +15,7 @@
 -export([init/0,
          reset/0,
          force_reset/0,
+         force_load_on_next_boot/0,
          is_virgin_node/0, is_virgin_node/1,
          dir/0,
          ensure_dir_exists/0]).
@@ -89,6 +90,22 @@ force_reset_using_mnesia() ->
       #{domain => ?RMQLOG_DOMAIN_DB}),
     rabbit_mnesia:force_reset().
 
+-spec force_load_on_next_boot() -> Ret when
+      Ret :: ok.
+%% @doc Requests that the database to be forcefully loaded during next boot.
+%%
+%% This is necessary when a node refuses to boot when the cluster is in a bad
+%% state, like a critical members are MIA.
+
+force_load_on_next_boot() ->
+    force_load_on_next_boot_using_mnesia().
+
+force_load_on_next_boot_using_mnesia() ->
+    ?LOG_DEBUG(
+      "DB: resetting node forcefully",
+      #{domain => ?RMQLOG_DOMAIN_DB}),
+    rabbit_mnesia:force_load_next_boot().
+
 -spec is_virgin_node() -> IsVirgin when
       IsVirgin :: boolean().
 %% @doc Indicates if this RabbitMQ node is virgin.

deps/rabbit/src/rabbit_db_cluster.erl

@@ -135,7 +135,7 @@ check_consistency() ->
 check_consistency_using_mnesia() ->
     rabbit_mnesia:check_cluster_consistency().
 
--spec status() -> ClusterStatus when
+-spec cli_cluster_status() -> ClusterStatus when
       ClusterStatus :: [{nodes, [{rabbit_db_cluster:node_type(), [node()]}]} |
                         {running_nodes, [node()]} |
                         {partitions, [{node(), [node()]}]}].

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

@@ -33,16 +33,17 @@ defmodule RabbitMQ.CLI.Ctl.Commands.ClusterStatusCommand do
   use RabbitMQ.CLI.Core.RequiresRabbitAppRunning
 
   def run([], %{node: node_name, timeout: timeout}) do
-    status = case :rabbit_misc.rpc_call(node_name, :rabbit_db_cluster, :cli_cluster_status, []) do
-      {:badrpc, {:EXIT, {:undef, _}}} ->
-        :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :status, [])
+    status =
+      case :rabbit_misc.rpc_call(node_name, :rabbit_db_cluster, :cli_cluster_status, []) do
+        {:badrpc, {:EXIT, {:undef, _}}} ->
+          :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :status, [])
 
-      {:badrpc, _} = err ->
-        err
+        {:badrpc, _} = err ->
+          err
 
-      status ->
-        status
-    end
+        status ->
+          status
+      end
 
     case status do
       {:badrpc, _} = err ->

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

@@ -24,7 +24,16 @@ defmodule RabbitMQ.CLI.Ctl.Commands.ForceBootCommand do
   end
 
   def run([], %{node: node_name} = opts) do
-    case :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :force_load_next_boot, []) do
+    ret =
+      case :rabbit_misc.rpc_call(node_name, :rabbit_db, :force_load_on_next_boot, []) do
+        {:badrpc, {:EXIT, {:undef, _}}} ->
+          :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :force_load_next_boot, [])
+
+        ret0 ->
+          ret0
+      end
+
+    case ret do
       {:badrpc, :nodedown} ->
         case Config.get_option(:data_dir, opts) do
           nil ->

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

@@ -48,13 +48,14 @@ defmodule RabbitMQ.CLI.Ctl.Commands.ForgetClusterNodeCommand do
     atom_name = to_atom(node_to_remove)
     args = [atom_name, false]
 
-    ret = case :rabbit_misc.rpc_call(node_name, :rabbit_db_cluster, :forget_member, args) do
-      {:badrpc, {:EXIT, {:undef, _}}} ->
-        :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :forget_cluster_node, args)
-
-      ret0 ->
-        ret0
-    end
+    ret =
+      case :rabbit_misc.rpc_call(node_name, :rabbit_db_cluster, :forget_member, args) do
+        {:badrpc, {:EXIT, {:undef, _}}} ->
+          :rabbit_misc.rpc_call(node_name, :rabbit_mnesia, :forget_cluster_node, args)
+
+        ret0 ->
+          ret0
+      end
 
     case ret do
       {:error, {:failed_to_remove_node, ^atom_name, {:active, _, _}}} ->

deps/rabbitmq_management/src/rabbit_mgmt_wm_node.erl

@@ -59,7 +59,7 @@ augment(Key, ReqData, Node, Data) ->
     end.
 
 node_data(Node, ReqData) ->
-    S = rabbit_mnesia:status(),
+    S = rabbit_db_cluster:cli_cluster_status(),
     Nodes = proplists:get_value(nodes, S),
     Running = proplists:get_value(running_nodes, S),
     Type = find_type(Node, Nodes),

deps/rabbitmq_management/src/rabbit_mgmt_wm_nodes.erl

@@ -47,7 +47,7 @@ all_nodes(ReqData) ->
     end.
 
 all_nodes_raw() ->
-    S = rabbit_mnesia:status(),
+    S = rabbit_db_cluster:cli_cluster_status(),
     Nodes = proplists:get_value(nodes, S),
     Types = proplists:get_keys(Nodes),
     Running = proplists:get_value(running_nodes, S),