summaryrefslogtreecommitdiff
path: root/python/pgq/setconsumer.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pgq/setconsumer.py')
-rw-r--r--python/pgq/setconsumer.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/python/pgq/setconsumer.py b/python/pgq/setconsumer.py
index 268cd7a8..baa18b28 100644
--- a/python/pgq/setconsumer.py
+++ b/python/pgq/setconsumer.py
@@ -27,7 +27,7 @@ class SetConsumer(skytools.DBScript):
dst_node = self.load_node_info(dst_db)
if self.main_worker:
- self.consumer_name = dst_node.worker_name
+ self.consumer_name = dst_node.name
if not dst_node.up_to_date:
self.tag_node_uptodate(dst_db)
@@ -188,7 +188,7 @@ class SetConsumer(skytools.DBScript):
mbr_list = curs.dictfetchall()
db.commit()
- return NodeInfo(node_row, self.main_worker)
+ return NodeInfo(self.set_name, node_row, self.main_worker)
def tag_node_uptodate(self, dst_db):
dst_curs = dst_db.cursor()
@@ -197,8 +197,8 @@ class SetConsumer(skytools.DBScript):
dst_db.commit()
def copy_tick(self, dst_curs, src_queue, dst_queue):
- q = "select * from pgq.ticker(%s, %s)"
- dst_curs.execute(q, [dst_queue.queue_name, src_queue.cur_tick])
+ q = "select * from pgq.ticker(%s, %s, %s)"
+ dst_curs.execute(q, [dst_queue.queue_name, src_queue.cur_tick, src_queue.tick_time])
def set_tick_complete(self, dst_curs, tick_id):
q = "select * from pgq_set.set_completed_tick(%s, %s, %s)"