diff options
author | Marko Kreen | 2010-06-11 11:25:39 +0000 |
---|---|---|
committer | Marko Kreen | 2010-06-11 11:25:39 +0000 |
commit | bd00ea2953934375293bf18462e4339ded1c9fd8 (patch) | |
tree | 89eac3f539aa221fc2a53a47c6a148c2657b64b4 /python/pgq/consumer.py | |
parent | 64acef0e95879763330cc332b3483eb76cbae570 (diff) |
pgq.Consumer: Provide self.batch_info with batch details
Diffstat (limited to 'python/pgq/consumer.py')
-rw-r--r-- | python/pgq/consumer.py | 6 |
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.""" |