handle_fallback -> is_enabled

Author: dumbbell

deps/rabbit/src/rabbit_db_cluster.erl

@@ -134,16 +134,10 @@ forget_member(Node, RemoveWhenOffline) ->
             ?LOG_DEBUG(
                "DB: removing cluster member `~ts`", [Node],
                #{domain => ?RMQLOG_DOMAIN_DB}),
-            rabbit_khepri:handle_fallback(
-              #{mnesia => fun() ->
-                                  forget_member_using_mnesia(
-                                    Node, RemoveWhenOffline)
-                          end,
-                khepri => fun() ->
-                                  forget_member_using_khepri(
-                                    Node, RemoveWhenOffline)
-                          end
-               });
+            case rabbit_khepri:is_enabled() of
+                true  -> forget_member_using_khepri(Node, RemoveWhenOffline);
+                false -> forget_member_using_mnesia(Node, RemoveWhenOffline)
+            end;
         true ->
             {error, {failed_to_remove_node, Node, rabbit_still_running}}
     end.
@@ -172,10 +166,10 @@ forget_member_using_khepri(Node, false = _RemoveWhenOffline) ->
 
 change_node_type(NodeType) ->
     rabbit_mnesia:ensure_node_type_is_permitted(NodeType),
-    rabbit_khepri:handle_fallback(
-      #{mnesia => fun() -> change_node_type_using_mnesia(NodeType) end,
-        khepri => ok
-       }).
+    case rabbit_khepri:is_enabled() of
+        true  -> ok;
+        false -> change_node_type_using_mnesia(NodeType)
+    end.
 
 change_node_type_using_mnesia(NodeType) ->
     rabbit_mnesia:change_cluster_node_type(NodeType).
@@ -285,10 +279,10 @@ check_consistency_using_khepri() ->
 %% command.
 
 cli_cluster_status() ->
-    rabbit_khepri:handle_fallback(
-      #{mnesia => fun cli_cluster_status_using_mnesia/0,
-        khepri => fun cli_cluster_status_using_khepri/0
-       }).
+    case rabbit_khepri:is_enabled() of
+        true  -> cli_cluster_status_using_khepri();
+        false -> cli_cluster_status_using_mnesia()
+    end.
 
 cli_cluster_status_using_mnesia() ->
     rabbit_mnesia:status().
@@ -297,13 +291,13 @@ cli_cluster_status_using_khepri() ->
     rabbit_khepri:cli_cluster_status().
 
 rename(Node, NodeMapList) ->
-    rabbit_khepri:handle_fallback(
-      #{mnesia => fun() -> rabbit_mnesia_rename:rename(Node, NodeMapList) end,
-        khepri => {error, not_supported}
-       }).
+    case rabbit_khepri:is_enabled() of
+        true  -> {error, not_supported};
+        false -> rabbit_mnesia_rename:rename(Node, NodeMapList)
+    end.
 
 update_cluster_nodes(DiscoveryNode) ->
-    rabbit_khepri:handle_fallback(
-      #{mnesia => fun() -> rabbit_mnesia:update_cluster_nodes(DiscoveryNode) end,
-        khepri => {error, not_supported}
-       }).
+    case rabbit_khepri:is_enabled() of
+        true  -> {error, not_supported};
+        false -> rabbit_mnesia:update_cluster_nodes(DiscoveryNode)
+    end.