From 480df2e89ef20f1b2034f1566fef4858bc14436f Mon Sep 17 00:00:00 2001
From: Florian Hines <florian.hines@gmail.com>
Date: Wed, 28 Sep 2011 01:00:31 -0500
Subject: [PATCH] Query only specific zone via swift-recon

Add support to query only specific zones, as well a --all shortcut flag to run all checks.
Also skip deleted devices when trying to grab hosts from the ring.

Change-Id: I441ec76c90857c2e74262a7a9e2d36de89b28631
---
 bin/swift-recon | 81 +++++++++++++++++++++++++++++--------------------
 1 file changed, 48 insertions(+), 33 deletions(-)

diff --git a/bin/swift-recon b/bin/swift-recon
index 8df268daeb..68284a5df6 100755
--- a/bin/swift-recon
+++ b/bin/swift-recon
@@ -18,11 +18,13 @@ VERBOSE = False
 SUPPRESS_ERRORS = False
 
 
-def getdevices():
-    #todo , fitler by zone[s]
-    ring_file = "/etc/swift/object.ring.gz"
+def get_devices(zone_filter, ring_file):
     ring_data = Ring(ring_file)
-    ips = set((n['ip'], n['port']) for n in ring_data.devs if n)
+    if zone_filter:
+        ips = set((n['ip'], n['port']) for n in ring_data.devs if n \
+            if n['zone'] == zone_filter)
+    else:
+        ips = set((n['ip'], n['port']) for n in ring_data.devs if n)
     return ips
 
 
@@ -90,11 +92,10 @@ def scout_quarantine(host):
     return url, content, status
 
 
-def get_ringmd5(ringfile):
+def get_ringmd5(hosts, ringfile):
     stats = {}
     matches = 0
     errors = 0
-    hosts = getdevices()
     md5sum = md5()
     with open(ringfile, 'rb') as f:
         block = f.read(4096)
@@ -126,10 +127,8 @@ def get_ringmd5(ringfile):
     print "=" * 79
 
 
-def async_check():
-    ASYNC_COUNTER = 0
+def async_check(hosts):
     stats = {}
-    hosts = getdevices()
     pool = eventlet.GreenPool(20)
     now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print "[%s] Checking async pendings on %s hosts..." % (now, len(hosts))
@@ -148,9 +147,8 @@ def async_check():
     print "=" * 79
 
 
-def umount_check():
+def umount_check(hosts):
     stats = {}
-    hosts = getdevices()
     pool = eventlet.GreenPool(20)
     now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print "[%s] Getting unmounted drives from %s hosts..." % (now, len(hosts))
@@ -163,9 +161,8 @@ def umount_check():
     print "=" * 79
 
 
-def replication_check():
+def replication_check(hosts):
     stats = {}
-    hosts = getdevices()
     pool = eventlet.GreenPool(20)
     now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print "[%s] Checking replication times on %s hosts..." % (now, len(hosts))
@@ -184,11 +181,10 @@ def replication_check():
     print "=" * 79
 
 
-def load_check():
+def load_check(hosts):
     load1 = {}
     load5 = {}
     load15 = {}
-    hosts = getdevices()
     pool = eventlet.GreenPool(20)
     now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print "[%s] Checking load avg's on %s hosts..." % (now, len(hosts))
@@ -211,11 +207,10 @@ def load_check():
     print "=" * 79
 
 
-def quarantine_check():
+def quarantine_check(hosts):
     objq = {}
     conq = {}
     acctq = {}
-    hosts = getdevices()
     pool = eventlet.GreenPool(20)
     now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     print "[%s] Checking quarantine dirs on %s hosts..." % (now, len(hosts))
@@ -238,8 +233,7 @@ def quarantine_check():
     print "=" * 79
 
 
-def disk_usage():
-    hosts = getdevices()
+def disk_usage(hosts):
     stats = {}
     highs = []
     lows = []
@@ -315,6 +309,10 @@ def main():
         help="Get cluster quarantine stats")
     args.add_option('--objmd5', action="store_true",
         help="Get md5sums of object.ring.gz and compare to local copy")
+    args.add_option('--all', action="store_true",
+        help="Perform all checks. Equivelent to -arudlq --objmd5")
+    args.add_option('--zone', '-z', type="int",
+        help="Only query servers in specified zone")
     args.add_option('--swiftdir', default="/etc/swift",
         help="Default = /etc/swift")
     options, arguments = args.parse_args()
@@ -323,24 +321,41 @@ def main():
         args.print_help()
 
     swift_dir = options.swiftdir
+    obj_ring = os.path.join(swift_dir, 'object.ring.gz')
+    con_ring = os.path.join(swift_dir, 'container.ring.gz')
+    acct_ring = os.path.join(swift_dir, 'account.ring.gz')
 
     VERBOSE = options.verbose
     SUPPRESS_ERRORS = options.suppress
 
-    if options.async:
-        async_check()
-    if options.unmounted:
-        umount_check()
-    if options.replication:
-        replication_check()
-    if options.loadstats:
-        load_check()
-    if options.diskusage:
-        disk_usage()
-    if options.objmd5:
-        get_ringmd5(os.path.join(swift_dir, 'object.ring.gz'))
-    if options.quarantined:
-        quarantine_check()
+    if options.zone:
+        hosts = get_devices(options.zone, obj_ring)
+    else:
+        hosts = get_devices(None, obj_ring)
+
+    if options.all:
+        async_check(hosts)
+        umount_check(hosts)
+        replication_check(hosts)
+        load_check(hosts)
+        disk_usage(hosts)
+        get_ringmd5(hosts, obj_ring)
+        quarantine_check(hosts)
+    else:
+        if options.async:
+            async_check(hosts)
+        if options.unmounted:
+            umount_check(hosts)
+        if options.replication:
+            replication_check(hosts)
+        if options.loadstats:
+            load_check(hosts)
+        if options.diskusage:
+            disk_usage(hosts)
+        if options.objmd5:
+            get_ringmd5(hosts, obj_ring)
+        if options.quarantined:
+            quarantine_check(hosts)
 
 
 if __name__ == '__main__':