Skip to content

Commit

Permalink
Add start_link interface
Browse files Browse the repository at this point in the history
  • Loading branch information
NelsonVides committed Apr 22, 2024
1 parent 0f69ec9 commit 95d4ae6
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 0 deletions.
11 changes: 11 additions & 0 deletions src/cets.erl
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
-behaviour(gen_server).

-export([
start_link/2,
start/2,
stop/1,
insert/2,
Expand Down Expand Up @@ -70,6 +71,7 @@
]).

-ignore_xref([
start_link/2,
start/2,
stop/1,
insert/2,
Expand Down Expand Up @@ -275,6 +277,15 @@
%% We recommend to define that function if keys could have conflicts.
%% This function would be called once for each conflicting key.
%% We recommend to keep that function pure (or at least no blocking calls from it).
-spec start_link(table_name(), start_opts()) -> gen_server:start_ret().
start_link(Tab, Opts) when is_atom(Tab) ->
case check_opts(Opts) of
[] ->
gen_server:start_link({local, Tab}, ?MODULE, {Tab, Opts}, []);
Errors ->
{error, Errors}
end.

-spec start(table_name(), start_opts()) -> gen_server:start_ret().
start(Tab, Opts) when is_atom(Tab) ->
case check_opts(Opts) of
Expand Down
22 changes: 22 additions & 0 deletions test/cets_SUITE.erl
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ groups() ->

cases() ->
[
start_link_inits_and_accepts_records,
inserted_records_could_be_read_back,
insert_many_with_one_record,
insert_many_with_two_records,
Expand All @@ -52,6 +53,7 @@ cases() ->
join_works_with_existing_data_with_conflicts_and_defined_conflict_handler_and_more_keys,
join_works_with_existing_data_with_conflicts_and_defined_conflict_handler_and_keypos2,
bag_with_conflict_handler_not_allowed,
bag_with_conflict_handler_not_allowed_for_start_link,
insert_new_works,
insert_new_works_with_table_name,
insert_new_works_when_leader_is_back,
Expand Down Expand Up @@ -274,6 +276,12 @@ end_per_testcase(_, _Config) ->
log_modules() ->
[cets, cets_call, cets_long, cets_join, cets_discovery].

start_link_inits_and_accepts_records(Config) ->
Tab = make_name(Config),
start_link_local(Tab),
cets:insert(Tab, {alice, 32}),
[{alice, 32}] = ets:lookup(Tab, alice).

inserted_records_could_be_read_back(Config) ->
Tab = make_name(Config),
start_local(Tab),
Expand Down Expand Up @@ -747,6 +755,10 @@ bag_with_conflict_handler_not_allowed(Config) ->
{error, [bag_with_conflict_handler]} =
cets:start(make_name(Config), #{handle_conflict => fun resolve_highest/2, type => bag}).

bag_with_conflict_handler_not_allowed_for_start_link(Config) ->
{error, [bag_with_conflict_handler]} =
cets:start_link(make_name(Config), #{handle_conflict => fun resolve_highest/2, type => bag}).

join_with_the_same_pid(Config) ->
Tab = make_name(Config),
{ok, Pid} = start_local(Tab),
Expand Down Expand Up @@ -2917,6 +2929,16 @@ still_works(Pid) ->
ok = cets:insert(Pid, {1}),
{ok, [{1}]} = cets:remote_dump(Pid).

start_link_local(Name) ->
start_link_local(Name, #{}).

start_link_local(Name, Opts) ->
catch cets:stop(Name),
wait_for_name_to_be_free(node(), Name),
{ok, Pid} = cets:start_link(Name, Opts),
schedule_cleanup(Pid),
{ok, Pid}.

start_local(Name) ->
start_local(Name, #{}).

Expand Down

0 comments on commit 95d4ae6

Please sign in to comment.