londiste.syncer: load per-table handler

this is preparation for compare/repair for subset of table.

1 files changed, 8 insertions, 1 deletions

diff --git a/python/londiste/syncer.py b/python/londiste/syncer.py
index b5dd58d0..65fbb6e7 100644
--- a/python/londiste/syncer.py
+++ b/python/londiste/syncer.py
@@ -3,12 +3,17 @@
 """
 
 import sys, time, skytools
+from londiste.handler import build_handler, load_handler_modules
 
 class ATable:
     def __init__(self, row):
         self.table_name = row['table_name']
         self.dest_table = row['dest_table'] or row['table_name']
         self.merge_state = row['merge_state']
+        attrs = row['table_attrs'] or ''
+        self.table_attrs = skytools.db_urldecode(attrs)
+        hstr = self.table_attrs.get('handler', '')
+        self.plugin = build_handler(self.table_name, hstr, row['dest_table'])
 
 class Syncer(skytools.DBScript):
     """Walks tables in primary key order and checks if data matches."""
@@ -35,6 +40,8 @@ class Syncer(skytools.DBScript):
         if self.pidfile:
             self.pidfile += ".repair"
 
+        load_handler_modules(self.cf)
+
     def set_lock_timeout(self, curs):
         ms = int(1000 * self.lock_timeout)
         if ms > 0:
@@ -74,7 +81,7 @@ class Syncer(skytools.DBScript):
         Returns tuple of (dict(name->ATable), namelist)"""
 
         curs = db.cursor()
-        q = "select table_name, merge_state, dest_table"\
+        q = "select table_name, merge_state, dest_table, table_attrs"\
             " from londiste.get_table_list(%s) where local"
         curs.execute(q, [self.queue_name])
         rows = curs.fetchall()