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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
-module(fabric_delete_db).
-author(brad@cloudant.com).
-export([delete_db/2]).
-include("../../couch/src/couch_db.hrl").
-include("../../dynomite/include/membership.hrl").
%% @doc Delete a database, and all its partition files across the cluster
%% Options is proplist with user_ctx, n, q
-spec delete_db(binary(), list()) -> {ok, #db{}} | {error, any()}.
delete_db(DbName, Options) ->
Fullmap = partitions:all_parts(DbName),
RefPartMap = send_delete_calls(Fullmap, Options),
Acc0 = {true, length(RefPartMap)},
case fabric_util:receive_loop(
RefPartMap, 1, fun handle_delete_msg/3, Acc0) of
{ok, _Results} ->
delete_fullmap(DbName),
ok;
Error -> Error
end.
%%
%% internal
%%
%% @doc delete the partitions on all appropriate nodes (rexi calls)
-spec send_delete_calls(fullmap(), list()) -> [{reference(), part()}].
send_delete_calls(Parts, Options) ->
lists:map(fun(#shard{node=Node, name=ShardName} = Part) ->
Ref = rexi:async_server_call({couch_server, Node},
{delete, ShardName, Options}),
{Ref, Part}
end, Parts).
handle_delete_msg(_, not_found, {NotFound, N}) ->
{ok, {NotFound, N-1}};
handle_delete_msg(_, {rexi_EXIT, _Reason}, {NotFound, N}) ->
{ok, {NotFound, N-1}};
handle_delete_msg(_, {rexi_DOWN, _, _, _}, _Acc) ->
{error, delete_db_fubar};
handle_delete_msg(_, _, {NotFound, 1}) ->
if
NotFound -> {stop, not_found};
true -> {stop, ok}
end;
handle_delete_msg(_, ok, {_NotFound, N}) ->
{ok, {false, N-1}}.
delete_fullmap(DbName) ->
case couch_db:open(<<"dbs">>, []) of
{ok, Db} ->
{ok, Doc} = couch_api:open_doc(Db, DbName, nil, []),
couch_api:update_doc(Db, Doc#doc{deleted=true});
Error -> Error
end.
|
