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
59
60
61
62
63
64
65
66
67
68
69
|
% Licensed under the Apache License, Version 2.0 (the "License"); you may not
% use this file except in compliance with the License. You may obtain a copy of
% the License at
%
% http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
% WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
% License for the specific language governing permissions and limitations under
% the License.
-module(smoosh).
-include_lib("couch/include/couch_db.hrl").
-include_lib("mem3/include/mem3.hrl").
-export([suspend/0, resume/0, enqueue/1, status/0]).
-export([enqueue_all_dbs/0, enqueue_all_dbs/1, enqueue_all_views/0]).
suspend() ->
smoosh_server:suspend().
resume() ->
smoosh_server:resume().
enqueue(Object) ->
smoosh_server:enqueue(Object).
sync_enqueue(Object) ->
smoosh_server:sync_enqueue(Object).
sync_enqueue(Object, Timeout) ->
smoosh_server:sync_enqueue(Object, Timeout).
status() ->
smoosh_server:status().
enqueue_all_dbs() ->
fold_local_shards(fun(#shard{name=Name}, _Acc) ->
sync_enqueue(Name) end, ok).
enqueue_all_dbs(Timeout) ->
fold_local_shards(fun(#shard{name=Name}, _Acc) ->
sync_enqueue(Name, Timeout) end, ok).
enqueue_all_views() ->
fold_local_shards(fun(#shard{name=Name}, _Acc) ->
catch enqueue_views(Name) end, ok).
fold_local_shards(Fun, Acc0) ->
mem3:fold_shards(fun(Shard, Acc1) ->
case node() == Shard#shard.node of
true ->
Fun(Shard, Acc1);
false ->
Acc1
end
end, Acc0).
enqueue_views(ShardName) ->
DbName = mem3:dbname(ShardName),
{ok, DDocs} = fabric:design_docs(DbName),
[sync_enqueue({ShardName, id(DDoc)}) || DDoc <- DDocs].
id(#doc{id=Id}) ->
Id;
id({Props}) ->
couch_util:get_value(<<"_id">>, Props).
|
