diff options
| -rw-r--r-- | python/londiste/setup.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/python/londiste/setup.py b/python/londiste/setup.py index 20d3f5c0..1ce79598 100644 --- a/python/londiste/setup.py +++ b/python/londiste/setup.py @@ -40,6 +40,8 @@ class LondisteSetup(CascadeAdmin): self.set_name = self.queue_name + self.lock_timeout = self.cf.getfloat('lock_timeout', 10) + londiste.handler.load_handler_modules(self.cf) def init_optparse(self, parser=None): @@ -117,6 +119,13 @@ class LondisteSetup(CascadeAdmin): def is_root(self): return self.queue_info.local_node.type == 'root' + def set_lock_timeout(self, curs): + ms = int(1000 * self.lock_timeout) + if ms > 0: + q = "SET LOCAL statement_timeout = %d" % ms + self.log.debug(q) + curs.execute(q) + def cmd_add_table(self, *args): """Attach table(s) to local node.""" @@ -198,6 +207,8 @@ class LondisteSetup(CascadeAdmin): tbl_exists = skytools.exists_table(dst_curs, dest_table) dst_db.commit() + self.set_lock_timeout(dst_curs) + if dest_table == tbl: desc = tbl else: |