summaryrefslogtreecommitdiff
path: root/bin/swift-get-nodes
blob: 279f8d82624e8200417763dd233cc935bc305914 (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
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#!/usr/bin/env python
# Copyright (c) 2010-2012 OpenStack, LLC.
#
# 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.

import optparse
import sys
import urllib

from swift.common.ring import Ring
from swift.common.utils import hash_path


parser = optparse.OptionParser()
parser.add_option('-a', '--all', action='store_true',
                  help='Show all handoff nodes')
parser.add_option('-p', '--partition', metavar='PARTITION',
                  help='Show nodes for a given partition')
(options, args) = parser.parse_args()

if (len(args) < 2 or len(args) > 4) and \
   (options.partition is None or not args):
    print 'Usage: %s [-a] <ring.gz> <account> [<container>] [<object>]' \
        % sys.argv[0]
    print '   Or: %s [-a] <ring.gz> -p partition' % sys.argv[0]
    print '   Note: account, container, object can also be a single arg ' \
        'separated by /'
    print 'Shows the nodes responsible for the item specified.'
    print 'Example:'
    print '    $ %s /etc/swift/account.ring.gz MyAccount' % sys.argv[0]
    print '    Partition 5743883'
    print '    Hash 96ae332a60b58910784e4417a03e1ad0'
    print '    10.1.1.7:8000 sdd1'
    print '    10.1.9.2:8000 sdb1'
    print '    10.1.5.5:8000 sdf1'
    print '    10.1.5.9:8000 sdt1 # [Handoff]'
    sys.exit(1)

if len(args) == 2 and '/' in args[1]:
    # Parse single path arg, as noted in above help text.
    path = args[1].lstrip('/')
    args = [args[0]] + [p for p in path.split('/', 2) if p]

ringloc = None
account = None
container = None
obj = None

if len(args) == 4:
    # Account, Container and Object
    ring_file, account, container, obj = args
    ring = Ring(ring_file)
    hash_str = hash_path(account, container, obj)
    part, nodes = ring.get_nodes(account, container, obj)
    target = "%s/%s/%s" % (account, container, obj)
    loc = 'objects'
elif len(args) == 3:
    # Account, Container
    ring_file, account, container = args
    ring = Ring(ring_file)
    hash_str = hash_path(account, container)
    part, nodes = ring.get_nodes(account, container)
    target = "%s/%s" % (account, container)
    loc = 'containers'
elif len(args) == 2:
    # Account
    ring_file, account = args
    ring = Ring(ring_file)
    hash_str = hash_path(account)
    part, nodes = ring.get_nodes(account)
    target = "%s" % (account)
    loc = 'accounts'
elif len(args) == 1:
    # Partition
    ring_file = args[0]
    ring = Ring(ring_file)
    hash_str = None
    part = int(options.partition)
    nodes = ring.get_part_nodes(part)
    target = ''
    loc = ring_file.rsplit('/', 1)[-1].split('.', 1)[0]
    if loc in ('account', 'container', 'object'):
        loc += 's'
    else:
        loc = '<type>'

more_nodes = []
for more_node in ring.get_more_nodes(part):
    more_nodes.append(more_node)
    if not options.all and len(more_nodes) >= len(nodes):
        break

print '\nAccount \t%s' % account
print 'Container\t%s' % container
print 'Object   \t%s\n' % obj
print '\nPartition\t%s' % part
print 'Hash       \t%s\n' % hash_str

for node in nodes:
    print 'Server:Port Device\t%s:%s %s' % (node['ip'], node['port'],
                                            node['device'])
for mnode in more_nodes:
    print 'Server:Port Device\t%s:%s %s\t [Handoff]' \
        % (mnode['ip'], mnode['port'], mnode['device'])
print "\n"
for node in nodes:
    print 'curl -I -XHEAD "http://%s:%s/%s/%s/%s"' \
        % (node['ip'], node['port'], node['device'], part,
           urllib.quote(target))
for mnode in more_nodes:
    print 'curl -I -XHEAD "http://%s:%s/%s/%s/%s" # [Handoff]' \
        % (mnode['ip'], mnode['port'], mnode['device'], part,
           urllib.quote(target))
print "\n"
print 'Use your own device location of servers:'
print 'such as "export DEVICE=/srv/node"'
for node in nodes:
    if hash_str:
        print 'ssh %s "ls -lah ${DEVICE:-/srv/node}/%s/%s/%s/%s/%s/"' % (
            node['ip'], node['device'], loc, part, hash_str[-3:], hash_str)
    else:
        print 'ssh %s "ls -lah ${DEVICE:-/srv/node}/%s/%s/%s/"' % (
            node['ip'], node['device'], loc, part)
for mnode in more_nodes:
    if hash_str:
        print 'ssh %s "ls -lah ${DEVICE:-/srv/node}/%s/%s/%s/%s/%s/" '\
              '# [Handoff]' % (mnode['ip'], mnode['device'], loc, part,
                               hash_str[-3:], hash_str)
    else:
        print 'ssh %s "ls -lah ${DEVICE:-/srv/node}/%s/%s/%s/" # [Handoff]' % (
            mnode['ip'], mnode['device'], loc, part)