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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
module LeapCli; module Commands
desc 'Database commands.'
command :db do |db|
db.desc 'Destroy one or more databases. If present, limit to FILTER nodes. For example `leap db destroy --db sessions,tokens testing`.'
db.arg_name 'FILTER', :optional => true
db.command :destroy do |destroy|
destroy.flag :db, :arg_name => "DATABASES", :desc => 'Comma separated list of databases to destroy (no space). Use "--db all" to destroy all databases.', :optional => true
destroy.flag :user, :arg_name => "USERS", :desc => 'Comma separated list of usernames. The storage databases for these user(s) will be destroyed.', :optional => true
destroy.action do |global_options,options,args|
dbs = (options[:db]||"").split(',')
users = (options[:user]||"").split(',')
if dbs.empty? && users.empty?
bail!('Either --db or --user is required.')
end
nodes = manager.filter(args)
if nodes.any?
nodes = nodes[:services => 'couchdb']
end
unless nodes.any?
bail! 'No db nodes selected.'
end
if users.any?
unless global_options[:yes]
say 'You are about to permanently destroy user databases for [%s] for nodes [%s].' % [users.join(', '), nodes.keys.join(', ')]
bail! unless agree("Continue? ")
end
destroy_user_dbs(nodes, users)
elsif dbs.any?
unless global_options[:yes]
if dbs.include?('all')
say 'You are about to permanently destroy all database data for nodes [%s].' % nodes.keys.join(', ')
else
say 'You are about to permanently destroy databases [%s] for nodes [%s].' % [dbs.join(', '), nodes.keys.join(', ')]
end
bail! unless agree("Continue? ")
end
if dbs.include?('all')
destroy_all_dbs(nodes)
else
destroy_dbs(nodes, dbs)
end
say 'You must run `leap deploy` in order to create the databases again.'
end
end
end
end
private
def destroy_all_dbs(nodes)
ssh_connect(nodes) do |ssh|
ssh.run('/etc/init.d/couchdb stop && test ! -z "$(ls /var/lib/couchdb 2> /dev/null)" && rm -r /var/lib/couchdb/* && echo "All DBs destroyed" || echo "DBs already destroyed"')
end
end
def destroy_dbs(nodes, dbs)
nodes.each_node do |node|
ssh_connect(node) do |ssh|
dbs.each do |db|
ssh.run(DESTROY_DB_COMMAND % {:db => db})
end
end
end
end
def destroy_user_dbs(nodes, users)
nodes.each_node do |node|
ssh_connect(node) do |ssh|
users.each do |user|
ssh.run(DESTROY_USER_DB_COMMAND % {:user => user})
end
end
end
end
DESTROY_DB_COMMAND = %{
if [ 200 = `curl -ns -w "%%{http_code}" -X GET "127.0.0.1:5984/%{db}" -o /dev/null` ]; then
echo "Result from DELETE /%{db}:" `curl -ns -X DELETE "127.0.0.1:5984/%{db}"`;
else
echo "Skipping db '%{db}': it does not exist or has already been deleted.";
fi
}
DESTROY_USER_DB_COMMAND = %{/srv/leap/couchdb/scripts/destroy-user-db --username %{user}}
end; end
|
