wa_raft is open-source and cannot depend on WA specific applications

Summary: wa_pt back to persistent_term

Reviewed By: jaher, robertoaloi

Differential Revision: D85642877

fbshipit-source-id: cf9805e3f79e145538d6169af7d1c4b9d46b36b8

Author: Alexander Shturm

include/wa_raft.hrl

@@ -80,7 +80,7 @@
 %%-------------------------------------------------------------------
 
 -define(RAFT_METRICS_MODULE_KEY, {?RAFT_APPLICATION, raft_metrics_module}).
--define(RAFT_METRICS_MODULE, (wa_pt:get(?RAFT_METRICS_MODULE_KEY, wa_raft_metrics))).
+-define(RAFT_METRICS_MODULE, (persistent_term:get(?RAFT_METRICS_MODULE_KEY, wa_raft_metrics))).
 -define(RAFT_COUNT(Metric), ?RAFT_METRICS_MODULE:count(Metric)).
 -define(RAFT_COUNTV(Metric, Value), ?RAFT_METRICS_MODULE:countv(Metric, Value)).
 -define(RAFT_GATHER(Metric, Value), ?RAFT_METRICS_MODULE:gather(Metric, Value)).

src/wa_raft_log_catchup.erl

@@ -97,7 +97,7 @@
 
 -spec init_tables() -> term().
 init_tables() ->
-    wa_pt:put(?COUNTER_KEY, counters:new(?COUNTER_COUNT, [atomics])),
+    persistent_term:put(?COUNTER_KEY, counters:new(?COUNTER_COUNT, [atomics])),
     ?MODULE = ets:new(?MODULE, [set, public, named_table, {read_concurrency, true}]).
 
 -spec child_spec(Options :: #raft_options{}) -> supervisor:child_spec().
@@ -228,7 +228,7 @@ send_logs(Peer, NextLogIndex, LeaderTerm, LeaderCommitIndex, Witness, #state{nam
     LockoutMillis = maps:get(Peer, Lockouts, 0),
     NewState = case LockoutMillis =< StartMillis of
         true ->
-            Counters = wa_pt:get(?COUNTER_KEY),
+            Counters = persistent_term:get(?COUNTER_KEY),
             case counters:get(Counters, ?COUNTER_CONCURRENT_CATCHUP) < ?RAFT_MAX_CONCURRENT_LOG_CATCHUP() of
                 true ->
                     counters:add(Counters, ?COUNTER_CONCURRENT_CATCHUP, 1),
@@ -241,7 +241,7 @@ send_logs(Peer, NextLogIndex, LeaderTerm, LeaderCommitIndex, Witness, #state{nam
                                 [Name, LeaderTerm, Peer, T, E, S]
                             )
                     after
-                        counters:sub(wa_pt:get(?COUNTER_KEY), ?COUNTER_CONCURRENT_CATCHUP, 1)
+                        counters:sub(persistent_term:get(?COUNTER_KEY), ?COUNTER_CONCURRENT_CATCHUP, 1)
                     end,
                     EndMillis = erlang:system_time(millisecond),
                     ?RAFT_GATHER('raft.leader.catchup.duration', (EndMillis - StartMillis) * 1000),

src/wa_raft_metrics.erl

@@ -58,7 +58,7 @@
 %% RAFT metrics with the provided module.
 -spec install(Module :: module()) -> ok.
 install(Module) ->
-    wa_pt:put(?RAFT_METRICS_MODULE_KEY, Module).
+    persistent_term:put(?RAFT_METRICS_MODULE_KEY, Module).
 
 %%-------------------------------------------------------------------
 %% Default Implementation

src/wa_raft_part_sup.erl

@@ -85,13 +85,13 @@ start_link(Application, Spec) ->
     %% The RAFT options for a table are not expected to change during the
     %% runtime of the RAFT application and so repeated updates should not
     %% result in any GC load. Warn if this is case.
-    PrevOptions = wa_pt:get(?OPTIONS_KEY(Table, Partition), Options),
+    PrevOptions = persistent_term:get(?OPTIONS_KEY(Table, Partition), Options),
     PrevOptions =/= Options andalso
         ?RAFT_LOG_WARNING(
             ?MODULE_STRING " storing changed options for RAFT partitition ~0p/~0p",
             [Table, Partition]
         ),
-    ok = wa_pt:put(?OPTIONS_KEY(Table, Partition), Options),
+    ok = persistent_term:put(?OPTIONS_KEY(Table, Partition), Options),
 
     supervisor:start_link({local, Name}, ?MODULE, Options).
 
@@ -131,7 +131,7 @@ registered_partition_path(Table, Partition) ->
 
 -spec options(Table :: wa_raft:table(), Partition :: wa_raft:partition()) -> #raft_options{} | undefined.
 options(Table, Partition) ->
-    wa_pt:get(?OPTIONS_KEY(Table, Partition), undefined).
+    persistent_term:get(?OPTIONS_KEY(Table, Partition), undefined).
 
 -spec normalize_spec(Application :: atom(), Spec :: wa_raft:args()) -> #raft_options{}.
 normalize_spec(Application, #{table := Table, partition := Partition} = Spec) ->
@@ -174,7 +174,7 @@ normalize_spec(Application, #{table := Table, partition := Partition} = Spec) ->
 -spec prepare_spec(Application :: atom(), Spec :: wa_raft:args()) -> #raft_options{}.
 prepare_spec(Application, Spec) ->
     Options = #raft_options{table = Table, partition = Partition} = normalize_spec(Application, Spec),
-    ok = wa_pt:put(?OPTIONS_KEY(Table, Partition), Options),
+    ok = persistent_term:put(?OPTIONS_KEY(Table, Partition), Options),
     Options.
 
 %%-------------------------------------------------------------------

src/wa_raft_sup.erl

@@ -94,7 +94,7 @@ child_spec(Application, RaftArgs, Options) ->
 
 -spec start_link(Application :: atom(), Specs :: [wa_raft:args()], Options :: options()) -> supervisor:startlink_ret().
 start_link(Application, RaftArgs, Options) ->
-    ok = wa_pt:put(?OPTIONS_KEY(Application), normalize_spec(Application, Options)),
+    ok = persistent_term:put(?OPTIONS_KEY(Application), normalize_spec(Application, Options)),
     case supervisor:start_link({local, default_name(Application)}, ?MODULE, Application) of
         {ok, Pid} = Result ->
             [
@@ -160,7 +160,7 @@ registered_config_apps(Application) ->
 
 -spec options(Application :: atom()) -> #raft_application{} | undefined.
 options(Application) ->
-    wa_pt:get(?OPTIONS_KEY(Application), undefined).
+    persistent_term:get(?OPTIONS_KEY(Application), undefined).
 
 -spec normalize_spec(Application :: atom(), Options :: options()) -> #raft_application{}.
 normalize_spec(Application, Options) ->
@@ -180,7 +180,7 @@ prepare_application(Application) ->
 -spec prepare_application(Application :: atom(), Options :: options()) -> ok.
 prepare_application(Application, Options) ->
     RaftApplication = normalize_spec(Application, Options),
-    ok = wa_pt:put(?OPTIONS_KEY(Application), RaftApplication).
+    ok = persistent_term:put(?OPTIONS_KEY(Application), RaftApplication).
 
 %%-------------------------------------------------------------------
 %% Supervisor callbacks