1 files changed, 45 insertions, 0 deletions
diff --git a/python/londiste/compare.py b/python/londiste/compare.py
new file mode 100644
index 00000000..0029665b
--- /dev/null
+++ b/python/londiste/compare.py
@@ -0,0 +1,45 @@
+#! /usr/bin/env python
+
+"""Compares tables in replication set.
+
+Currently just does count(1) on both sides.
+"""
+
+import sys, os, time, skytools
+
+__all__ = ['Comparator']
+
+from syncer import Syncer
+
+class Comparator(Syncer):
+    def process_sync(self, tbl, src_db, dst_db):
+        """Actual comparision."""
+
+        src_curs = src_db.cursor()
+        dst_curs = dst_db.cursor()
+
+        self.log.info('Counting %s' % tbl)
+
+        q = "select count(1) from only _TABLE_"
+        q = self.cf.get('compare_sql', q)
+        q = q.replace('_TABLE_', tbl)
+
+        self.log.debug("srcdb: " + q)
+        src_curs.execute(q)
+        src_row = src_curs.fetchone()
+        src_str = ", ".join(map(str, src_row))
+        self.log.info("srcdb: res = %s" % src_str)
+
+        self.log.debug("dstdb: " + q)
+        dst_curs.execute(q)
+        dst_row = dst_curs.fetchone()
+        dst_str = ", ".join(map(str, dst_row))
+        self.log.info("dstdb: res = %s" % dst_str)
+
+        if src_str != dst_str:
+            self.log.warning("%s: Results do not match!" % tbl)
+
+if __name__ == '__main__':
+    script = Comparator(sys.argv[1:])
+    script.start()
+