summaryrefslogtreecommitdiff
path: root/python/pgq/consumer.py
diff options
context:
space:
mode:
authorMarko Kreen2010-06-11 11:25:39 +0000
committerMarko Kreen2010-06-11 11:25:39 +0000
commitbd00ea2953934375293bf18462e4339ded1c9fd8 (patch)
tree89eac3f539aa221fc2a53a47c6a148c2657b64b4 /python/pgq/consumer.py
parent64acef0e95879763330cc332b3483eb76cbae570 (diff)
pgq.Consumer: Provide self.batch_info with batch details
Diffstat (limited to 'python/pgq/consumer.py')
-rw-r--r--python/pgq/consumer.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py
index 1f6615d4..20b83fe6 100644
--- a/python/pgq/consumer.py
+++ b/python/pgq/consumer.py
@@ -144,6 +144,8 @@ class Consumer(skytools.DBScript):
pgq_min_interval = None
pgq_min_lag = None
+ batch_info = None
+
def __init__(self, service_name, db_name, args):
"""Initialize new consumer.
@@ -298,8 +300,8 @@ class Consumer(skytools.DBScript):
q = "select * from pgq.next_batch_custom(%s, %s, %s, %s, %s)"
curs.execute(q, [self.queue_name, self.consumer_name,
self.pgq_min_lag, self.pgq_min_count, self.pgq_min_interval])
- inf = curs.fetchone()
- return inf['batch_id']
+ self.batch_info = curs.fetchone()
+ return self.batch_info['batch_id']
def _flush_retry(self, curs, batch_id, list):
"""Tag retry events."""