summaryrefslogtreecommitdiff
path: root/python/pgq/consumer.py
diff options
context:
space:
mode:
authorPetr Jelinek2013-01-24 13:04:31 +0000
committerPetr Jelinek2013-01-24 13:04:31 +0000
commitc5155488cac0c73774aa4e2788f34f37e28818c2 (patch)
tree196433412e6bfcd7c4c79429ada2d249f510ed3b /python/pgq/consumer.py
parent37d41a17cd2227fd73d2e8aa11ecb70b06948d32 (diff)
Initialize Event properly in BaseConsumer
Diffstat (limited to 'python/pgq/consumer.py')
-rw-r--r--python/pgq/consumer.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py
index 19fbf5ef..3d78105f 100644
--- a/python/pgq/consumer.py
+++ b/python/pgq/consumer.py
@@ -31,12 +31,13 @@ class RetriableBatchWalker(BaseBatchWalker):
"""BatchWalker that returns RetriableEvents
"""
- _event_class = RetriableWalkerEvent
-
def __init__(self, curs, batch_id, queue_name, fetch_size = 300, consumer_filter = None):
super(RetriableBatchWalker, self).__init__(self, curs, batch_id, queue_name, fetch_size, consumer_filter)
self.status_map = {}
+ def _make_event(self, queue_name, row):
+ return RetriableWalkerEvent(self, queue_name, row)
+
def tag_event_done(self, event):
if event.id in self.status_map:
del self.status_map[event.id]