summaryrefslogtreecommitdiff
path: root/apps/fabric/src/fabric_db_delete.erl
blob: 57eefa9ed4fed9c32cf28b073add637434430797 (plain)
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
-module(fabric_db_delete).
-export([go/2]).

-include("fabric.hrl").
-include_lib("mem3/include/mem3.hrl").

go(DbName, Options) ->
    Shards = mem3:shards(DbName),
    Workers = fabric_util:submit_jobs(Shards, delete_db, [Options, DbName]),
    Acc0 = fabric_dict:init(Workers, nil),
    case fabric_util:recv(Workers, #shard.ref, fun handle_message/3, Acc0) of
    {ok, ok} ->
        ok;
    {ok, not_found} ->
        erlang:error(database_does_not_exist);
    Error ->
        Error
    end.

handle_message(Msg, Shard, Counters) ->
    C1 = fabric_dict:store(Shard, Msg, Counters),
    case fabric_dict:any(nil, C1) of
    true ->
        {ok, C1};
    false ->
        final_answer(C1)
    end.

final_answer(Counters) ->
    Successes = [X || {_, M} = X <- Counters, M == ok orelse M == not_found],
    case fabric_view:is_progress_possible(Successes) of
    true ->
        case lists:keymember(ok, 2, Successes) of
        true ->
            {stop, ok};
        false ->
            {stop, not_found}
        end;
    false ->
        {error, internal_server_error}
    end.